【51NOD 1501】【51NOD 算法马拉松19】石头剪刀布威力加强版

Description

小A和小B在玩石头剪刀布，他们每个人写出一个序列。
小A写出了n个数。
小B写出了m个数。
其中0代表石头，1代表剪刀，2代表布
0>1,1>2,2>0。
他们总共进行k轮，第一轮选择第一个数字，后面每一轮两个人都选择序列的下一个数进行比赛（序列结尾的下一个位置在序列开头）。
问小A和小B每人赢了几局。

Solution

这题很显然先把循环节给减掉，
设g=gcd(n,m)
循环节内的，对于a_i，它能访问到的b_j一定有imodg=jmodg，
对于循环节外的，很显然，对于位置i，所有位置j只要符合：jmodg=imodg，它在每轮循环中，依次访问b的点的顺序一定是循环同构的，
所以对于b中的1~g的数全部在a中跑一遍看能到哪些点，倍长以后预处理一下前缀和，
再O(n)扫一遍即可。

复杂度：O(n)

Code

#include <iostream>#include <cstdio>#include <cstdlib>#define fo(i,a,b) for(int i=a;i<=b;i++)using namespace std;typedef long long LL;const int N=1000500;int read(int &n){    char ch=' ';int q=0,w=1;    for(;(ch!='-')&&((ch<'0')||(ch>'9'));ch=getchar());    if(ch=='-')w=-1,ch=getchar();    for(;ch>='0' && ch<='9';ch=getchar())q=q*10+ch-48;n=q*w;return n;}bool JH;int m,n;LL ans,m1,n1,ans1,M1c;int a[N],b[N],c[N];LL f[N*3][3];int gcd(int x,int y){return y?gcd(y,x%y):x;}int main(){    int q;    read(n),read(m);scanf("%lld",&m1);M1c=m1;    fo(i,1,n)read(a[i]);    fo(i,1,m)read(b[i]);    if(n>m){fo(i,1,n)swap(a[i],b[i]);swap(n,m);JH=1;}    n1=gcd(n,m);    fo(i,1,n)f[i%n1][a[i]]++;    fo(i,1,m)ans+=f[i%n1][(b[i]+1)%3],ans1+=f[i%n1][b[i]];    ans=ans*(m1/(n/n1*m));ans1=ans1*(m1/(n/n1*m));    m1%=(n/n1*m);    fo(i,0,n1)fo(j,0,2)f[i][j]=0;    LL w=m1;m1/=m;q=0;    fo(i,1,n1)    {        int t=q;        for(int J=i-1;!c[J+1];J=(J+m)%n)f[++q][a[J+1]]++,c[J+1]=q;        fo(j,1,q-t)fo(k,0,2)f[q+j][k]=f[t+j][k];        q+=q-t;        fo(j,t,q)fo(k,0,2)f[j][k]+=f[j-1][k];    }    fo(i,1,m)    {        ans+=f[c[(i-1)%n+1]+m1-1][(b[i]+1)%3]-f[c[(i-1)%n+1]-1][(b[i]+1)%3];        ans1+=f[c[(i-1)%n+1]+m1-1][b[i]]-f[c[(i-1)%n+1]-1][b[i]];    }    q=(m1*m%n)%n;w=w%m;    fo(i,1,w)        if(b[i]==a[(q+i-1)%n+1])ans1++;            else if((b[i]+1)%3==a[(q+i-1)%n+1])ans++;    ans1=M1c-ans-ans1;    if(JH)swap(ans,ans1);    printf("%lld %lld\n", ans1,ans);    return 0;}