[Codility] Leader - Dominator 풀이

예문

https://app.codility.com/programmers/lessons/8-leader/dominator/

An array A consisting of N integers is given. The dominator of array A is the value that occurs in more than half of the elements of A.

For example, consider array A such that

 A[0] = 3    A[1] = 4    A[2] =  3
 A[3] = 2    A[4] = 3    A[5] = -1
 A[6] = 3    A[7] = 3

The dominator of A is 3 because it occurs in 5 out of 8 elements of A (namely in those with indices 0, 2, 4, 6 and 7) and 5 is more than a half of 8.

Write a function

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

that, given an array A consisting of N integers, returns index of any element of array A in which the dominator of A occurs. The function should return −1 if array A does not have a dominator.

For example, given array A such that

 A[0] = 3    A[1] = 4    A[2] =  3
 A[3] = 2    A[4] = 3    A[5] = -1
 A[6] = 3    A[7] = 3

the function may return 0, 2, 4, 6 or 7, as explained above.

Write an efficient algorithm for the following assumptions:

N is an integer within the range [0..100,000];
each element of array A is an integer within the range [−2,147,483,648..2,147,483,647].

해석

  • 배열A : 정수 N으로 구성
  • 배열A의 지배자 : A의 요소의 절반 이상에서 발생하는 값.
  • 배열A의 지배자가 발생하는 요소의 인덱스 중 하나를 반환하라!
    • 만약 지배자가 없으면 -1을 반환

풀이

  • 배열의 절반 이상을 차지해야 하므로 리더는 한명만 존재!
  • 리더 선출에 stack 사용
  • stack에 값이 없으면 push
  • 이전 값이랑 현재 값이 다르면 이전값이랑 현재값 제거
  • 이전 값이랑 현재 값이 같으면 stack에 쌓음
  • 최종적으로 stack에 남아 있는 값이 리더
  • 중간에 stack에 남아있는 값이 배열요소의 절반이 남으면 리더
  • stack에 남아 있는 값과 동일한 값인지 비교해서 리더의 모든 인덱스 찾음

제약사항

  • N의 범위는 정수 [0..100,000]
  • 배열 A 의 요소의 범위는 [−2,147,483,648..2,147,483,647]

코드

import java.util.Stack;
public int solution(String S) {
  Stack<Character> stack = new Stack<>();
  char bracket;

  for (int i = 0; i < S.length(); i++) {
    bracket = S.charAt(i);

    if (stack.isEmpty()) {
      stack.push(bracket);
      continue;
    }

    if ('(' == stack.peek() && ')' == bracket) stack.pop();
    else stack.push(bracket);
  }
  return stack.size() > 0 ? 0 : 1;
}

결과