3-H0ngJu

[H0ngJu]

🔗 문제 링크

N번째 큰 수

✔️ 소요된 시간

30-40분

✨ 수도 코드

이 문제는 배열을 처음부터 끝까지 정렬해서 arr[line] 혹은 arr[len(arr)-line]으로 답은 구할 수 있지만, 메모리 초과로 통과는 안됩니다..!

그래서 heap을 통해 풀었습니다.

1. heap의 크기가 line 미만이면 push
2. heap의 크기가 line 이상이면, 최솟값과 비교해서 크면 pop하고 새 원소를 push

처음에는 아래 코드와 같이, 힙으로 풀었는데도 메모리 초과가 났습니다 ㅜ

import sys
import heapq

line = int(sys.stdin.readline().strip())

array = [list(map(int, sys.stdin.readline().strip().split())) for _ in range(line)]

heap = []
heapq.heapify(heap)

for i in range(len(array)):
    for k in range(len(array[i])):
        if len(heap) < line:
            heapq.heappush(heap, array[i][k])
        else:
            if array[i][k] > heap[0]:
                heapq.heappop(heap)
                heapq.heappush(heap, array[i][k])

print(heap[0])

그래서 모든 원소(line*line)를 저장하는 배열 array의 문제라고 보고, 배열의 저장은 한 줄씩만 하도록 for문 안으로 넣었더니 통과가 되었습니다 -!

📚 새롭게 알게된 내용

• heap이라는 자료구조

1. 큐 + 우선순위 = heap (이진 트리 구조)
2. 최대힙, 최소힙으로 나눌 수 있고, 최대힙은 최소힙에서 -를 붙여 구현
3. 부모 노드와 자식 노드의 관계
   a. 왼쪽 자식 = (부모 index) * 2
   b. 오른쪽 자식 = (부모 index) * 2 + 1
   c. 부모 index = (자식 index) / 2

[tgyuuAn]

저도 처음엔 메모리가 초과날 것은 알고 있었지만,

'혹시나 느슨한 테스트 케이스로 인해 통과될까?' 라는 실험 정신으로 완탐으로 제출해봤어요.

(만약 통과되면 문제가 원하는 바가 아닐 것 같다는 생각이 들어서 테스트 케이스 추가를 요청하려고..)

import sys

def input(): return sys.stdin.readline().rstrip()

N = int(input())

numbers = []
for _ in range(N):
    numbers.extend(map(int,input().split()))
    
print(sorted(numbers, reverse = True)[N-1])

역시나 메모리 초과가 나더군요.

225만개의 정수형 데이터를 모두 저장하기에는 12MB로는 터무니가 없기 떄문이죠...

int형이 대부분 4바이트인데 파이썬은 더 많은 바이트를 소모하니까.. 대충 계싼해보면 초과가 난다는 것을 알 수 있었어요.

그럼 도대체 어떻게 하면 좋을까 곰곰히 생각해봤는데,

결국에는 N번째 큰 수를 찾는 것이므로 N개의 데이터만 저장하면 될 것 같다는 생각이 들었어요.

최근에 홍주님이 제출하셨던 deque로 이용했던 버퍼(?) 혹은 슬라이딩 윈도우 방식을 도입하되,

어떤 자료구조가 가장 적절할까? 를 고민해보니까,

heap이 바로 떠올라서 heap으로 구현했더니 바로 통과되었씁니다!

import sys
from heapq import *

def input(): return sys.stdin.readline().rstrip()

N = int(input())

board = []
for _ in range(N):
    temp = list(map(lambda x : -1*int(x),input().split()))
    heapify(temp)
    
    # board가 비어있을 경우
    if not board: 
        board.extend(map(lambda x : -1*x, temp))
        heapify(board)
    
    else: 
        while -1*temp[0] > board[0]:
            heappop(board)
            heappush(board, -1*heappop(temp))
            
    # print(board)
            
print(board[0])

[tgyuuAn]

지금은 뭐 프로젝트도 아니지만 i라는 변수명하고 s라는 변수명은 지양하는 게 좋을 것 같아요!

코드 리뷰를 받는 시스템이다 보니까...

PS할 때 클린 코드는 오버 엔지니어링이긴 하지만,

요즘은 코드가 길어지는 구현 문제도 많이 나오기 때문에 너무 많은 시간을 들이지는 말고 최소한의 노력(?) 정도는 해주면 좋다고 생각합니다!

그리고 지금은 시간 제한이 있는 시험도 아니기에 이런 클린 코드를 체화 시키기에는 좋을 것 같다고 생각이 되어서요!

[H0ngJu]

오오 좋은 리뷰 감사합니다 (_ _)

아마 저 당시에는 i는 단순 반복문을 위한 변수, s는 단순히 원소를 data에서 꺼내온다고 생각해서 썼던 것 같네용

피드백을 반영하면, for _ in range(line):과 for num in data:로 각각 i와 s를 수정할 수 있겠네요!

[pknujsp]

이 함수는 여기에서 호출안하셔도 돼요
heap 리스트 원소 트리를 재 구성하는거라 비어있을 때는 불필요한 호출이랍니다

[pknujsp]

한달 전에 메모리 초과때문에 푸는 데 시간이 좀 걸렸던 문제네요.

처음은 밑에 처럼 리스트에 원소를 전부다 넣고
heapify() 로 최대 힙을 만든 후
N - 1번 최댓값을 뺸 뒤에, 그 다음 최댓값을 출력하도록 했었습니다.

이렇게 하니까 메모리가 터지더라구요.
이 부분은 홍주님이 써놓으신 내용이군요

for _ in range(N):
    heap.extend(map(lambda v: -int(v), stdin.readline().split()))
heapify(heap)

for _ in range(N - 1):
    heappop(heap)

print(-heappop(heap))

단순 정렬을 하는 것 뿐인데 다시 보니 정말 이상한 코드네요
똑같이 메모리 문제가 있지만
리스트 정렬하고 N번째 인덱스 값을 보여주게 하는게 맞는 방식인데.

어떻게 고쳐야하나 고민해보니

전부다 리스트에 넣고 최대 힙으로 만들 필요없이
최소 힙을 사용하고 크기를 N으로 유지시키면 되더라구요.

예를 들어 [7, 4, 3, 9, 1, 2]에서 3번째 큰 수를 찾는다는 경우를 봤을때
최소 힙 리스트 X를 만들고, 원소들을 순서대로 순회하면서
경우에 따라 추가해가면 3번째 큰 수가 힙의 최솟값이 된다는 걸 알게됬습니다.

• X : [7, 4, 3] : X크기가 3보다 작으니, 7, 4, 3을 순서대로 추가
• [7, 4, 9] : 3을 빼고, 9를 추가, 1, 2는 3번째 큰 수가 될 수 없으니 추가X
• [7, 4, 9] : 최솟값인 4가 3번째 큰 수

좀 길어지긴 했는데
우선순위 큐를 올바르게 사용하는 법을 좀 더 배울수 있었던 문제입니다

제 코드가 홍주님 코드랑 거의 똑같네요

N = int(stdin.readline())
heap = []

for _ in range(N):
    for v in map(int, stdin.readline().split()):
        if len(heap) < N:
            heappush(heap, v)
        elif v >= heap[0]:
            heappush(heap, v)
            heappop(heap)

print(heap[0])

[Munbin-Lee]

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);

    int N;
    cin >> N;

    vector<int> v(N * N);

    for (int &num: v) {
        cin >> num;
    }

    sort(v.rbegin(), v.rend());

    cout << v[N - 1];

    return 0;
}

띠요옹 C++는 되네요

[Munbin-Lee]

#include <iostream>
#include <queue>

using namespace std;

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);

    int N;
    cin >> N;

    priority_queue<int, vector<int>, greater<>> pq;

    for (int i = 0; i < N; i++) {
        int num;
        cin >> num;
        pq.emplace(num);
    }

    for (int i = 0; i < N * (N - 1); i++) {
        int num;
        cin >> num;
        pq.emplace(num);
        pq.pop();
    }

    cout << pq.top();

    return 0;
}

자세한 내용은 리뷰로 달아놓겠습니다.

[Munbin-Lee]

결국 힙 자체가 크기 비교를 위한 자료구조이기 때문에, 조건문을 달아놓을 필요 없이 바로 push 후 pop하면 됩니다!

heapq.heappush(heap, s)
heapq.heappop(heap)

heappushpop()을 사용하면 한 줄로도 되구요.