C++完成矩陣原地轉置算法。本站提示廣大學習愛好者:(C++完成矩陣原地轉置算法)文章只能為提供參考,不一定能成為您想要的結果。以下是C++完成矩陣原地轉置算法正文
本文實例描寫了C++完成矩陣原地轉置算法,是一個異常經典的算法,信任關於進修C++算法的同伙有很年夜的贊助。詳細以下:
1、成績描寫
微軟面試題:將一個MxN的矩陣存儲在一個一維數組中,編程完成矩陣的轉置。
請求:空間龐雜度為O(1)
2、思緒剖析
上面以一個4x2的矩陣A={1,2,3,4,5,6,7,8}停止剖析,轉置進程以下圖:
圖中右下角的白色數字表現在一維數組中的下標。矩陣的轉置其實就是數組中元素的挪動,詳細的挪動進程以下圖:
我們發明,這些挪動的元素的下標是一個個環,下標1的元素挪動到4,下標4的元素挪動到2,下標2的元素挪動到1。在編寫法式的時刻,我們須要處理兩個成績:第一個是若何剖斷環能否反復(已處置過);第二個是若何盤算以後元素下標的先驅與後繼。
第一個成績:若何斷定環是反復已處置過的?由於我們遍歷全部數組時下標是從小到年夜的,所以假如是第一次遍歷該環,則第一個下標確定是這個環中最小的。假如一個環被處置過,那末總能找到一個它的後繼是小於它的。從上圖可以顯著看出來。
第二個成績:若何盤算以後元素下標的先驅與後繼?假定轉置前某個元素的數組下標為i,則它地點行列為(i/N, i%N),轉置後地點行列則為(i%N, i/N),可盤算轉置後數組下標為(i%N)*M+i/N,此為i的後繼。假定轉置後某個元素的數組下標為i,則它地點行列為(i/M, i%M),則轉置前地點行列為(i%M, i/M),可盤算此時下標為(i%M)*N+i/M,此為i的先驅。
3、代碼完成以下:
/************************************************************************* > File Name: matrix_transpose.cpp > Author: SongLee ************************************************************************/ #include<iostream> using namespace std; /* 後繼 */ int getNext(int i, int m, int n) { return (i%n)*m + i/n; } /* 先驅 */ int getPre(int i, int m, int n) { return (i%m)*n + i/m; } /* 處置以下標i為終點的環 */ void movedata(int *mtx, int i, int m, int n) { int temp = mtx[i]; // 暫存 int cur = i; // 以後下標 int pre = getPre(cur, m, n); while(pre != i) { mtx[cur] = mtx[pre]; cur = pre; pre = getPre(cur, m, n); } mtx[cur] = temp; } /* 轉置,即輪回處置一切環 */ void transpose(int *mtx, int m, int n) { for(int i=0; i<m*n; ++i) { int next = getNext(i, m, n); while(next > i) // 若存在後繼小於i解釋反復 next = getNext(next, m, n); if(next == i) // 處置以後環 movedata(mtx, i, m, n); } } /* 輸入矩陣 */ void print(int *mtx, int m, int n) { for(int i=0; i<m*n; ++i) { if((i+1)%n == 0) cout << mtx[i] << "\n"; else cout << mtx[i] << " "; } } /* 測試 */ int main() { int matrix[4*2] = {1,2,3,4,5,6,7,8}; cout << "Before matrix transposition:" << endl; print(matrix, 4, 2); transpose(matrix, 4, 2); cout << "After matrix transposition:" << endl; print(matrix, 2, 4); return 0; }
運轉成果以下圖所示: