Algorithm/Codility
[코딜리티/Java] TapeEquilibrium
wavid
2020. 12. 26. 23:20

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문을 피하는 것이 포인트.