洛谷 P1659 [国家集训队]拉拉队排练(manacher)

题目链接:[国家集训队]拉拉队排练 - 洛谷

思路:一看n这么大,要求回文长度,肯定是manacher无疑了。注意只计算奇数长度,这个也好办,计算结果的时候只看奇数长度即可。重点是如何处理这个前 k 长的奇数回文长度的问题。朴素的做法是用优先队列存储前k长的长度。如下:

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int mod = 19930726;
int n, k;
string s,t;
priority_queue<int, vector<int>, greater<int> > q;
int p[23000000]; //要开到字符串长度两倍多
void manacher(const string& s){
	t="@#"; //头标记+分隔符
	for(char ch:s) {t+=ch; t+='#';} //加入字符和间隔符
	t += '*'; //尾标记
	int R=0, mid=0, res=0;
	for(int i=1; i<t.size()-1; i++){
		p[i] = i<R ? min(p[2*mid-i], R-i) : 1;
		while(t[i+p[i]] == t[i-p[i]]) p[i]++;
		if(i+p[i] > R) {R=i+p[i]; mid=i;}
		if(i%2==0 && (p[i]-1)%2){
			int x = p[i]-1;
			while(x>0){
				q.push(x);
				if(q.size() > k) q.pop();
				x-=2;
			}
		}
	}
}
signed main(){
	ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
	cin>>n>>k>>s;
	manacher(s);
	if(q.size() < k) {cout<<-1; return 0;}
	int ans = 1;
	while(k--){
		int top = q.top(); q.pop();
		ans *= top; ans %= mod;
	}
	cout<<ans;
}

理所当然地TLE了,因为数量实在太大。 

注意到每个回文中心对应的最大的长度与所有长度之间的关系:前者是一个数2k+1,后者是一个集合{1,3,5,7...2k+1}。举个例子,有一个字符串是abcba,那么以c为中心的回文长度最大是5,所有长度是{1,3,5}。根据这个连续性,我们只需要记录最大回文长度,用一个桶来存储。然后计算的时候算出前缀和。正解代码如下:

(补充一个要点:计算幂次要用快速幂,不然也会TLE)

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int mod = 19930726;
int n, k;
string s,t;
int p[2000050]; //要开到字符串长度两倍多
int cnt[2000050]; //记录最大长度的桶
int Pow(int a, int b){
    int res = 1, base = a;
    while(b>0){
        if(b&1) res = res*base%mod;
        base = base*base%mod;
        b >>= 1;
    }
    return res;
}
void manacher(const string& s){ //马拉车板子
	t="@#"; //头标记+分隔符
	for(char ch:s) {t+=ch; t+='#';} //加入字符和间隔符
	t += '*'; //尾标记
	int R=0, mid=0, res=0;
	for(int i=1; i<t.size()-1; i++){
		p[i] = i<R ? min(p[2*mid-i], R-i) : 1;
		while(t[i+p[i]] == t[i-p[i]]) p[i]++;
		if(i+p[i] > R) {R=i+p[i]; mid=i;}
		if((p[i]-1)%2) cnt[p[i]-1]++; //记录奇数最大长度
	}
}
signed main(){
	ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
	cin>>n>>k>>s;
	manacher(s);
	int sum = 0, ans = 1;
	for(int i=(n%2)?n:(n+1); i>=1; i-=2){ //只看奇数部分
		sum += cnt[i]; //长度为i的回文的总数量(前缀累加)
		if(k >= sum){
			ans = ans * Pow(i,sum) % mod;
			k -= sum;
		} else {
			ans = ans * Pow(i, k) % mod;
			k = 0;
		}
	}
	if(k > 0) {cout<<-1; return 0;} //个数不够
	else cout<<ans;
}

上一篇:CF1513C.Add One


下一篇:scanf,sscanf高级用法