day03 756 蛇形矩阵（偏移量技巧）

756 蛇形矩阵

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

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

输入格式

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

输出格式

输出满足要求的矩阵。

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

数据范围

$1\le n,m\le 100$

输入样例：

3 3

输出样例：

1 2 3
8 9 4
7 6 5

难度：简单
时/空限制：1s / 64MB
来源：语法题 , 微软面试题
算法标签 数组 模拟

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int n = scanner.nextInt();
        int m = scanner.nextInt();
        int[] dx = {-1,0,1,0};//定义辅助改变方向的数组，上、右、下、左
        int[] dy = {0,1,0,-1};
        int[][] arr = new int[n][m];
        boolean[][] isVisited = new boolean[n][m];//用于标记当前位置是否走过了
        int a = 0,b=0,d=1;//d = 1表示最开始是往右走的
        for(int i = 1;i <= n*m;i++){//共有n*m个数，故走n*m步
            arr[a][b] = i;
            isVisited[a][b] = true;
            int x = a + dx[d];//尝试更新下一步要走的坐标
            int y = b + dy[d];
            //下一步要走的坐标越界或者已经有值了，说明需要拐弯了
            if(x >= n || x < 0 || y >= m || y < 0 || isVisited[x][y]){
                d = (d + 1) % 4;
                x = a + dx[d];//更新正确的下一步要走的坐标
                y = b + dy[d];
            }
            a = x;//下一步要走的坐标赋给（a,b)
            b = y;
        }
        for(int i = 0;i < n;i++){
            for(int j = 0;j < m;j++){
                System.out.print(arr[i][j] + " ");
            }
            System.out.println();
        }
    }
}