题目内容
给定一个M*N大小的矩阵(M行,N列),要求返还一个以下图所示方法得到的一个一维数组
这道题从题意来看,很简洁,要求按照对角线以S型遍历整个数组,从而得到一个特定顺序的一维数组,如果是人来完成这个任务,恐怕五六岁的小朋友都能轻易做到。但要怎么让电脑完成这件事呢?你需要注意到这几个点:
- 上图中任一对角线上的所有点, 横纵坐标之和是相同的。
- 横纵坐标之和的奇偶性决定了对角线应该向右上方遍历,还是应该向左下方遍历,即横纵坐标的变化方法;当和为奇数时,向左下方遍历;和为偶数时,向右上方遍历。
- 最重要的一点,每次对角线遍历到达上下左右四个边界后,应该怎么选择起点以及更新横纵坐标变换的方式,下面我将结合代码对这一点加以阐述。
class Solution {
public:
vector<int> findDiagonalOrder(vector<vector<int>>& matrix) {
vector<int> result;
if (matrix.size() == 0) return result;
// 初始遍历方向为右上方
int row = matrix.size(), col = matrix[0].size();
int sum = 0, x = 0, y = 0, dx = -1, dy = 1;
while (x < row && y < col) {
result.push_back(matrix[x][y]);
x += dx;
y += dy;
if (x < 0 || y < 0 || x >= row || y >= col) {
++sum;
if (x < 0) {
if (y < 0) {
// 这种情况不会发生
} else if (y >= col) {
// 遍历接触到右侧边界线
x += 2;
y = col - 1;
} else {
// 遍历接触到上侧边界线
x = 0;
}
} else if (x >= row) {
x = r - 1;
if (y < 0) {
// 遍历接触到下侧边界线
y += 2;
} else if (y >= col) {
// this condition won't happen
} else {
// 遍历接触到下侧边界线
y += 2;
}
} else {
if (y < 0) {
// 遍历接触到左侧边界线
y = 0;
} else if (y >= col) {
// 遍历接触到右侧边界线
y = col - 1;
x += 2;
}
}
// 根据奇偶性确定下一次的遍历方向
if (sum % 2 == 1) {
dx = 1;
dy = -1;
} else {
dx = -1;
dy = 1;
}
}
}
return result;
}
};