[SWEA][JAVA]D3 : 농작물 수확하기

https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AV7GLXqKAWYDFAXB 

SW Expert Academy

 

---

 

아직 그림문제에 익숙하지 않은데, 

고민하다가 마름모를 나눠서 구하면 된다.

 

가운데 가로줄을 포함하여 상단, 그리고 하단의 삼각형 부분.

 

 

import java.util.Scanner;
class Solution{
    public static void main(String args[]) throws Exception{
        
         Scanner kb = new Scanner(System.in);
        int t = kb.nextInt();
        for (int test_case = 1; test_case <= t; test_case++) {
            int n=kb.nextInt();
            int[][] arr=new int[n][n];
            for(int i=0; i<n; i++){
                String str=kb.next();
                for(int j=0; j<n; j++){
                    arr[i][j]=str.charAt(j)-'0';
                }
            }
            int sum=0;
            // 중간부 ~ 상단 
            for(int i=0; i<=n/2; i++){
                for(int j=n/2-i; j<=n/2+i; j++){
                    sum+=arr[i][j];
                }
            }
            // 중간부제외 ~ 하단 삼각형
            int a=1; // a는 행의 시작점 ( 점점 ++ )
            for(int i=n/2+1; i<n; i++){
                for(int j=a; j<=n-1-a; j++){ // j는 행의 끝점 ( 점점 -- )
                    sum+=arr[i][j];
                }
                a++;
            }
            System.out.println("#" + test_case +" "+sum);
        }
    }
}