蛇形矩阵

蛇形矩阵

输入两个整数n和m，输出一个n行m列的矩阵，将数字 1 到 n*m 按照回字蛇形填充至矩阵中。

具体矩阵形式可参考样例。

输入格式
输入共一行，包含两个整数n和m。

输出格式
输出满足要求的矩阵。

矩阵占n行，每行包含m个空格隔开的整数。

数据范围

1≤n,m≤100

输入样例：

3 3

输出样例：

1 2 3
8 9 4
7 6 5

题目分析

将每个方向的偏移量放在dx和dy数组内

C++代码

#include<iostream> 

using namespace std;

const int N = 110;

int n,m;
int q[N][N];

int main()
{
	cin >> n >> m;
	int dx[] = {-1,0,1,0},dy[] = {0,1,0,-1};
	int x = 0, y = 0 , d = 1;
	for(int i = 1;i <= n*m;i++){
		q[x][y] = i;
		int a = x + dx[d], b = y + dy[d];
		if( a< 0 ||	a >= n || b < 0 || b >= m || q[a][b]){
			d = (d + 1)%4;
			a = x + dx[d],b = y + dy[d];
		}
		x = a; y = b;
	}
	for(int i = 0;i < n;i++){
		for(int j = 0; j < m;j++){
			cout << q[i][j] << ' '; 	
		}
		cout << endl;
	}	
} 

Java代码

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        int N = 110;
        int n,m;
        int[][] p = new int[N][N];
        int[] dx = {-1,0,1,0};
        int[] dy = {0,1,0,-1};
        Scanner sc = new Scanner(System.in);
        n = sc.nextInt();
        m = sc.nextInt();
        int x = 0, y = 0, d = 1;
        for (int i = 1;i<=n*m;i++){
            p[x][y] = i;
            int a = x + dx[d];
            int b = y + dy[d];
            if(a<0 || a>=n || b < 0 || b >=m || p[a][b] >= 1){
                d = (d + 1) %4;
                a = x + dx[d];
                b = y + dy[d];
            }
            x = a;
            y = b;
        }

        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                System.out.print(p[i][j] + " ");
            }
            System.out.println();
        }
    }
}