【蓝桥杯BASIC-25】回形取数 Java版

基础练习 回形取数

时间限制：1.0s 内存限制：512.0MB

问题描述
　　回形取数就是沿矩阵的边取数，若当前方向上无数可取或已经取过，则左转90度。一开始位于矩阵左上角，方向向下。
输入格式
　　输入第一行是两个不超过200的正整数m, n，表示矩阵的行和列。接下来m行每行n个整数，表示这个矩阵。
输出格式
　　输出只有一行，共mn个数，为输入矩阵回形取数得到的结果。数之间用一个空格分隔，行末不要有多余的空格。
样例输入
3 3
1 2 3
4 5 6
7 8 9
样例输出
1 4 7 8 9 6 3 2 5
样例输入
3 2
1 2
3 4
5 6
样例输出
1 3 5 6 4 2
分析
用四个循环表示在各个方向的前进，将取过的数标记为-1。用最外层的循环控制程序在取过所有的数后结束。

Java 代码：

import java.io.*;
public class Main{
    public static void main(String[] args)throws Exception{
        BufferedReader in=new BufferedReader(new InputStreamReader(System.in));
        String[] s1=in.readLine().split(" ");
        int n=Integer.parseInt(s1[0]);
        int m=Integer.parseInt(s1[1]);
        int[][] nm=new int[n][m];
        for(int i=0;i<n;i++){
            String[] s2=in.readLine().split(" ");
            for(int j=0;j<m;j++){
                nm[i][j]=Integer.parseInt(s2[j]);
            }
        }
        int i=-1,j=0;
        int num=n*m;
        while(num>0){
            while(i+1<n && nm[i+1][j]!=-1){
                System.out.print(" "+nm[i+1][j]);
                nm[i+1][j]=-1;
                i++;
                num--;
            }
            while(j+1<m && nm[i][j+1]!=-1){
                System.out.print(" "+nm[i][j+1]);
                nm[i][j+1]=-1;
                j++;
                num--;
            }
            while(i-1>=0 && nm[i-1][j]!=-1){
                System.out.print(" "+nm[i-1][j]);
                nm[i-1][j]=-1;
                i--;
                num--;
            }
            while(j-1>=0 && nm[i][j-1]!=-1){
                System.out.print(" "+nm[i][j-1]);
                nm[i][j-1]=-1;
                j--;
                num--;
            }
        }

    }
}