题目地址:传送门
先用dp求出所有的符合要求的半回文串,标记出来。然后构造字典树。然后再dfs一遍求出所有节点的子树和,最后搜一遍就能找出第k个来了。
代码如下:
#include <iostream>
#include <string.h>
#include <math.h>
#include <queue>
#include <algorithm>
#include <stdlib.h>
#include <map>
#include <set>
#include <stdio.h>
#include <time.h>
using namespace std;
#define LL __int64
#define pi acos(-1.0)
//#pragma comment(linker, "/STACK:1024000000")
const int mod=1e9+7;
const int INF=0x3f3f3f3f;
const double eqs=1e-9;
const int MAXN=400000+10;
int ok[5001][5001];
int k, cnt;
char s[6000], str[6000];
struct node
{
int flag, sum;
node *next[2];
};
node *newnode()
{
int i;
node *p=new node;
for(i=0;i<2;i++){
p->next[i]=NULL;
}
p->flag=0;
p->sum=0;
return p;
}
void Insert(int st, int len, node *root)
{
int i, x;
node *p=root;
for(i=st;i<len;i++){
x=s[i]-‘a‘;
if(p->next[x]==NULL)
p->next[x]=newnode();
p=p->next[x];
if(ok[st][i]) {
p->flag++;
p->sum++;
}
}
}
void dfs(node *u)
{
for(int i=0;i<2;i++){
if(u->next[i]!=NULL){
dfs(u->next[i]);
u->sum+=u->next[i]->sum;
}
}
}
void seach(node *root)
{
node *p=root;
while(1){
k-=p->flag;
if(k<=0){
return ;
}
if(p->next[0]!=NULL){
if(p->next[0]->sum>=k){
str[cnt]=‘a‘;
p=p->next[0];
cnt++;
continue ;
}
else k-=p->next[0]->sum;
}
str[cnt]=‘b‘;
p=p->next[1];
cnt++;
}
}
int main()
{
int i, j, len, h;
while(scanf("%s",s)!=EOF){
scanf("%d",&k);
len=strlen(s);
memset(ok,0,sizeof(ok));
for(i=0;i<len;i++){
ok[i][i]=1;
if(i<len-1&&s[i]==s[i+1]) ok[i][i+1]=1;
if(i<len-2&&s[i]==s[i+2]) ok[i][i+2]=1;
if(i<len-3&&s[i]==s[i+3]) ok[i][i+3]=1;
}
for(i=5;i<=len;i++){
for(j=0;j<len-i+1;j++){
h=j+i-1;
if(s[j]==s[h]&&ok[j+2][h-2]) ok[j][h]=1;
}
}
node *root;
root=newnode();
for(i=0;i<len;i++){
Insert(i, len, root);
}
dfs(root);
cnt=0;
seach(root);
for(i=0;i<cnt;i++){
printf("%c",str[i]);
}
puts("");
}
return 0;
}
版权声明:本文为博主原创文章,未经博主允许不得转载。
时间: 2024-10-23 16:14:27