美团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]
分析:
这个题目是一个模拟题,我们需要模拟一下打印的走势即可。
1 2 3 4
5 6 7 8
9 10 11 12
13 14 15 16
从这个线条的模拟方式,我们可以看出,只需要模拟每一个箭头的开始和结尾,如果到达一个箭头的尾部,就调到下一个线的末端。我们可以看到,末尾的位置[i,j]一定满足 i = n 或者 j = n;开始的位置一定是 i = 1,或者j = 1;(一般我们表示的时候是0和n-1)。如果i = n或者 j = n;我们如何转到下一个线开始呢?
如果是j = n;则下个开始位置是i = 1; j = k;(其中k一个模拟的位置,k = n,一直递减)。同理当 i=n时,下个位置是i=k,j=1;(k是一个模拟位置,一直递加)。
Code:
#include<iostream>
#include<stdio.h>
#include<vector>
using namespace std;
int a[][4] = {{1,2,3,4},{5,6,7,8},{9,10,11,12},{13,14,15,16}};
vector<int> print(int n){
vector<int> result;
int begin=0,end=n-1,k = 0;
//前半部分
while(end > -1){
//每次的行都是从0开始,列递加
int i = end;
begin = 0;
while(i <= n-1){
result.push_back(a[begin][i]);
begin ++;
i ++;
}
--end;
}
//下半部分
begin = 1; end = 0;
while(begin < n){
//每次的列都是从0开始,行递加
int i = begin;
end = 0;
while(i <= n-1){
result.push_back(a[i][end]);
i++;
end++;
}
++begin;
}
return result;
}
int main()
{
vector<int> result = print(4);
for(int i=0;i<result.size();i++){
cout<<result[i]<<" ";
}
cout<<endl;
return 0;
}