C语言实现数组转置的代码详解

  地址:    ... | +0     | +1     | ... | +N-1   | +N     | +N+1  | ...
  元素:    ... | a[0][0]| a[0][1]| ... | a[0][N-1]| a[1][0]| a[1][1]| ...

for (i = 0; i < n; ++i)
  for (j = i+1; j < n; ++j)
    swap(A[i*n + j], A[j*n + i]);

for (ii = 0; ii < rows; ii += B)
  for (jj = 0; jj < cols; jj += B)
    // 对矩阵子块 (ii..ii+B-1, jj..jj+B-1) 进行单独转置
    for (i = ii; i < min(ii+B, rows); ++i)
      for (j = jj; j < min(jj+B, cols); ++j)
        dst[j*rows + i] = src[i*cols + j];

#pragma omp parallel for collapse(2)
for (i = 0; i < rows; ++i)
  for (j = 0; j < cols; ++j)
    dst[j*rows + i] = src[i*cols + j];

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <omp.h>
 
#ifndef MIN
#define MIN(a,b) (((a)<(b))?(a):(b))
#endif
 
/**
 * 基本额外空间转置
 * @param src   原矩阵指针
 * @param rows  行数
 * @param cols  列数
 * @param dst   目标矩阵指针（已分配 rows*cols 大小）
 */
void transpose_with_buffer(int *src, int rows, int cols, int *dst) {
    for (int i = 0; i < rows; ++i) {
        for (int j = 0; j < cols; ++j) {
            dst[j * rows + i] = src[i * cols + j];
        }
    }
}
 
/**
 * 方阵原地转置
 * 适用于 n x n 方阵
 */
void transpose_inplace(int *a, int n) {
    for (int i = 0; i < n; ++i) {
        for (int j = i + 1; j < n; ++j) {
            int tmp = a[i * n + j];
            a[i * n + j] = a[j * n + i];
            a[j * n + i] = tmp;
        }
    }
}
 
/**
 * 块转置 (块大小 block_size)
 * @param src         原矩阵
 * @param rows,cols   原矩阵尺寸
 * @param block_size  块大小
 * @param dst         目标矩阵
 */
void transpose_block(int *src, int rows, int cols, int block_size, int *dst) {
    for (int ii = 0; ii < rows; ii += block_size) {
        for (int jj = 0; jj < cols; jj += block_size) {
            int max_i = MIN(ii + block_size, rows);
            int max_j = MIN(jj + block_size, cols);
            for (int i = ii; i < max_i; ++i) {
                for (int j = jj; j < max_j; ++j) {
                    dst[j * rows + i] = src[i * cols + j];
                }
            }
        }
    }
}
 
/**
 * OpenMP 并行转置 (基本方法)
 */
void transpose_omp(int *src, int rows, int cols, int *dst) {
    #pragma omp parallel for collapse(2)
    for (int i = 0; i < rows; ++i) {
        for (int j = 0; j < cols; ++j) {
            dst[j * rows + i] = src[i * cols + j];
        }
    }
}
 
/**
 * 性能测试主函数
 */
int main(int argc, char *argv[]) {
    int rows = 4096, cols = 4096;
    int *A = (int*)malloc(sizeof(int) * rows * cols);
    int *B = (int*)malloc(sizeof(int) * rows * cols);
    if (!A || !B) {
        fprintf(stderr, "内存分配失败\n");
        return EXIT_FAILURE;
    }
 
    // 初始化
    for (int i = 0; i < rows; ++i)
        for (int j = 0; j < cols; ++j)
            A[i * cols + j] = i * cols + j;
 
    struct timespec t1, t2;
    double elapsed;
 
    // 1. 额外空间转置
    clock_gettime(CLOCK_MONOTONIC, &t1);
    transpose_with_buffer(A, rows, cols, B);
    clock_gettime(CLOCK_MONOTONIC, &t2);
    elapsed = (t2.tv_sec - t1.tv_sec) + (t2.tv_nsec - t1.tv_nsec)/1e9;
    printf("Buffer Transpose: %.6f s\n", elapsed);
 
    // 2. 原地方阵转置 (只针对方阵 A)
    clock_gettime(CLOCK_MONOTONIC, &t1);
    transpose_inplace(A, cols);
    clock_gettime(CLOCK_MONOTONIC, &t2);
    elapsed = (t2.tv_sec - t1.tv_sec) + (t2.tv_nsec - t1.tv_nsec)/1e9;
    printf("In-place Square Transpose: %.6f s\n", elapsed);
 
    // 3. 块转置
    clock_gettime(CLOCK_MONOTONIC, &t1);
    transpose_block(A, rows, cols, 64, B);
    clock_gettime(CLOCK_MONOTONIC, &t2);
    elapsed = (t2.tv_sec - t1.tv_sec) + (t2.tv_nsec - t1.tv_nsec)/1e9;
    printf("Block Transpose (64): %.6f s\n", elapsed);
 
    // 4. OpenMP 并行转置
    clock_gettime(CLOCK_MONOTONIC, &t1);
    transpose_omp(A, rows, cols, B);
    clock_gettime(CLOCK_MONOTONIC, &t2);
    elapsed = (t2.tv_sec - t1.tv_sec) + (t2.tv_nsec - t1.tv_nsec)/1e9;
    printf("OpenMP Parallel Transpose: %.6f s\n", elapsed);
 
    free(A);
    free(B);
    return 0;
}