【九校3D2T2】交错的字符串
程序员文章站
2024-03-20 18:11:52
...
Description
Mark Douglas是一名律师。他的客户Yuri Ball在一场车祸中不幸去世。为了帮助Yuri的亲属拿到他的遗产,Mark需要得到一个密码。在Yuri的笔记本上,有一个长为2n的只包含小写字母的字符串,Mark知道密码恰好是将这个字符串分解为两个长度为n的子序列且它们构成的字符串恰好相反的方案数。我们认为两种分解方法是不同的,当且仅当两个下标集合构成的集合{S1, S2}是不同的,注意{S1, S2}和{S2, S1}我们认为是相同的分解方法。如cabaacba的合法分解共有cabaacba和cabaacba两种。Mark希望你能帮助他计算出密码,事成之后他决定分给你six million five hundred thousand dollars并邀请你去柬埔寨度假。
Input
输入文件名为string.in。
第一行为一个正整数n。
第二行为一个长度为2n的字符串,仅包含小写字母。
Output
输出文件名为string.out。
输出仅一行表示答案。
Sample Input
4
cabaacba
Sample Output
2
Hint
【数据范围】
对于1~7号测试点(28%):1<=n<=10。
对于8~10号测试点(12%):字符串中仅有小写字母a。
对于11~14号测试点(16%):字符串中除去两个b外其余全是a。
对于15~18号测试点(16%):字符串中仅有a, b两种字符。
对于所有测试点(100%):1<=n<=18。
折半枚举。小技巧左边h1+右边h1'=左边h0+右边h0'->h1-h0=h0'-h1',这样枚举出前一半了后面一半O(1)计算,时间复杂度2^n
需要注意的是因为涉及减法?所以单哈希也许会错。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int Maxn=45;
const ll e1=233,p1=1e9+7,e2=131,p2=998244353;
ll f1[Maxn]={1},f2[Maxn]={1};
char s[Maxn];
int n,tot;
map<pair<ll,ll>,ll>Count;
int main(){
scanf("%d%s",&n,s+1);
for(int i=1;i<=n;++i)
f1[i]=f1[i-1]*e1%p1,f2[i]=f2[i-1]*e2%p2;
tot=(1<<n)-1;
for(int i=0;i<=tot;++i){
ll _h0=0,_h1=0,h0_=0,h1_=0,st=0,ed=n;
for(int j=0;j<n;++j)
if(i&(1<<j)){
++st;
(_h1+=f1[st]*(s[j+1]-'a'+1)%p1)%=p1;
(h1_+=f2[st]*(s[j+1]-'a'+1)%p2)%=p2;
}else {
(_h0+=f1[ed]*(s[j+1]-'a'+1)%p1)%=p1;
(h0_+=f2[ed]*(s[j+1]-'a'+1)%p2)%=p2;
--ed;
}
++Count[make_pair(_h1-_h0,h1_-h0_)];
}
long long ans=0;
for(int i=0;i<=tot;++i){
ll _h0=0,_h1=0,h0_=0,h1_=0,st=n-__builtin_popcount(i),ed=st;
for(int j=0;j<n;++j)
if(i&(1<<j)){
++st;
(_h1+=f1[st]*(s[n+j+1]-'a'+1)%p1)%=p1;
(h1_+=f2[st]*(s[n+j+1]-'a'+1)%p2)%=p2;
}else {
(_h0+=f1[ed]*(s[n+j+1]-'a'+1)%p1)%=p1;
(h0_+=f2[ed]*(s[n+j+1]-'a'+1)%p2)%=p2;
--ed;
}
ans+=Count[make_pair(_h0-_h1,h0_-h1_)];
}
cout<<ans/2<<endl;
return 0;
}