1. 程式人生 > >【leetcode 498. 對角線遍歷】Diagonal Traverse

【leetcode 498. 對角線遍歷】Diagonal Traverse

給定一個含有 M x N 個元素的矩陣(M行,N列),請以對角線遍歷的順序返回這個矩陣中的所有元素,對角線遍歷如下圖所示。

示例:

輸入:
[
 [ 1, 2, 3 ],
 [ 4, 5, 6 ],
 [ 7, 8, 9 ]
]
輸出:  [1,2,4,7,5,3,6,8,9]
解釋:

說明:

  1. 給定矩陣中的元素總數不會超過 100000 。

分析題目,遍歷分左下到左上和右上到右下

觀察每個對角線上的元素髮現在同一對角線上元素的座標和為一定值,如第一的對角線為2,依次為3,4,5,6。最大到行數和列數之和。

故可從2開始到行數和列數和遍歷。每一個值對應一條對角線,行依次減,列依次對應加為左下到左上。行依次加,列依次減對應右上到右下,注意邊界條件即可。

/**
 * Return an array of size *returnSize.
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* findDiagonalOrder(int** matrix, int matrixRowSize, int matrixColSize, int* returnSize) {
    *returnSize = matrixRowSize * matrixColSize;  
    int size = matrixRowSize + matrixColSize;
    int bit = 0,row,col,state=0;
    int *obj = (int*)malloc((*returnSize) * sizeof(int));
    
    for(int i=2;i<=size;i++){       
        if(bit==0){
            if(i>matrixRowSize + 1) col = i - matrixRowSize;
            else col = 1;
            
            for(int j=col;j<i && j-1 < matrixColSize && i-j-1<matrixRowSize;j++){
                obj[state] = matrix[i-j-1][j-1];
                state++;               
            }           
            bit = 1;
            continue;
        }
        if(bit==1){
             if(i>matrixColSize + 1) row = i - matrixColSize;
            else row = 1;
            
             for(int j=row;j<i && j-1<matrixRowSize && i-j-1<matrixColSize;j++){
                 obj[state] = matrix[j-1][i-j-1];
                state++;                
            }
            
            bit = 0; 
        }
    }   
    return obj;
    
}