主对角线打印二维数组---美团点评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]

// ConsoleApplication5.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"
#include <iostream>
#include <vector>
#include <string>
#include <map>

using namespace::std;

class Printer {
public:
	vector<int> arrayPrint(vector<vector<int> > vec, int n) {
		// write code here
		vector<int> result;

		int col = n - 1;
		for (int i = col; i >= 0; --i) {
			int row = 0;
			int k = n - i;
			for (int j = i; k > 0; ++j, --k) {
				result.push_back(vec[row][j]);
				++row;
			}
		}

		int row = 1;
		for (int i = row; i <= n - 1; ++i) {
			int k = n - i;
			int col = 0;
			for (int j = i; k > 0; --k, ++j) {
				result.push_back(vec[j][col]);
				++col;
			}
		}

		return result;
	}
};
int main() {
	vector<vector<int>> vec = { { 1, 2, 3, 4 }, { 5, 6, 7, 8 }, { 9, 10, 11, 12 }, { 13, 14, 15, 16 } };
	Printer obj;
	vector<int> result = obj.arrayPrint(vec, vec.size());

	return 0;
}