티스토리 뷰

N개의 숫자로 이루어진 배열 A를 받아서, P값 기준으로 나눠서 부분합의 차이가 가장 작은 값을 구하는 문제.

A[0], A[1], ... , A[P-1] / A[P+1], A[P+2], ... , A[N-1]

P를 N-1까지 증가시키면서 매번 위처럼 나뉜 두 그룹의 차이의 최소값을 return하면 된다.

class Solution {
    public int solution(int[] A) {
        int totalSum = 0;
        // 먼저 총합을 구하고
        for (int i = 0; i < A.length; i++) {
            totalSum+=A[i];
        }
        int result = Integer.MAX_VALUE;
        int sum = 0;
        for (int i = 0; i < A.length - 1; i++) {
            // i까지의 합
            sum+=A[i];
            // i까지의 합 - (총합 - i까지의 합)과 현재 result와 비교해서 작은 값을 result에
            result = Math.min(result, Math.abs(sum - (totalSum - sum)));
        }
        return result;
    }
}

미리 총합을 구해놓고 부분합과 계산해서 처리하는 식으로 이중 for문을 피하는 것이 포인트.

댓글
공지사항
최근에 올라온 글
최근에 달린 댓글
Total
Today
Yesterday
링크
«   2024/09   »
1 2 3 4 5 6 7
8 9 10 11 12 13 14
15 16 17 18 19 20 21
22 23 24 25 26 27 28
29 30
글 보관함