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

https://swexpertacademy.com/main/main.do

SW Expert Academy

 

SWEA 2805번

i가 0 ~ (N/2)까지인 경우랑 (N/2)+1 ~ (N-1)인 경우를 나눠서 코드 작성함

 

import java.util.Scanner;

public class P2805 {
    public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        int T = sc.nextInt();
        for(int t=1; t<=T; t++){
            int N = sc.nextInt();
            int[][] a = new int[N][N];
            for(int i=0; i<N; i++){
                String str = sc.next();
                char[] ch = str.toCharArray();
                for(int j=0; j<N; j++){
                    a[i][j] = Integer.parseInt(String.valueOf(ch[j]));
                }
            }
            int sum = 0;
            // 0~(N/2)까지
            for(int i=0; i<=N/2; i++){
                for(int j=N/2-i; j<=N/2+i; j++){
                    sum += a[i][j];
                }
            }
            // (N/2)+1~(N-1)까지
            int lt=0, rt=N-1;
            for(int i=(N/2)+1; i<N; i++){
                for(int j=lt+(i-N/2); j<=rt-(i-N/2); j++){
                    sum+=a[i][j];
                }
            }

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