LeetCode对角线遍历

本题参考LeetCode部分代码，标注自己的理解完成以方便自己今后查看。

#include<iostream>
#include<vector>
using namespace std;
vector<int> findDiagonalOrder(vector<vector<int> >& matrix) {

	vector<int> nums;
	int m = matrix.size();
	int n = matrix[0].size();
	int i = 0;	// i 是 x + y 的和
	while (i < m + n-1)
	{
		// 第 1 3 5 ... 趟
		int x1 = (i < m) ? i : m - 1;	//x从大到小，x尽量取大，x最大为i或m-1 
		int y1 = i - x1;
		while (x1 >= 0 && y1 < n)
		{
			nums.push_back(matrix[x1][y1]);
			// cout<<"x1="<<x1<<"y1="<<y1<<endl;
			x1--;
			y1++;
		}
		i++;
		// cout<<"第"<<i<<"趟"<<endl; 
		if (i >= m + n-1) break;
		// 第 2 4 6 ... 趟
		int y2 = (i < n) ? i : n - 1;	//x从小到大，y尽量取大，y最大为i或n-1 
		int x2 = i - y2;
		while (y2 >= 0 && x2 < m)
		{
			nums.push_back(matrix[x2][y2]);
			// cout<<"x2="<<x2<<"y2="<<y2<<endl;
			x2++;
			y2--;
		}
		i++;
		// cout<<"第"<<i<<"趟"<<endl; 
	}
	return nums;
    
}

int main(){
	vector<int> a1;
	for(int i=1;i<=3;i++){
		a1.push_back(i);
	}
	vector<int> a2;
	for(int i=4;i<=6;i++){
		a2.push_back(i);
	}
	vector<int> a3;
	for(int i=7;i<=9;i++){
		a3.push_back(i);
	}
	vector<vector<int> > v;
	v.push_back(a1);
	v.push_back(a2);
	v.push_back(a3);

//	vector<vector<int> > v = {{1,2,3},{4,5,6},{7,8,9}};
	vector<int> c = findDiagonalOrder(v);
	vector<int>::iterator it= c.begin();
	for(int i=0;i<c.size();i++){
		cout<<*(it+i)<<" ";
	}
	return 0;
}