이번에 풀어본 문제는
백준 9252번 LCS 2 입니다.
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Stack;
public class Main {
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
String str1 = br.readLine();
String str2 = br.readLine();
int len1 = str1.length();
int len2 = str2.length();
char [] arr1 = new char[len1 + 1];
char [] arr2 = new char[len2 + 1];
for (int i = 0; i < len1; i++) arr1[i + 1] = str1.charAt(i);
for (int i = 0; i < len2; i++) arr2[i + 1] = str2.charAt(i);
int [][] dp = new int[len1 + 1][len2 + 1];
for (int i = 1; i <= len1; i++) {
for (int j = 1; j <= len2; j++) {
if (arr1[i] == arr2[j]) {
dp[i][j] = dp[i - 1][j - 1] + 1;
}
else {
dp[i][j] = Math.max(dp[i - 1][j], dp[i][j - 1]);
}
}
}
StringBuilder answer = new StringBuilder();
int lcs = dp[len1][len2];
answer.append(lcs).append("\n");
if (lcs > 0) {
int i = len1, j = len2;
Stack<Character> stk = new Stack<>();
while (i > 0 && j > 0) {
int cur = dp[i][j];
if (cur == dp[i - 1][j]) {
i--;
}
else if (cur == dp[i][j - 1]) {
j--;
}
else {
stk.add(arr1[i]);
i--;
j--;
}
}
while(!stk.isEmpty()) {
answer.append(stk.pop());
}
}
System.out.print(answer);
}
}
최장 공통 부분 수열을 구하고, 길이와 문자열을 출력하는 문제입니다.
길이를 구하는 방법은 LCS 문제와 동일하며,
해당 위치를 기준으로 인덱스를 이동하며 문자열을 완성시켜줍니다.
왼쪽이나 위쪽과 누적된 dp값이 동일할 경우, 해당 위치로 이동해주고, 값이 다르다면 [i-1][j-1]로 이동하고, 해당 인덱스의 문자를 스택에 담아 최종 출력값에 반영해줍니다.
마지막으로, 모든 추적이 끝나고 스택을 pop하면서 문자열을 완성해주면 해결할 수 있습니다.