bzoj2160 拉拉队排练

题目：

艾利斯顿商学院篮球队要参加一年一度的市篮球比赛了。拉拉队是篮球比赛的一个看点，好的拉拉队往往能帮助球队增加士气，赢得最终的比赛。所以作为拉拉队队长的楚雨荨同学知道，帮助篮球队训练好拉拉队有多么的重要。拉拉队的选拔工作已经结束，在雨荨和校长的挑选下，n位集优秀的身材、舞技于一体的美女从众多报名的女生中脱颖而出。这些女生将随着篮球队的小伙子们一起，和对手抗衡，为艾利斯顿篮球队加油助威。一个阳光明媚的早晨，雨荨带领拉拉队的队员们开始了排练。n个女生从左到右排成一行，每个人手中都举了一个写有26个小写字母中的某一个的牌子，在比赛的时候挥舞，为小伙子们呐喊、加油。雨荨发现，如果连续的一段女生，有奇数个，并且他们手中的牌子所写的字母，从左到右和从右到左读起来一样，那么这一段女生就被称作和谐小群体。现在雨荨想找出所有和谐小群体，并且按照女生的个数降序排序之后，前K个和谐小群体的女生个数的乘积是多少。由于答案可能很大，雨荨只要你告诉她，答案除以19930726的余数是多少就行了。

Input

输入为标准输入。第一行为两个正整数n和K，代表的东西在题目描述中已经叙述。接下来一行为n个字符，代表从左到右女生拿的牌子上写的字母。

Output

输出为标准输出。输出一个整数，代表题目描述中所写的乘积除以19930726的余数，如果总的和谐小群体个数小于K，输出一个整数-1。

思路：

Manacher算法。

因为求的是奇数长度的回文串，因此不需要在字符串的中间加特殊字符，只需要在字符串的首尾加上不相同的特殊字符就行了，这样求出的回文串就是奇数长度的回文串。

假设以i为中心的回文串的长度是R，那么需要区间修改[1,R]，我们可以用差分数组，d[1]++,d[R+1]--，最后累加得到每个长度的回文串的个数

代码：

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<iostream>
#include<algorithm>
#include<ctime>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<string>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<stack>
#include<list>
#include<numeric>
using namespace std;
#define LL long long
#define ULL unsigned long long
#define INF 0x3f3f3f3f3f3f3f3f
#define mm(a,b) memset(a,b,sizeof(a))
#define PP puts("*********************");
template<class T> T f_abs(T a){ return a > 0 ? a : -a; }
template<class T> T gcd(T a, T b){ return b ? gcd(b, a%b) : a; }
template<class T> T lcm(T a,T b){return a/gcd(a,b)*b;}
// 0x3f3f3f3f3f3f3f3f

const LL MOD=19930726;
const int maxn=1e6+50;
char s[maxn];
int p[maxn];
LL ans[maxn],d[maxn];
int Manacher(char *s){
    int i,j=0,len=strlen(s),mx=0,res=0;
    for(i=1;i<len;i++){
        if(mx>i)
            p[i]=min(p[2*j-i],mx-i);
        else
            p[i]=1;
        for(;s[i+p[i]]==s[i-p[i]];p[i]++);
        if(p[i]+i>mx){
            mx=p[i]+i;
            j=i;
        }
        res=max(res,p[i]-1);
    }
    return res;//返回最长回文子串的长度
}
LL pow_mod(LL a,LL b){
    LL res=1;
    while(b){
        if(b&1) res=res*a%MOD;
        a=a*a%MOD;
        b/=2;
    }
    return res;
}
int main(){

    int n;
    LL k;
    while(~scanf("%d%lld",&n,&k)){
        scanf("%s",s+1);
        s[0]='$';
        s[n+1]='#';
        s[n+2]='\0';
        Manacher(s);
        mm(d,0);
        for(int i=1;i<=n;i++){
            d[1]++;
            d[p[i]+1]--;
        }
        ans[0]=0;
        int pos=0;
        for(int i=1;i<=n;i++){
            ans[i]=ans[i-1]+d[i];
            if(ans[i]>0)
                pos=i;
        }
        LL mul=1;
        for(int i=pos;i>=1;i--){
            LL b=min(k,ans[i]);
            mul=mul*pow_mod((LL)(2*i-1),b)%MOD;
            k-=b;
            if(k==0)
                break;
        }
        if(k==0)
            printf("%lld\n",mul);
        else
            printf("-1\n");
    }
    return 0;
}

回文树写法

代码：

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<iostream>
#include<algorithm>
#include<ctime>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<string>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<stack>
#include<list>
#include<numeric>
using namespace std;
#define LL long long
#define ULL unsigned long long
#define INF 0x3f3f3f3f3f3f3f3f
#define mm(a,b) memset(a,b,sizeof(a))
#define PP puts("*********************");
template<class T> T f_abs(T a){ return a > 0 ? a : -a; }
template<class T> T gcd(T a, T b){ return b ? gcd(b, a%b) : a; }
template<class T> T lcm(T a,T b){return a/gcd(a,b)*b;}
// 0x3f3f3f3f3f3f3f3f

const LL MOD=19930726;
const int MAXN = 1e6+50, SIZE = 26;
struct Palindromic_Tree {
	int next[MAXN][SIZE];//next指针
	int fail[MAXN];//fail指针
	int cnt[MAXN];//表示节点i表示的回文串的个数(建树时求出的不是完全的,最后Count()函数跑一遍以后才是正确的)
	int num[MAXN];//表示以节点i表示的最长回文串的最右端点为回文串结尾的回文串个数
	int len[MAXN];//len[i]表示节点i表示的回文串的长度（一个节点表示一个回文串）
	int S[MAXN];//存放添加的字符
	int last;//指向上一个字符所在的节点,方便下一次add
	int n;//字符数组指针
	int p;//节点指针
	int NewNode(int L) {//新建节点
		for(int i = 0; i < SIZE; ++i)
			next[p][i] = 0;
		cnt[p] = num[p] = 0;
		len[p] = L;
		return p++;
	}
	void Init() {//初始化
		p = n = 0;
		NewNode(0); NewNode(-1);
		last = 0;
		S[n] = -1;//开头放一个字符集中没有的字符,减少特判
		fail[0] = 1;
	}
	int GetFail(int x) {//和KMP一样,失配后找一个尽量最长的
		while(S[n - len[x] - 1] != S[n]) x = fail[x];
		return x;
	}
	void Add(int c) {
		S[++n] = c;
		int cur = GetFail(last);//通过上一个回文串找这个回文串的匹配位置
		if(!next[cur][c]) {//如果这个回文串没有出现过,说明出现了一个新的本质不同的回文串
			int now = NewNode(len[cur] + 2);//新建节点
			fail[now] = next[GetFail(fail[cur])][c];//和AC自动机一样建立fail指针,以便失配后跳转
			next[cur][c] = now;
			num[now] = num[fail[now]] + 1;
		}
		last = next[cur][c];
		cnt[last]++;
	}
	void Count() {
		for(int i = p - 1; i >= 0; --i)
			cnt[fail[i]] += cnt[i];
		//父亲累加儿子的cnt,因为如果fail[v]=u,则u一定是v的子回文串！
	}
}tree;
LL pow_mod(LL a,LL b){
    LL res=1;
    while(b){
        if(b&1) res=res*a%MOD;
        a=a*a%MOD;
        b/=2;
    }
    return res;
}
char s[MAXN];
int num[MAXN];
int main(){

    int n;
    LL k;
    while(~scanf("%d%lld",&n,&k)){
        scanf("%s",s);
        tree.Init();
        for(int i=0;s[i]!='\0';i++)
            tree.Add(s[i]-'a');
        tree.Count();
        mm(num,0);
        for(int i=2;i<tree.p;i++)
            num[tree.len[i]]+=tree.cnt[i];
        LL mul=1;
        for(int i=n;i>=1;i--){
            if(i%2==0) continue;
            LL b=min(k,(LL)num[i]);
            mul=mul*pow_mod((LL)i,b)%MOD;
            k-=b;
            if(k==0)
                break;
        }
        if(k==0)
            printf("%lld\n",mul);
        else
            printf("-1\n");
    }
    return 0;
}