wojiushimogui的博客

正在路上的编程学习者

《leetCode》:Spiral Matrix II

题目

Given an integer n, generate a square matrix filled with elements from 1 to n2 in spiral order.

For example,
Given n = 3,

You should return the following matrix:
[
 [ 1, 2, 3 ],
 [ 8, 9, 4 ],
 [ 7, 6, 5 ]
]

题目大意:用1~n^2的数字按顺时针方向来填充一个n*n的矩阵。

思路

有了上篇博文的思路,这个题就比较简单了。
都是控制好方向和各个方向上的长度,进行赋值即可实现。

实现代码如下:

/**
 * Return an array of arrays.
 * Note: The returned array must be malloced, assume caller calls free().
 */
//此题比较简单;2015年11月21日20:00:12 
int** generateMatrix(int n) {
    if(n<0){
        return NULL;
    }
    int row=n;
    int col=n;

    //分配一个空间来保存结果
    int **result=(int **)malloc(row*sizeof(int *));
    if(result==NULL){
        exit(EXIT_FAILURE);
    } 
    for(int j=0;j<row;j++){
        result[j]=(int *)malloc(col*sizeof(int));
        if(result[j]==NULL){
            exit(EXIT_FAILURE);
        }
    }
    //nx数组和ny数组用来控制顺时针打印时,下标的变化
    //例如:向右访问时,matrix[y][x]中x每次添加1个单位,y不变;因此nx数组中第一个元素为1,ny数组中第一个元素为0;
    int nx[4]={1,0,-1,0};
    int ny[4]={0,1,0,-1};
    int i=0;//用来控制方向,i=0是表示向右,i=1时表示向下,i=2时表示向左,i=3时表示向上
    int x=-1;
    int y=0;
    int value=1;
    while(row>0&&col>0){
        int k;
        if(i%2==0){
            k=col;
            row--;
        }
        else{
            k=row;
            col--;
        }
        while(k>0){
            x+=nx[i];
            y+=ny[i];
            result[y][x]=value;
            value++;
            k--;
        }
        i=(i+1)%4;
    }
    return result; 
}

测试代码如下:

int main(void){
    int n;
    while(scanf("%d",&n)!=EOF&&n>0){
        int **result=generateMatrix(n);
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                printf("%d  ",result[i][j]);
            }
            printf("\n"); 
        }
    }
}
阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/u010412719/article/details/49967425
文章标签: square leetcode matrix
个人分类: leetCode
上一篇《leetCode》:Spiral Matrix
下一篇《leetCode》:Merge Intervals
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭