ALGORITHM NOTE1

BOJ 10942 - 팰린드롬?

뒤집어도 같은지 한번 보자

#algorithm#boj#gold#dp
아카이브로 돌아가기

문제 링크

문제

명우는 홍준이와 함께 팰린드롬 놀이를 해보려고 한다.

먼저, 홍준이는 자연수 N개를 칠판에 적는다. 그 다음, 명우에게 질문을 총 M번 한다.

각 질문은 두 정수 S와 E(1 ≤ S ≤ E ≤ N)로 나타낼 수 있으며, S번째 수부터 E번째 까지 수가 팰린드롬을 이루는지를 물어보며, 명우는 각 질문에 대해 팰린드롬이다 또는 아니다를 말해야 한다.

예를 들어, 홍준이가 칠판에 적은 수가 1, 2, 1, 3, 1, 2, 1라고 하자.

  • S = 1, E = 3인 경우 1, 2, 1은 팰린드롬이다.
  • S = 2, E = 5인 경우 2, 1, 3, 1은 팰린드롬이 아니다.
  • S = 3, E = 3인 경우 1은 팰린드롬이다.
  • S = 5, E = 7인 경우 1, 2, 1은 팰린드롬이다.

자연수 N개와 질문 M개가 모두 주어졌을 때, 명우의 대답을 구하는 프로그램을 작성하시오.

입력

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

둘째 줄에는 홍준이가 칠판에 적은 수 N개가 순서대로 주어진다. 칠판에 적은 수는 100,000보다 작거나 같은 자연수이다.

셋째 줄에는 홍준이가 한 질문의 개수 M (1 ≤ M ≤ 1,000,000)이 주어진다.

넷째 줄부터 M개의 줄에는 홍준이가 명우에게 한 질문 S와 E가 한 줄에 하나씩 주어진다.

출력

총 M개의 줄에 걸쳐 홍준이의 질문에 대한 명우의 답을 입력으로 주어진 순서에 따라서 출력한다. 팰린드롬인 경우에는 1, 아닌 경우에는 0을 출력한다.

풀이

질의마다 구간이 팰린드롬인지 바로 답하려면, 모든 구간의 팰린드롬 여부를 미리 구해 두는 DP가 좋다. dp[l][r]를 l부터 r까지가 팰린드롬인지 여부로 두면 된다.

길이 1은 항상 팰린드롬이고, 길이 2는 두 값이 같은지만 보면 된다. 길이 3 이상은 양 끝 값이 같고 안쪽 구간 dp[l+1][r-1]이 팰린드롬일 때만 참이 된다.

짧은 구간부터 차례대로 채워 두면 이후 질의는 배열 조회만으로 끝난다.

코드

java
import java.util.*;
import java.io.*;
 
public class Main {
 
    static int[][] isPalindrome;
    static int[] numbers;
 
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();
 
        int N = Integer.parseInt(br.readLine());
        numbers = new int[N + 1];
 
        StringTokenizer st = new StringTokenizer(br.readLine());
        for (int i = 1; i <= N; i++)
            numbers[i] = Integer.parseInt(st.nextToken());
 
        checkPalindrome(N);
 
        int M = Integer.parseInt(br.readLine());
        while (M-- > 0) {
            st = new StringTokenizer(br.readLine());
            int S = Integer.parseInt(st.nextToken());
            int E = Integer.parseInt(st.nextToken());
            sb.append(isPalindrome[S][E]).append('\n');
        }
 
        System.out.println(sb);
    }
 
    static void checkPalindrome(int N) {
        isPalindrome = new int[N + 1][N + 1];
 
        // 길이 1
        for (int i = 1; i <= N; i++)
            isPalindrome[i][i] = 1;
 
        // 길이 2
        for (int i = 1; i <= N - 1; i++)
            if(numbers[i] == numbers[i + 1])
                isPalindrome[i][i + 1] = 1;
 
        // 길이 3+
        for (int i = N - 2; i >= 1; i--)
            for (int j = i + 2; j <= N; j++)
                if (numbers[i] == numbers[j] && isPalindrome[i + 1][j - 1] == 1)
                    isPalindrome[i][j] = 1;
    }
}

복잡도

  • 시간 복잡도: 전처리 O(N2)O(N^2), 질의당 O(1)O(1)
  • 공간 복잡도: O(N2)O(N^2)

마무리

구간 팰린드롬 문제는 질의마다 다시 보면 늦다. 짧은 구간부터 DP로 미리 채워 두면 이후 답은 바로 꺼낼 수 있다.