codechef CBAL

来源：互联网发布：将mdf文件导入数据库编辑：程序博客网时间：2024/06/05 02:37

传送门：https://www.codechef.com/problems/CBAL

思路：

先求一遍出现次数前缀和，我们只管每个字母出现次数奇偶性，所以可以把状态压缩一下，离散化之后就只有最多n个状态
对于子串[l,r]如果s[l-1]==s[r]，那么[l,r]合法
f 前i块，状态j，出现下标k次方之和，ans 块i到j的答案
完整的块直接得答案，外面剩余的一个一个加进来
假设现在加进来的位置是b，状态为s[b]
那么对于0次方就不说什么了，就是s[b]状态的出现次数
对于1次方由Σ(b-i)^1=b*sum[i][[0]-sum[i][1]（s[i]==s[b]）可算得
对于2次方由Σ(b-i)^2=(b^2-2*b*sum[i][1]+sum[i][2])(s[i]==s[b])可算得

#include<cmath>#include<cstdio>#include<cstring>#include<cassert>#include<iostream>#include<algorithm>const int maxn=100010,maxb=320;typedef long long ll;using namespace std;int cas,n,Q,sta[maxn],pw[29],tmp[maxn],cnt,bcnt,bsz,bel[maxn],seq[maxn];char s[maxn];ll A,B,lastans,f[maxb][maxn][3],ans[maxb][maxb][3],sum[maxn][3];struct seg{int l,r;}blk[maxb];void init(){pw[0]=1;for (int i=1;i<=26;i++) pw[i]=pw[i-1]<<1;scanf("%s",s+1),n=strlen(s+1),scanf("%d",&Q);sta[0]=tmp[0]=0;for (int i=1;i<=n;i++) sta[i]=sta[i-1]^pw[s[i]-'a'],tmp[i]=sta[i];sort(tmp,tmp+n+1),cnt=unique(tmp,tmp+n+1)-tmp;for (int i=0;i<=n;i++) sta[i]=upper_bound(tmp,tmp+cnt,sta[i])-tmp;}void trans(ll a[],ll sum[],int b,int op){a[0]+=sum[0];a[1]+=op*(sum[0]*b-sum[1]);a[2]+=sum[0]*b*b-2ll*sum[1]*b+sum[2];sum[0]++,sum[1]+=b,sum[2]+=1ll*b*b;}void prework(){for (int i=1;i<maxb;i++) blk[i].l=0;bsz=sqrt(n);for (int i=0;i<=n;i++){bel[i]=i/bsz+1;if (!blk[bel[i]].l) blk[bel[i]].l=i;blk[bel[i]].r=i;}bcnt=bel[n];for (int i=1;i<=bcnt;i++){for (int j=1;j<=cnt;j++) for (int k=0;k<3;k++) f[i][j][k]=f[i-1][j][k];for (int j=blk[i].l;j<=blk[i].r;j++){int st=sta[j];f[i][st][0]++;f[i][st][1]+=j;f[i][st][2]+=1ll*j*j;}}for (int i=1;i<=bcnt;i++){int cnt=0;for (int j=i;j<=bcnt;j++){for (int k=0;k<3;k++) ans[i][j][k]=ans[i][j-1][k];for (int k=blk[j].l;k<=blk[j].r;k++){trans(ans[i][j],sum[sta[k]],k,1);seq[++cnt]=sta[k];}}for (int j=1;j<=cnt;j++) for (int k=0;k<3;k++) sum[seq[j]][k]=0;}}ll getf(int l,int r,int st,int type){return f[r][st][type]-f[l-1][st][type];}ll query(int L,int R,int type){int st=bel[L],ed=bel[R],cnt=0;ll res[3]={0,0,0};if (st<ed){for (int i=0;i<3;i++) res[i]=ans[st+1][ed-1][i];for (int i=blk[st].r;i>=L;i--){int stt=sta[i];if (!sum[stt][0]){seq[++cnt]=stt;for (int k=0;k<3;k++)sum[stt][k]=getf(st+1,ed-1,stt,k);}trans(res,sum[stt],i,-1);}for (int i=blk[ed].l;i<=R;i++){int stt=sta[i];if (!sum[stt][0]){seq[++cnt]=stt;for (int k=0;k<3;k++)sum[stt][k]=getf(st+1,ed-1,stt,k);}trans(res,sum[stt],i,1);}}else{for (int i=L;i<=R;i++){int stt=sta[i];if (!sum[stt][0]) seq[++cnt]=stt;trans(res,sum[stt],i,1);}}for (int i=1;i<=cnt;i++) for (int j=0;j<3;j++) sum[seq[i]][j]=0;return res[type];}void work(){lastans=A=B=0;for (int i=1,x,y,type;i<=Q;i++){scanf("%d%d%d",&x,&y,&type);x=(x+A)%n+1,y=(y+B)%n+1;if (x>y) swap(x,y);lastans=query(x-1,y,type);printf("%lld\n",lastans);A=B,B=lastans;}}int main(){scanf("%d",&cas);while (cas--) init(),prework(),work();return 0;}

0 0