Careercup

2014-05-06 01:40

题目链接

原题：

Give a N*N matrix, print it out diagonally. 
Follow up, if it is a M*N matrix, how to print it out. 
Example: 
1 2 3 
4 5 6 
7 8 9 
print: 
1 
2 4 
3 5 7 
6 8 
9

题目：按照题目示例给出的方式，输出一个矩阵。

解法：按题意写代码即可。

代码：

// http://www.careercup.com/question?id=5661939564806144
#include <cstdio>
#include <vector>
using namespace std;

class Solution {
public:
    void printDiagonal(vector<vector<int> > &matrix) {
        int n, m;
        
        n = (int)matrix.size();
        if (n == 0) {
            return;
        }
        m = (int)matrix[0].size();
        if (m == 0) {
            return;
        }
        
        int i, j;
        bool first;
        
        for (i = 0; i < n + m; ++i) {
            first = true;
            for (j = (i < m ? i : m - 1); j >= 0; --j) {
                if (i - j < 0 || i - j > n - 1) {
                    break;
                }
                printf((first ? "%d" : " %d"), matrix[i - j][j]);
                first = false;
            }
            printf("
");
        }
    };
};