[백준/BOJ] platinum5 - 14003번 가장 긴 증가하는 부분 수열5 (Python)

▶14003 - 가장 긴 증가하는 부분 수열

▶문제

수열 A가 주어졌을 때, 가장 긴 증가하는 부분 수열을 구하는 프로그램을 작성하시오.

예를 들어, 수열 A = {10, 20, 10, 30, 20, 50} 인 경우에 가장 긴 증가하는 부분 수열은 A = {10, 20, 10, 30, 20, 50} 이고, 길이는 4이다.

 

▶입력

첫째 줄에 수열 A의 크기 N (1 ≤ N ≤ 1,000,000)이 주어진다.

둘째 줄에는 수열 A를 이루고 있는 Ai가 주어진다. (-1,000,000,000 ≤ Ai ≤ 1,000,000,000)

 

 

▶출력

첫째 줄에 수열 A의 가장 긴 증가하는 부분 수열의 길이를 출력한다.

둘째 줄에는 정답이 될 수 있는 가장 긴 증가하는 부분 수열을 출력한다.

 

▶풀이

두번정도 틀렸다.

런타임에러도 아니고, 시간초과도 아니고 그냥 틀렸습니다...

 

from bisect import bisect_left

num = int(input())
a = list(map(int, input().split()))
stack = [0]

for i in a:
    if stack[-1] < i:
        stack.append(i)
    else:
        stack[bisect_left(stack, i)] = i

print(len(stack) - 1)
for i in range(1, len(stack)):
    print(stack[i], end=' ')

 

python에는 bisect라는 함수가 있어서 쉽게 문제를 풀 수 있을 거 같았다.

이미 그것을 이용해서 풀었지만...

뭔가 실수가 있는거 같아서 재도전!!

 

재도전해서 성공!!

​

사실 bisect를 이용하는 법을 모르고 있었고 이 문제를 풀면서 알게 되었다.

그냥 구현하면  시간복잡도가 너무 커지기때문에 bisect를 이용해야한다.

 

from bisect import bisect_left

num = int(input())
a = list(map(int, input().split()))
stack = []
tmp = []
for i in a:
    if not stack or stack[-1] < i:
        stack.append(i)
        tmp.append((len(stack)-1, i))
    else:
        stack[bisect_left(stack, i)] = i
        tmp.append((bisect_left(stack, i), i))

ans = []
last_idx = len(stack)-1
for i in range(len(tmp)-1, -1, -1):
    if tmp[i][0] == last_idx:
        ans.append(tmp[i][1])
        last_idx -= 1

print(len(stack))
print(*ans[::-1])