[Codility/Java] 문제 6. TapeEquilibrium

TapeEquilibrium

 

Task

A non-empty array A consisting of N integers is given. Array A represents numbers on a tape.

Any integer P, such that 0 < P < N, splits this tape into two non-empty parts: A[0], A[1], ..., A[P − 1] and A[P], A[P + 1], ..., A[N − 1].

The difference between the two parts is the value of: |(A[0] + A[1] + ... + A[P − 1]) − (A[P] + A[P + 1] + ... + A[N − 1])|

In other words, it is the absolute difference between the sum of the first part and the sum of the second part.

For example, consider array A such that:

A[0] = 3 A[1] = 1 A[2] = 2 A[3] = 4 A[4] = 3

We can split this tape in four places:

• P = 1, difference = |3 − 10| = 7
• P = 2, difference = |4 − 9| = 5
• P = 3, difference = |6 − 7| = 1
• P = 4, difference = |10 − 3| = 7

Write a function:

class Solution { public int solution(int[] A); }

that, given a non-empty array A of N integers, returns the minimal difference that can be achieved.

For example, given:

A[0] = 3 A[1] = 1 A[2] = 2 A[3] = 4 A[4] = 3

the function should return 1, as explained above.

Write an efficient algorithm for the following assumptions:

• N is an integer within the range [2..100,000];
• each element of array A is an integer within the range [−1,000..1,000].

 

Solution

class Solution {
 public int solution(int[] A) {
	 int result = Integer.MAX_VALUE;
	 int total = 0;
	 int left = 0;
	 int right = 0;
	 
	 for(int i=0; i<A.length; i++) {
		 total += A[i];
	 }
	 
	 for(int j=1; j<A.length; j++) {
		 left += A[j-1];
		 right = total - left;
		 result = Math.min(Math.abs(left - right), result);
	 }
	 return result;
 }
}

 

Notion

처음에 생각한 방법은 이중 for문을 사용하는 것이었다. 근데 반복문을 이중으로 사용하면 시간복잡도가 너무 높아지므로 다른 방법을 생각해봤다.

left part와 right part를 나누어서 계산해야겠다는 것은 생각했는데 어떤 방법으로 구현할 지 정하지 못해 다른 사람들의 코드를 참고했다.

Math.min으로 최솟값을 구하려면 result의 초기값은 Integer.MAX_VALUE(Integer로 표현할 수 있는 최댓값) 이어야 함을 잊지말자!