矩阵转置函数——指针自增的陷阱

#include"algorithm.h" int** transpose(int**po, int &m, int &n) { int*p = (int*)malloc(m*n*sizeof(int));//转置矩阵空间 int**pn = (int**)malloc(n*sizeof(int*));//转置矩阵数组指针 //指针备份 int *save = po[0]; int *nSave = p; //布置转置矩阵 // [1 2 3 4 5 6] --> [1 4 2 5 3 6] for (int j = 0; j < n; j++) { for (int i = 0; i < m; i++) { /*注意这里都有自增，指针的值变化了，就可能出现堆头错误!*/ *p++ = *po[i];//按地址录入元素 po[i] = po[i] + 1;//更新地址 } } //还原指针指向堆头,坑!!! p = nSave; po[0] = save; //布置转置矩阵数组指针 for (int j = 0; j < n; j++) { pn[j] = &p[j*m]; } //释放原矩阵空间 free(po[0]); free(po); swap(m, n); return pn; } int main() { int m, n; scanf("%d%d", &m, &n); int**p = new_arr2(m, n); if (!p) return -1; scan_arr2(p, m, n); p = transpose(p, m, n); print_arr2(p, m, n); free(p[0]); free(p); return 0; }

指针变化后不再指向原处，如果指针指向的是堆(malloc来的)，则free时不注意就会导致堆头错误，上面有二级指针的话二级指针也无法正确解包拿到自增前的数据，比如这次实验就产生矩阵转置后全是杂乱值的情况，那是一级指针已经飘了，二级也遭殃了，所以要注意指针的变化是有潜在的陷阱的。

---

该程序改进方法是不要对传进的参数直接修改，而是新建临时指针变量用于处理，就像ps一般不直接对原图处理而是新建副本再处理。