CF557E Ann and Half-Palindrome 字典树+dp

现在看这道题也不难啊，不知道考场上为啥没切~

code: 

#include <bits/stdc++.h> 
#define N 5006   
#define setIO(s) freopen(s".in","r",stdin) 
using namespace std; 
struct trie
{
    int size,tag,ch[2];    
}p[N*2500];  
char S[N];    
int f[N][N],tot;           
void dfs(int u) 
{
    p[u].size=p[u].tag;    
    for(int i=0;i<2;++i) 
    {
        if(p[u].ch[i]) 
        {
            int v=p[u].ch[i]; 
            dfs(v);   
            p[u].size+=p[v].size;     
        }   
    }
}
void dfs2(int u,int kth) 
{   
    kth-=p[u].tag;   
    if(kth<=0) 
    {    
        exit(0);    
    }   
    for(int i=0;i<2;++i) 
    {
        if(p[u].ch[i]) 
        { 
            int v=p[u].ch[i]; 
            if(kth>p[v].size) kth-=p[v].size;         
            else {
                printf("%c",i+'a');   
                dfs2(v, kth);      
                break;    
            }
        }
    }
}
int main() 
{ 
    int i,j,n,k,rt=0;    
    // setIO("input"); 
    scanf("%s%d",S+1,&k);  
    n=strlen(S+1);   
    for(i=1;i<=n;++i) f[i][i]=1;    
    for(i=n-1;i>=1;--i) 
    {
        for(j=i+1;j<=n;++j) 
        {    
            if(j-2>=i+2)                
            {   
                if(S[i]==S[j] && f[i+2][j-2]) f[i][j]=1;    
            }
            else
            {
                if(S[i]==S[j]) f[i][j]=1;       
            }
        }
    }       
    for(i=1;i<=n;++i) 
    { 
        rt=0;   
        for(j=i;j<=n;++j) 
        {    
            int c=S[j]-'a';    
            if(!p[rt].ch[c]) 
            {
                p[rt].ch[c]=++tot;   
            }    
            rt=p[rt].ch[c];     
            p[rt].tag+=f[i][j];        
        }
    }    
    dfs(0);     
    dfs2(0,k);    
    return 0; 
}