[SWEA] 2805. 농작물 수확하기

N x N 배열에서 다이아몬드 모양의 값들만 더한다.

---

import java.util.Scanner;

public class 농장 {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);

        int T = scanner.nextInt();

        for(int tc = 1; tc <= T; tc++) {
            int n = scanner.nextInt();
            int[][] arr = new int[n][n];
            int sum = 0;

            for(int i = 0; i < n; i++) {
                String str = scanner.next();

                for(int j = 0; j < str.length(); j++) {
                    arr[i][j] = str.charAt(j) - '0';
                }
            }

            for(int i = 0; i < n; i++) {
                if(i <= n / 2) {
                    for(int j = n / 2 - i; j <= n / 2 + i; j++) {
                        sum += arr[i][j];
                    }
                } else {
                    for(int j = n / 2 - (n - i - 1) ; j <= n / 2 + (n - i - 1) ; j++) {
                        sum += arr[i][j];
                    }
                }
            }

            System.out.println("#" + tc + " " + sum);
        }

        scanner.close();
    }
}

N의 크기에 따라 다이아몬드 모양이 어떻게 나오는지 i, j값을 따져보면 된다.