[JAVA] CodeUp 1462 : [기초-배열연습] 2차원 배열 순서대로 채우기 1-3

https://codeup.kr/problem.php?id=1462 

[기초-배열연습] 2차원 배열 순서대로 채우기 1-3

문제 분류 : 기초5-3. 2차원 배열

문제 설명

다음과 같은 n*n 배열 구조를 출력해보자.

입력이 3인 경우 다음과 같이 출력한다.
1 4 7
2 5 8
3 6 9

입력이 5인 경우는 다음과 같이 출력한다.
1 6 11 16 21
2 7 12 17 22
3 8 13 18 23
4 9 14 19 24
5 10 15 20 25

입력이 n인 경우의 2차원 배열을 출력해보자.

입력

첫 번째 줄에 배열의 크기 n이 입력된다.
[입력값의 정의역]
1<= n <= 100

출력

n*n 크기의 배열을 설명과 같이 채워 출력한다.

입력 예시

4

출력 예시

1 5 9 13

2 6 10 14

3 7 11 15

4 8 12 16

도움말

 

 

내 답안

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;

public class Main {

  public static void main(String[] args) throws IOException {
    BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

    int n = Integer.parseInt(br.readLine());

    int [][] arr = new int [n][n];

    for(int i = 0; i < n; i++) {
      for(int j = 0; j < n; j++) {
        arr[i][j] = i+1 + n*(j);
        bw.write(String.valueOf(arr[i][j]+" "));
      }
      bw.write("\n");
    }

    bw.flush();
    bw.close();
    br.close();
  }
}