2022년 04월 07일 목요일 - 오늘은 생일!!
오늘 올려볼 문제는 1046번 Last Stone Weight 이라는 문제이다.
사진을 클릭하면 해당 문제로 이동합니다.
오늘도 LeetCode 사이트 오늘의 문제를 가지고 왔다.
뭔가 깔삼하게 풀지는 모태서...
입력
사진에서도 볼 수 있듯이 int 배열 1개가 입력으로 들어온다.
풀이 및 코드
가장 큰 수 2개씩 없애서 마지막에 남는 값을 리턴하는 문제다.
수를 없애는 규칙은 두 수가 같을 때는 둘 다 없애고 아닐 때는 큰 수에서 작은 수를 뺀 값만 남긴다.
오늘은 처음부터 정답을 생각해냈다.
priority queue에 넣고 규칙대로 문제를 풀었다.
이제 코드를 봐보자!
풀이코드
class Solution {
public int lastStoneWeight(int[] stones) {
PriorityQueue<Integer> pq = new PriorityQueue<>(Collections.reverseOrder());
for(int num : stones) pq.add(num);
while(pq.size() > 1)
{
int n1 = pq.poll();
int n2 = pq.poll();
if(n1 == n2) continue;
pq.add(n1 - n2);
}
return pq.size() == 0 ? 0 : pq.poll();
}
}
제출 화면
오늘 문제는 nlogn으로 풀어서 살짝 멋이 없었다.
내일은 멋있게 풀어봐야겠다!
내일도 문제를 풀어서 블로그에 글을 쓸 수 있으면 좋겠다.
'LeetCode 문제 풀이' 카테고리의 다른 글
[LeetCode] 347번 문제를 풀어보았다. (ft. java) (0) | 2022.04.09 |
---|---|
[LeetCode] 703번 문제를 풀어보았다. (ft. java) (0) | 2022.04.08 |
[LeetCode] 923번 문제를 풀어보았다. (ft. java) (0) | 2022.04.06 |
[LeetCode] 1721번 문제를 풀어보았다. (ft. java) (0) | 2022.04.04 |
[LeetCode] 31번 문제를 풀어보았다. (ft. java) (0) | 2022.04.03 |