[코딩 테스트 연습(파이썬/Python)] 백준(BOJ) 9252번 _ LCS 2

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

9252번: LCS 2

[문제]

LCS(Longest Common Subsequence, 최장 공통 부분 수열)문제는 두 수열이 주어졌을 때, 모두의 부분 수열이 되는 수열 중 가장 긴 것을 찾는 문제이다.

예를 들어, ACAYKP와 CAPCAK의 LCS는 ACAK가 된다.

 

[입력]

첫째 줄과 둘째 줄에 두 문자열이 주어진다. 문자열은 알파벳 대문자로만 이루어져 있으며, 최대 1000글자로 이루어져 있다.

[출력]

첫째 줄에 입력으로 주어진 두 문자열의 LCS의 길이를, 둘째 줄에 LCS를 출력한다.

LCS가 여러 가지인 경우에는 아무거나 출력하고, LCS의 길이가 0인 경우에는 둘째 줄을 출력하지 않는다.

 

[아이디어]

1) dp에 현재까지의 최장 길이와 그 문자열을 보관한다.

2) dp[i][j]는 s1의 i번째 문자와 s2의 j번째 문자가 같을 때와 그렇지 않을 때로 나누어 생각할 수 있다.

s1=input()
s2=input()
n1=len(s1); n2=len(s2)
dp = [[[0,''] for _ in range (n2+1)] for _ in range(n1+1)]
for i in range(n1):
    for j in range(n2):
        if s1[i]==s2[j]:
            dp[i+1][j+1][0] = dp[i][j][0] + 1
            dp[i+1][j+1][1] = dp[i][j][1] + s1[i]
        elif dp[i+1][j][0] < dp[i][j+1][0]:
            dp[i+1][j+1] = dp[i][j+1]
        else:
            dp[i+1][j+1] = dp[i+1][j]
print(*dp[-1][-1], sep='\n')

문자열만을 dp에 보관하는 아래 풀이도 가능하다. (이게 좀 더 빠르다)

s1=input()
s2=input()
n1=len(s1); n2=len(s2)
dp = [['']*(n2+1) for _ in range(n1+1)]
for i in range(n1):
    for j in range(n2):
        if s1[i]==s2[j]:
            dp[i+1][j+1] = dp[i][j] + s1[i]
        elif len(dp[i+1][j]) < len(dp[i][j+1]):
            dp[i+1][j+1] = dp[i][j+1]
        else:
            dp[i+1][j+1] = dp[i+1][j]
print(len(dp[-1][-1]))
print(dp[-1][-1])