23、【数组】对角线遍历（C++版）

题目描述

题目分析

对一个3×3型数组的对角线遍历，重点在于对于数组索引的分析，首先来看一下上述图中，遍历索引的顺序

0,0					// 第一趟
0,1 1,0				// 第二趟
2,0 1,1 0,2			// 第三趟
1,2 2,1				// 第四趟
2,2					// 第五趟

设行索引编号为x,列索引编号为y，趟数值为z。
首先，从趟数和行列索引编号的关系上，可发现在对角线遍历时，
x + y = z - 1恒成立。因此，便可将此公式作为我们的遍历过程的便界条件。

然后，再从每一趟的情况来看，

---

第一趟，0,0；
第二趟，0,1 1,0，x先为0，然后再为1，过程中x逐渐增大，y逐渐减小；
第三趟，2,0 1,1 0,2，x先为2，然后再为1，然后再为0，过程中x逐渐减小，y逐渐增大；
第四趟，1,2 2,1，x先为1，再为2，过程中x逐渐增大，y逐渐减小；
第五趟，2 2。

---

可发现在2，4时，x先小后大，y先大后小，在3时，x先大后小，y先小后大。
那么我们假设，在奇数次的趟数时，均为x先大后小，y先小后大。在偶数次趟数时，均为x先小后大，y先大后小。

参考文章：
【对角线遍历】 分析题目-找准规律-代码清晰易读-不要让自己迷失在随意的ifelse中

代码实现

vector<int> findDiagonalOrder(vector<vector<int>>& matrix){
	vector<int> nums;
	int m = matrix.size();
	if(m == 0)		return nums;
	int n = matrix.size();
	if(n == 0)		return nums;

	bool bXFlag = true;		// 作为开关，控制数值的初始化，初始化为奇数趟次数的情况还是偶数趟次数的情况
	for(int i = 0; i < m + n; i++){
		// 为true时，初始化为奇数躺次数的情况；为false时，初始化为偶数趟次数的情况。
		int pm = bXFlag ? m : n;		
		int pn = bXFlag ? n : m;

		int x = (i < pm) ? i : pm - 1;
		int y = i - x;

		while(x >= 0 && y < pn){
			nums.push_back(bXFlag ? matrix[x][y] : matrix[y][x]);
		}
	}
}