[백준] 5619 파이썬(python) : 세 번째 - 메모리 줄이는 법

https://www.acmicpc.net/problem/5619

5619번: 세 번째

 

 

일본 대회가 출처로 적혀있다. itertools에서 permutations 사용했는데 테스트 케이스는 맞고 채점에서는 메모리 초과가 나와서 틀렸다. 순열은 모든 경우를 다 따져주니까 그런것 같다.

 

메모리 초과 나온 코드)

import sys, itertools
n = int(sys.stdin.readline())
arr = [ int(sys.stdin.readline()) for _ in range(n) ]
l, ans = [], []
for a, b in itertools.permutations(arr, 2):
    x = str(a) + str(b)
    ans.append(int(x))
ans.sort()
print(ans[2])

 

import sys
n = int(sys.stdin.readline())
arr = [ int(sys.stdin.readline()) for _ in range(n) ]
ans = []
arr.sort() #1
for a in arr[:4]: #2
    for b in arr[:4]:
        if a != b: #3
            x = str(a) + str(b)
            ans.append(int(x)) #4
ans.sort() #5
print(ans[2]) #6

 

#1 : 입력받은 값을 오름차순으로 정렬

#2 : 정렬된 상태에서 4개까지 자른다. 그것을 이중for문으로 만든다

#3 : 둘이 같지 않으면

#4 : 문자열로 만들고 합쳐서 정수형으로 바꾼뒤 삽입

#5 : 정렬해준뒤

#6 : 세 번째 숫자를 출력한다

 

permutations로 해결하지 못해서 결국 다른분의 블로그를 봤는데 더 좋은 방법이 생각이 나지 않았다. 생각해내지 못했다는것에 기분이 나쁘지만 뻐대지 말고 그냥 받아드리는게 좋을것 같다.