백준 9252 LCS 2 (Java,자바)

jonghyukLee·2022년 11월 12일
0

이번에 풀어본 문제는
백준 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하면서 문자열을 완성해주면 해결할 수 있습니다.

profile
머무르지 않기!

0개의 댓글