三角螺旋阵

方阵的主对角线之上称为“上三角”。
请你设计一个用于填充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;
    }
};