HDU 6129 Just do it【杨辉三角+思维+Lucas定理】

题目链接

题意：$a_1$到$a_n$这n个数，每次操作使$a_i=a_1$^$a_2$^…^$a_i$，把这个操作重复$m$次，求最后所得的序列。

显然异或两次就等于没异或。考虑每个数在每一次操作后对每个数异或的次数，可以得到下面的矩阵。

$$\left[ \begin{matrix} 1&1&1&\cdots&1\\ 1&2&3&\cdots&n\\ 1&3&6&\cdots&dp[3][n]\\ 1&4&10&\cdots&dp[4][n]\\ \vdots&\vdots&\vdots&\ddots&\vdots\\ dp[m][1]&dp[m][2]&dp[m][3]&\cdots&dp[m][n] \end{matrix} \right]$$

其中第$i$行就是经过$i$次操作后所得的序列。

多写几个数可以发现规律，这是一个杨辉三角。（比赛时没想到这点）

杨辉三角第$n$行的第$m$个数等于$C_{n-1}^{m-1}$。对这题来说，单独考虑每个$a_i$，最后的数$dp[m][i]$是在杨辉三角中的第$i+m-1$行的第$i$个数，就等于$C_{i+m-2}^{i-1}$，而因为异或两次等于没异或，所以我们只要判断奇偶就可以了。

$C_n^m$为奇数当且仅当($n$&$m$)$==m$

还有一个Lucas定理可以用来判断$C(n,m)$% p <script type="math/tex" id="MathJax-Element-23">p</script>的值，顺便提一下。

想到这几点，后面的就很好写了。

#include <bits/stdc++.h>
using namespace std;

int a[200010];
int ans[200010];
int n,m;
int T;

int main(){
    scanf("%d",&T);
    while (T--){
        scanf("%d %d",&n,&m);
        memset(ans,0,sizeof(ans));
        for (int i=1;i<=n;i++)  scanf("%d",&a[i]);
        for (int i=1;i<=n;i++){
            int x=i+m-2;
            int y=i-1;
            if ((x&y)==y){
                for (int j=i;j<=n;j++){
                    ans[j]^=a[j-i+1];
                }
            }
        }
        for (int i=1;i<=n;i++){
            if (i==1)   printf("%d",ans[i]);
            else    printf(" %d",ans[i]);
        }
        putchar('\n');
    }
    return 0;
}