书接上上文串行算法的矩阵转置
以及上文块棋盘法多线程的矩阵转置
我们来介绍介绍多线程实现矩阵转置的另一种方法,直角划分法。
那么我们就一起来看看如何做到的吧。
首先,我们要知道什么是直角。
Q:什么是直角呢?
A:??不是吧阿sir,直角你都不知道,不是吧不是吧。
没错,就是我们想象中的那个直角,这道题大家都在第一层,没有第二层,更没有第五层。那么就显而易见了,只要把矩阵划分成一个一个直角即可,然后一个线程负责一个直角即可。
图。。我以我胎教的美术水平画一个。
#include <pthread.h>
#include <stdlib.h>
#include <stdio.h>
#include <math.h>
int **matrix;
int **tmp;
int T,n;
void init(){
matrix = (int **)malloc(n * sizeof(int *));
tmp = (int **)malloc(n * sizeof(int *));
for(int i=0;i<n;i++){
matrix[i] = (int *)malloc(n * sizeof(int));
tmp[i] = (int *)malloc(n * sizeof(int));
}
int cnt = 0;
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
matrix[i][j] = tmp[i][j] = cnt;
cnt++;
}
}
}
void *thread_function2(void *arg){
int begin = *(int *)arg;//确定该线程id
int piece = n/T;//确定线程负责的规模
int end = (begin == T-1) ? n : begin + piece;//确定线程的截至
begin = begin * piece; //确定线程的起始
for(int rank=begin;rank<end;rank++){
for(int i=0;i<rank;i++){
int tmp = matrix[rank][i];
matrix[rank][i] = matrix[i][rank];
matrix[i][rank] = tmp;
}
}
return NULL;
}
void free(int n){
free(matrix);
}
void print(int n){
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
printf("%d ", matrix[i][j]);
}
printf("\n");
}
}
int main(int argc, const char * argv[]) {
n = 10;//atoi(argv[1]);
T = 2;//atoi(argv[2]);
init();
int x[T];
pthread_t thread[T];
for(int i=0;i<T;i++){
x[i] = i;
pthread_create(&thread[i], NULL, thread_function2, &x[i]);
}
for(int i=0;i<T;i++){
pthread_join(thread[i], NULL);
}
print(n);
return 0;
}
结果如下
0 10 20 30 40 50 60 70 80 90
1 11 21 31 41 51 61 71 81 91
2 12 22 32 42 52 62 72 82 92
3 13 23 33 43 53 63 73 83 93
4 14 24 34 44 54 64 74 84 94
5 15 25 35 45 55 65 75 85 95
6 16 26 36 46 56 66 76 86 96
7 17 27 37 47 57 67 77 87 97
8 18 28 38 48 58 68 78 88 98
9 19 29 39 49 59 69 79 89 99
好了,这就是直角划分法实现矩阵转置的全部内容了。
其实程序还有改进空间,我们很明显看出线程的工作量是负载不均衡的,所以如何负载均衡也是个值得考虑的问题。
希望大家不吝赐教,共同进步。