[SW Expert Academy] 농작물 수확하기 (D3)

https://swexpertacademy.com/main/code/problem/problemDetail.do?problemLevel=3&contestProbId=AV7GLXqKAWYDFAXB&categoryId=AV7GLXqKAWYDFAXB&categoryType=CODE&problemTitle=&orderBy=PASS_RATE&selectCodeLang=ALL&select-1=3&pageSize=10&pageIndex=3

SW Expert Academy

 

 

import java.util.Scanner;

class Solution {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);

		int t = sc.nextInt();

		for (int tc = 1; tc <= t; tc++) {
			int n = sc.nextInt();

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

			int sum = 0;

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

			// 중간 냠
			for (int i = 0; i < n; i++) {
				sum += arr[n / 2][i];
			}

			// 밑 삼각형
			int idx = n / 2 + 1;
			int count = 1;

			for (int i = n / 2 - 1; i >= 0; i--) {

				for (int j = count; j < n - count; j++) {
					sum += arr[idx][j];
				}

				++idx;
				++count;

			}

			// 위 삼각형
			idx = n / 2 - 1;
			count = 1;

			for (int i = n / 2 - 1; i >= 0; i--) {

				for (int j = count; j < n - count; j++) {
					sum += arr[idx][j];
				}

				--idx;
				++count;

			}

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

		}

	}
}

 

나눠서 정복하기 기법을 사용했는데 다시 풀어보면 더 좋은 방법이 떠오를 수도 있을 것 같다.

 

 

start end 값 바꿔주는 코드

import java.util.Scanner;

class Solution {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);

		int t = sc.nextInt();

		for (int tc = 1; tc <= t; tc++) {
			int n = sc.nextInt();
			int map[][] = new int[n][n];

			for (int i = 0; i < n; i++) {
				String str = sc.next();
				for (int j = 0; j < str.length(); j++) {
					map[i][j] = str.charAt(j) - '0';
				}
			}

			int start = n / 2;
			int end = n / 2;
			int ans = 0;

			for (int i = 0; i < n; i++) {
				
				//System.out.println(start+" "+end);
				
				for (int j = start; j <= end; j++) {
					ans += map[i][j];
				}
				if (i < n / 2) {
					start--;
					end++;
				} else {
					start++;
					end--;
				}
			}
			
			System.out.println("#"+tc+" "+ans);

		}
	}
}