对角线打印二维数组----美团2016研发工程师编程题(二)

[编程题] 二维数组打印

有一个二维数组(n*n),写程序实现从右上角到左下角沿主对角线方向打印。

给定一个二位数组arr及题目中的参数n，请返回结果数组。

测试样例：

[[1,2,3,4],[5,6,7,8],[9,10,11,12],[13,14,15,16]],4

返回：[4,3,8,2,7,12,1,6,11,16,5,10,15,9,14,13]

没啥难度，就是细心

// 二维数组打印.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"
#include <vector>

using namespace::std;

vector<int> arrayPrint(vector<vector<int> > arr, int n) {
	int row = 0;
	int count = 1;
	vector<int> result;
	for (int i = 1; i <= n; i++)
	{
		for (int r = 0, c = n - count; r <= i - 1, c<n; r++, c++)
		{
			result.push_back(arr[r][c]);
			// cout<<r<<","<<c<<endl;
		}
		count++;
	}
	if (n <= 1)
		return result;
	count = n - 1;
	for (int i = n - 1; i >= 1; i--)
	{
		for (int r = n - count, c = 0; r <= n - 1, c <= i - 1; c++, r++)
			result.push_back(arr[r][c]);
		count--;
	}
	return result;
}
int _tmain(int argc, _TCHAR* argv[])
{
	vector<vector<int>> vec;
	vector<int> test1 = { 1, 2, 3, 4 };
	vector<int> test2 = { 5, 6, 7, 8 };
	vector<int> test3 = { 9, 10, 11, 12 };
	vector<int> test4 = { 13, 14, 15, 16 };

	vec.push_back(test1);
	vec.push_back(test2);
	vec.push_back(test3);
	vec.push_back(test4);

	vector<int> result = arrayPrint(vec, 4);
	return 0;
}