51nod - 1464 半回文

一个字符串t是半回文的条件是，对于所有的奇数i(1≤i≤|t|+12)，ti = t|t| − i + 1 始终成立，|t|表示字符串t的长度。下标从1开始。例如"abaa", "a", "bb", "abbbaa"都是半回文，而"ab", "bba"和"aaabaa"则不是。

现在有一个字符串s，只由小写字母a,b构成，还有一个数字k。现在要求找出s的半回文子串中字典序排在第k位的串，字符串可以是一样，只要所在的位置不同就是不一样的串。

样例解释：

这个样例中半回文子串是 a, a, a, a, aa, aba, abaa, abba, abbabaa, b, b, b, b, baab,bab, bb, bbab, bbabaab (按照字典序排序).

 

Input

单组测试数据。
第一行有一个字符串s(1 ≤ |s| ≤ 5000)，只包含'a' 和 'b'，|s|表示s的长度。
第二行有一个正整数k。k不超过s子串中是半回文串的总数目。

Output

输出排在第k位的半回文子串。

Input示例

abbabaab
7

Output示例

abaa

思路：

用dp[i][j]记录i,j之间能否构成半回文。vis[i]记录i为起点能构成半回文的最远距离。将字符串从左到右依次插入字典树中，并记录每个节点的半回文数量。 查找时先查a再查b。

#include <iostream>
#include <cstring>
using namespace std;

const int MAXN = 5005;
struct Node
{
    int count;
    char val;
    Node *child[2];
    Node()
    {
        count = 0;
        val = 'd';
        child[0] = NULL;
        child[1] = NULL;
    }
};
char s[MAXN];
int k;
bool dp[MAXN][MAXN];
Node *top = new Node();
char result[MAXN];
int t = 0;
int vis[MAXN];

void insert(int pos, int end)
{
    Node *p = top;
    int left = pos;
    while (pos <= end)
    {
        if (p->child[s[pos] - 'a'])
        {
            p = p->child[s[pos] - 'a'];
        }
        else
        {
            p->child[s[pos] - 'a'] = new Node();
            p = p->child[s[pos] - 'a'];
            p->val = s[pos];
        }
        if (dp[left][pos])
        {
            p->count++;
        }
        pos++;
    }
}

void find(Node *p)
{
    if (k < 1)
    {
        return;
    }
    
    if (p == top)
    {
        t = 0;
    }
    else
    {
        result[t] = p->val;
        t++;
    }
    
    k -= p->count;
    int prevT = t;
    if (k > 0 && p->child[0])
    {
        find(p->child[0]);
    }
       
    if (k > 0 && p->child[1])
    {
        t = prevT;
        find(p->child[1]);
    }
}

void init(int n)
{
    for (int i = n - 1; i >= 0; --i)
    {
        dp[i][i] = true;
        vis[i] = i;
        for (int 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;
            }
        }
    }
}

int main()
{
	cin >> s;
	cin >> k;
	memset(dp, false, sizeof(dp));
	int len = strlen(s);
	init(len);
	for (int i = 0; i < len; i++)
	{
	    insert(i, vis[i]);
	}
	
	find(top);
	
	for (int i = 0; i < t; i++)
	{
	    cout << result[i];
	}
	cout << endl;
	
	return 0;
}