ARC117 C - Tricolor Pyramid（思维，找规律，杨辉三角）

题意：

解法：

将三种颜色看作0,1,2,发现颜色x和y上面的颜色是-(x+y)%3.

相当于每个颜色x向上传递的时候会变成-x,
我们只需要想办法统计每个数向上传递的总次数即可.

发现每个格子=下面两个格子的和,和杨辉三角一致.

观察得最底层第i个颜色在顶端的出现次数是C(n-1,i-1),如果n是偶数,那么需要加上一个负号.

算出顶端格子的值就知道顶端格子的颜色了.

code：

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxm=2e6+5;
char s[maxm];
int c[3][3];
int a[maxm];
int n;
void init(){
    for(int i=0;i<3;i++){
        c[i][0]=1%3;
        for(int j=1;j<=i;j++){
            c[i][j]=(c[i-1][j]+c[i-1][j-1])%3;
        }
    }
}
int C(int n,int m){
    if(m<0||m>n)return 0;
    return c[n][m];
}
int lucas(int n,int m,int mod){
    if(m==0)return 1%mod;
    return C(n%mod,m%mod)*lucas(n/mod,m/mod,mod)%mod;
}
void solve(){
    init();
    cin>>n>>(s+1);
    for(int i=1;i<=n;i++){
        if(s[i]=='R')a[i]=0;
        else if(s[i]=='W')a[i]=1;
        else if(s[i]=='B')a[i]=2;
    }
    int ans=0;
    for(int i=1;i<=n;i++){
        ans=(ans+lucas(n-1,i-1,3)*a[i]%3)%3;
    }
    if(n%2==0)ans=(-ans+3)%3;
    if(ans==0)cout<<"R"<<endl;
    else if(ans==1)cout<<"W"<<endl;
    else if(ans==2)cout<<"B"<<endl;
}
signed main(){
    ios::sync_with_stdio(0);
    solve();
    return 0;
}

---