vector<int> findDiagonalOrder(vector<vector<int>>& mat) {

int m = mat.size();

int n = mat[0].size();

vector<int> res;

int i = 0, j = 0; // 从左上角开始遍历，先向右上方向走

int dir = 1; // 方向：0: 左下、1: 右上

while (true) {

res.emplace_back(mat[i][j]);

if (i == m-1 && j == n-1) break;

if (dir == 1) { // 走对角线：右上

i--, j++;

if (i == -1 && j == n) // 走到右上角

i = 1, j = n -1, dir = 0; // 向下走

else if (i == -1) // 走到第一行

i = 0, dir = 0; // 右右走

else if (j == n) // 走到最后一列

i += 2, j = n - 1, dir = 0; // 向下走

} else if (dir == 0) { // 走对角线：左下

i++, j--;

if (i == m && j == -1) // 走到左下角

i = m-1, j = 1, dir = 1; // 向右走

else if (j == -1) // 已走到第一列

j = 0, dir = 1; // 向下走

else if (i == m) // 走到最后一行

i = m - 1, j += 2, dir = 1; // 向右走

}

}

return res;

}