CodeForces245H - Number of Palindromes for Queries (구간 dp)

3500 단어
dp(i)(j)는 구간(i, j)에 몇 개의 회문열이 있는지 표시하고, dp(i)(j) = dp(i+1)(j)+dp(i)(j-1)-dp(i+1)(j-1)+flag(i, j).만약str(i)부터str(j)까지 회문열이면flag(i,j)는1
#include <bits/stdc++.h>
#define maxn 5005
#define INF 1e9
using namespace std;
typedef long long ll;

char str[maxn];
int dp[maxn][maxn];
int flag[maxn][maxn], len;
void solve(int i, int j){

    while(i >= 0 && j < len){

        if(str[i] == str[j])
           flag[i][j] = 1;
        else
         break;  
        i--;
        j++;    
    }
}
int main(){

// freopen("in.txt", "r", stdin);
    while(scanf("%s", str) == 1){

        memset(flag, 0, sizeof(flag));
        len = strlen(str);
        for(int i = 0; i < len; i++){
            solve(i, i);
            solve(i, i+1);
        }
        for(int i = 0; i < len; i++)
          dp[i][i] = 1;
        for(int i = 1; i < len; i++)
          for(int j = 0; i + j < len; j++){
            int k1 = j, k2 = i + j;
            dp[k1][k2] = dp[k1][k2-1] + dp[k1+1][k2] - dp[k1+1][k2-1] + flag[k1][k2];
          }
        int t;

        cin >> t;
        for(int i = 0; i < t; i++){

            int a, b;
            scanf("%d%d", &a, &b);
            printf("%d
"
, dp[a-1][b-1]); } } return 0; }

좋은 웹페이지 즐겨찾기