Jungol (Java) - 2046 : 숫자사각형4

www.jungol.co.kr/bbs/board.php?bo_table=pbank&wr_id=1316&sca=2010

JUNGOL

 

import java.util.*;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n, m;
        n = sc.nextInt();
        m = sc.nextInt();
        int arr[][] = new int[n][n];
        int num = 1;

        switch (m) {
            case 1:
                for (int i = 0; i < n; i++) {
                    for (int j = 0; j < n; j++) {
                        arr[j][i] = num;
                        num++;
                    }
                    num = 1;
                }
                break;
            case 2:
                for (int i = 0; i < n; i++) {
                    if (i % 2 == 0) {
                        num = 1;
                        for (int j = 0; j < n; j++) {
                            arr[i][j] = num;
                            num++;
                        }
                    } else {
                        num = 1;
                        for (int j = n - 1; j >= 0; j--) {
                            arr[i][j] = num;
                            num++;
                        }
                    }
                }
                break;
            case 3:
                for (int i = 0; i < n; i++) {
                    for (int j = 0; j < n; j++) {
                        arr[i][j] = (i + 1) * num;
                        num++;
                    }
                    num = 1;
                }
                break;
        }

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

    }
}