基础练习 回形取数

问题描述

　　回形取数就是沿矩阵的边取数，若当前方向上无数可取或已经取过，则左转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

import java.util.Scanner;  
      
    public class Main{   
        public static void main(String[] args) {
            int n,m;
            Scanner input = new Scanner(System.in);
            n = input.nextInt();
            m = input.nextInt();
            int[][] a= new int[n][m];
            for(int i=0;i<n;i++){
                for(int j=0;j<m;j++){
                    a[i][j] = input.nextInt();
                }
            }
            int i=0,j=0;
            int n1 = n;
            int m1 = m;
            int h = 0;
            int k = 1;
            if(n!=1){
                i = 1;
                System.out.print(a[0][0]);
            }
            else{
                System.out.print(a[0][0]);
                j = 1;
            }
                
            while(true){
                while(i<n1-1){
                    System.out.print(" "+a[i][j]);
                    k++;
                    i++;
                }
                if(k==n*m-1||k==n*m){
                    break;
                }
                while(j<m1-1){
                    System.out.print(" "+a[i][j]);
                    k++;
                    j++;
                }
                if(k==n*m-1||k==n*m){
                    break;
                }
                while(i>h){
                    System.out.print(" "+a[i][j]);
                    k++;
                    i--;
                }
                if(k==n*m-1||k==n*m){
                    break;
                }
                while(j>h+1){
                    System.out.print(" "+a[i][j]);
                    k++;
                    j--;
                }
                if(k==n*m-1||k==n*m){
                    break;
                }
                h++;
                n1--;
                m1--;
            }
            if(k==n*m-1)
                System.out.print(" "+a[i][j]);
        }
    
 }