方阵的主对角线之上称为“上三角”。
请你设计一个用于填充n阶方阵的上三角区域的程序。填充的规则是:使用1,2,3….的自然数列,从左上角开始,按照顺时针方向螺旋填充。
例如:当n=3时,输出:
1 2 3
6 4
5
当n=4时,输出:
1 2 3 4
9 10 5
8 6
7
当n=5时,输出:
1 2 3 4 5
12 13 14 6
11 15 7
10 8
9
程序运行时,要求用户输入整数n(3~20)
程序输出:方阵的上三角部分。
要求格式:每个数据宽度为4,右对齐。
//
// s_20.cpp
// swordOffer
//
// Created by YangKi on 2017/2/14.
// Copyright © 2017年 yangqi916. All rights reserved.
//
#include <iostream>
#include <vector>
#include <stack>
using namespace std;
class Solution {
private:
vector<int>res;
int currentDirection = 0;
int rowIncre[3] = {0, 1, -1};
int colIncre[3] = {1, -1, 0};
int rows = 0;
int cols = 0;
bool isValid(int row, int col, int n) {
if((row >= 0 && row < n) && (col >= 0 && col < n-row) )
return true;
return false;
}
public:
vector<int> printMatrix(vector<vector<int> > matrix, int n) {
rows = (int)matrix.size();
if (rows == 0)
return res;
cols = (int)(matrix[0].size());
if(cols == 0)
return res;
vector<vector<bool>>isVisited(rows ,vector<bool>(cols, false));
int curRow = 0;
int curCol = 0;
int cnt = 0;
int maxnodes = (n*(n+1)) / 2;
while (cnt < maxnodes) {
cnt++;
isVisited[curRow][curCol] = true;
res.push_back(matrix[curRow][curCol]);
if (!isValid(curRow + rowIncre[currentDirection], curCol + colIncre[currentDirection], n)
|| isVisited[curRow + rowIncre[currentDirection]][curCol + colIncre[currentDirection]] == true) {
// 沿着原来的方向取下一个位置是非法的,需要换方向
currentDirection = (currentDirection + 1) % 3;
}
//取下一个位置
curRow += rowIncre[currentDirection];
curCol += colIncre[currentDirection];
}
return res;
}
};