본문 바로가기
알고리즘/프로그래머스

[프로그래머스] 풍선 터트리기 JAVA

by Bhinney 2023. 11. 7.
 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr


❓ 문제

일렬로 나열된 n개의 풍선이 있습니다. 모든 풍선에는 서로 다른 숫자가 써져 있습니다. 당신은 다음 과정을 반복하면서 풍선들을 단 1개만 남을 때까지 계속 터트리려고 합니다.

  1. 임의의 인접한 두 풍선을 고른 뒤, 두 풍선 중 하나를 터트립니다.
  2. 터진 풍선으로 인해 풍선들 사이에 빈 공간이 생겼다면, 빈 공간이 없도록 풍선들을 중앙으로 밀착시킵니다.

여기서 조건이 있습니다. 인접한 두 풍선 중에서 번호가 더 작은 풍선을 터트리는 행위는 최대 1번만 할 수 있습니다. 즉, 어떤 시점에서 인접한 두 풍선 중 번호가 더 작은 풍선을 터트렸다면, 그 이후에는 인접한 두 풍선을 고른 뒤 번호가 더 큰 풍선만을 터트릴 수 있습니다.

당신은 어떤 풍선이 최후까지 남을 수 있는지 알아보고 싶습니다. 위에 서술된 조건대로 풍선을 터트리다 보면, 어떤 풍선은 최후까지 남을 수도 있지만, 어떤 풍선은 무슨 수를 쓰더라도 마지막까지 남기는 것이 불가능할 수도 있습니다.

일렬로 나열된 풍선들의 번호가 담긴 배열 a가 주어집니다. 위에 서술된 규칙대로 풍선들을 1개만 남을 때까지 터트렸을 때 최후까지 남기는 것이 가능한 풍선들의 개수를 return 하도록 solution 함수를 완성해주세요.



제한사항
  • a의 길이는 1 이상 1,000,000 이하입니다.
    • a[i]는 i+1 번째 풍선에 써진 숫자를 의미합니다.
    • a의 모든 수는 -1,000,000,000 이상 1,000,000,000 이하인 정수입니다.
    • a의 모든 수는 서로 다릅니다.

📎 예시

a 최후까지 남을 수 있는 풍선 return
[9, -1, -5] [9, -1, -5] 3
[-16, 27, 65, -2, 58, -92, -71, -68, -61, -33] [-16, -92, -71, -68, 61, -33] 6

✍🏻 풀이

  • 현재 위치한 풍선이 남을 수 있는지를 확인을 해주는 것이 문제를 푸는 방법이라고 생각하였다.
  • 현재 위치한 풍선의 좌측과 우측의 풍선들을 터트린 후, 현재 위치한 풍선이 남을 수 있는지 확인해 주는 방법으로 풀기로 했다.
  • 현재 풍선의 좌측의 최솟값과 우측의 최솟값을 알아 낸 후, 현재의 풍선과 비교해주었다.
  • 만약, 현재의 풍선이 두 값보다 모두 크면 해당 풍선은 터질 수 밖에 없다.
  • 더 큰 값은 단 한 번만 터트릴 수 있기 때문이다.
  • 즉, 현재의 풍선이 두 값 중에서 하나의 값보다 작으면 해당 풍선은 남을 수 있다.
  • 맨 앞의 풍선과 맨 뒤의 풍선은 좌측과 우측 값이 각각 없기 때문에, 무조건 남을 수 밖에 없다.

  • 배열 a의 길이가 1이면, 가능한 개수는 1개이므로, a의 길이를 리턴시켜주었다.
  • 좌측의 최솟값을 저장할 배열과 우측의 최솟값을 저장할 배열을 생성해주었다.
public int solution(int[] a) {
   if (a.length <= 1) return a.length;
   
   int[] left = new int[a.length]; // 해당 인덱스 원소 좌측에 최솟값 저장
   int[] right = new int[a.length]; // 해당 인덱스 원소 우측에 최솟값 저장
}

  • 반복문을 통해 좌측의 최솟값과 우측의 최솟값을 찾아 주었다.
public int solution(int[] a) {
   
   ..
   
   int l = a[0]; // 좌측 최솟값 비교
   for (int i = 1; i < a.length; i++) {
      left[i] = l;
      l = Math.min(l, a[i]);
   }
   
   int r = a[a.length - 1]; // 우측 최솟값 비교
   for (int i = a.length - 1; i >= 0; i--) {
      right[i] = r;
      r = Math.min(r, a[i]);
   }
}

  • 맨 처음과 맨 마지막의 풍선은 무조건 남기 때문에 ans는 2로 초기화 시켜주었다.
  • 반복문을 돌며, 좌측 최솟값과 우측 최솟값을 비교해주었다.
  • 만약, 두 개의 값보다 현재 풍선의 숫자가 더 크면 해당 풍선은 남을 수 없으므로 카운트 해주지 않았다.
public int solution(int[] a) {
   
   ...
   
   int ans = 2; // 맨 처음과 맨 끝 두 개는 무조건 가능함
   for (int i = 1; i < a.length - 1; i++) {
      if (a[i] > left[i] && a[i] > right[i]) continue;
      ans++;
   }
   
   return ans;
}​

✍🏻 최종 풀이

public int solution(int[] a) {
   if (a.length <= 1) return a.length;
   
   int[] left = new int[a.length]; // 해당 인덱스 원소 좌측에 최솟값 저장
   int[] right = new int[a.length]; // 해당 인덱스 원소 우측에 최솟값 저장
   
   int l = a[0]; // 좌측 최솟값 비교
   for (int i = 1; i < a.length; i++) {
      left[i] = l;
      l = Math.min(l, a[i]);
   }
   
   int r = a[a.length - 1]; // 우측 최솟값 비교
   for (int i = a.length - 1; i >= 0; i--) {
      right[i] = r;
      r = Math.min(r, a[i]);
   }
   
   int ans = 2; // 맨 처음과 맨 끝 두 개는 무조건 가능함
   for (int i = 1; i < a.length - 1; i++) {
      if (a[i] > left[i] && a[i] > right[i]) continue;
      ans++;
   }
   
   return ans;
}

✍🏻 다른 사람의 풀이

  • 좌측과 우측의 최솟값들을 Set에 저장해주면, 남을 수 있는 풍선만 저장이 된다.
import java.util.HashSet;

public int solution(int[] a) {
   int min1 = Integer.MAX_VALUE;
   int min2 = Integer.MAX_VALUE;
   HashSet<Integer> set = new HashSet<>();
   
   for (int i = 0; i < a.length; i++) {
      min1 = Math.min(min1, a[i]);
      min2 = Math.min(min2, a[a.length - 1 - i]);
      set.add(min1);
      set.add(min2);
   }
   
   return set.size();
}

 

댓글