[백준 11054] 가장 긴 바이토닉 부분 수열 ❗

코뉴·2022년 1월 20일
0

백준🍳

목록 보기
70/149
post-custom-banner

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

🥚문제


🥚입력/출력


🍳코드

import sys
input = sys.stdin.readline

n = int(input().strip())
a = list(map(int, input().split()))


def LIS(part_of_a, k):
    # dp[x] = 길이가 x를 만족하는 수열들 중, 마지막 숫자의 최소값
    dp = [0]

    for i in range(len(part_of_a)):
        # k (Sk를 나타냄)보다 크거나 같으면, 증가부에 들어갈 수 없음
        if part_of_a[i] >= k:
            continue

        if dp[-1] < part_of_a[i]:
            dp.append(part_of_a[i])

        else:
            # 이미 존재하는 값이면 스킵
            if part_of_a[i] in dp:
                continue
            # 들어갈 수 있는 자리 찾기
            for j in range(len(dp)):
                if dp[j] > part_of_a[i]:
                    dp[j] = part_of_a[i]
                    break
    return len(dp) - 1


# 수열 a의 각 숫자가 각각 Sk일 때, 그 중 가장 긴 것 찾기
max_len = 0
for k in range(n):

    # Sk를 포함하지 않은 좌측 (증가부)
    left = a[:k]
    # Sk를 포함하지 않은 우측 (감소부)
    right = a[k+1:]

    # 증가부의 길이
    left_len = LIS(left, a[k])

    # 감소부의 길이
    # right를 reverse해줘야 함!
    right.reverse()
    right_len = LIS(right, a[k])

    # Sk는 무조건 수열에 들어가 있어야 하므로,
    # 증가부 길이 + 감소부 길이 + 1 (Sk)
    total_len = left_len + right_len + 1
    max_len = max(max_len, total_len)

print(max_len)

🧂아이디어

최장 증가 부분 수열 (LIS) 활용

  • 예전에 풀었던 가장 긴 증가하는 부분 수열 (코드 보기) 에서 사용한 알고리즘을 활용

  • n개의 숫자 -> 각각 Sk일 때의 가장 긴 바이토닉 부분 수열을 구한다

  • Sk를 기준으로

    • 좌측(증가부)는 LIS 알고리즘 적용
    • 우측(감소부)는 reverse한 뒤, LIS 알고리즘 적용(⭐⭐⭐⭐⭐)
  • LIS 알고리즘 (O(NlogN))

    • dp[k] = 수열의 길이가 k인 것 중, 마지막 숫자가 될 수 있는 것의 최소값을 저장
      • 그래야 그 뒤에 붙을 수 있는 수의 경우가 더 많아지므로!
    • dp[0] = 0으로 초기화
    • 수열의 길이는 len(dp) - 1로 얻어냄 (dp[0] 제외)

🍯비슷한 문제

profile
코뉴의 도딩기록
post-custom-banner

0개의 댓글