• 周二. 7 月 16th, 2024

5G编程聚合网

5G时代下一个聚合的编程学习网

热门标签

螺旋矩阵问题

admin

11 月 28, 2021

刷的leetcode中的题,其实之前写过一篇,不过当时的写法比较傻,遇见矩阵大一点的貌似就有问题。

这次再刷到一样的题目,就重新写了一下。

给你一个正整数 n ,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。
思路:
一圈一圈的递归走下去就好了。
然后每一圈都分为上下左右4个方向前进。
我们要做的,就是做好上下左右的操作
#include<iostream>
#include<vector>
using namespace std;
class Solution {
public:
    void setValue(int point,int value,int size,vector<vector<int>>& v)
    {
        if(size<0)
        {
            return;
        }
        if(size==1)
        {
            v[point][point] = value;
            return;
        }
//        cout<<"point = "<<point<<" value = "<<value<<" size = "<<size<<endl;
        int total = 4*size-4;
        int array[4] = {};//对应上下左右中要计算的次数 
        array[0] = size;
        array[1] = 2*size - 2;//其实没有用到,直接简化成了2*size-2-size = size-2
        array[2] = 3*size - 2;
        array[3] = 4*size - 4;
        int nowindex = size;
        int y1 = point+size-1;
        for(int i=0;i<4;i++)
        {
            if(i==0)
            {
                for(int j=0;j<size;j++)//y不变,x++;
                {
                    nowindex = point + j;
                    v[point][nowindex] = value++;
                }
            }
            else if(i==1)
            {
                for(int j=0;j<size-2;j++)//x不变,y++ 
                {
                    nowindex = size + j;
                    v[point+1+j][y1] = value++;
                }
            }
            else if(i==2)
            {
                for(int j=0;j<size;j++)//y不变,x-- 
                {
                    v[y1][y1-j]=value++;
                }
            }
            else if(i==3)
            {
                for(int j=0;j<size-2;j++)//x不变,y-- 
                {
                    v[y1-1-j][point]=value++;
                }
            }
        }
        point++;
        size = size - 2;
        setValue(point,value,size,v);
    }
    vector<vector<int>> generateMatrix(int n) {
        vector<int> v1;
        vector<vector<int>> v;
        v1.resize(n);
        for(int i=0;i<n;i++)
        {
            v.push_back(v1);
        }
        setValue(0,1,n,v);
//        for(auto i:v)
//        {
//            for(auto j:i)
//            {
//                cout<<j<<" ";
//            }
//            cout<<endl;
//        }
        return v;
    }
};
int main()
{
    Solution s;
    s.generateMatrix(4);
} 

发表回复