51nod 1464 半回文

一开始想到后缀数组，然后发现好难处理对称的。参考了 f_zyj 的代码，发现用字典树做，对称采用了DP优化。这个DP优化好神奇啊。。。。

打完后想想，好像后缀数组也能做。。。。

#include<bits/stdc++.h>
using namespace std;

const int MAXN=200200;
char s[MAXN];
int cnt,n,k,vis[5050];
bool dp[5050][5050];
string ans;

struct Node
{
	int val;
	int son[2];
}trie[MAXN]; 

void insert(char *s,int index)
{
	for(int len=strlen(s),x=0,i=0;i<len&&i+index<=vis[index];i++)
	{
		if(!trie[x].son[s[i]-'a'])
			trie[x].son[s[i]-'a']=++cnt;
		if(dp[index][i+index])
			trie[trie[x].son[s[i]-'a']].val++;
		x=trie[x].son[s[i]-'a'];
	} 
}

void dfs(int x)
{
	string tmp=ans;
	for(int i=0;i<2;i++)
	{
		if(k>0&&trie[x].son[i])
		{
			k-=trie[trie[x].son[i]].val;
			ans=tmp+char('a'+i);
			dfs(trie[x].son[i]);
		}
	}
}

int main()
{
	int i,j;
	while(scanf(" %s %d",&s,&k)!=EOF)
	{
		n=strlen(s);
		memset(dp,0,sizeof(dp));
		for(i=n-1;i>=0;i--)
		{
			dp[i][i]=true;
			vis[i]=i;
			for(j=i+1;j<n;j++)
			{
				if(s[i]==s[j])
				{
					if(i+2>=j-2)
						dp[i][j]=true;
					else
						dp[i][j]=dp[i+2][j-2];
				}
				if(dp[i][j])
					vis[i]=j;
			}
		} 
		for(i=0;i<MAXN;i++)
		{
			trie[i].son[0]=trie[i].son[1]=0;
			trie[i].val=0;
		}
		cnt=0;
		for(i=0;i<n;i++)
			insert(s+i,i);
		ans="";
		dfs(0);
		cout<<ans<<endl;
	}
}