Codeforces 1832 E Combinatorics Problem 解析【难度 2200】

题目地址

https://codeforces.com/problemset/problem/1832/E

题目抽象

给 n,$a_1$,x,y,m,k 六个整数，需要进行如下4步操作

1. 根据 $a_i=(a_{i-1}\cdot x+y)\%m,i\in [2,n]$ 生成数组a
2. 根据 $b_i=({\sum }_{j=1}^i(C_{i-j+1}^k\cdot a_j))\%998244353$ 生成数组b
3. 根据 $c_i=b_i\cdot i$ 生成数组c
4. 计算 $c_1\oplus c_2\oplus \cdots \oplus c_n$ 的结果（$\oplus$ 代表位运算的异或操作）

数据量：
$1\le n\le 10^7$
$0\le a_1,x,y,\le m$
$2\le m\le 998244353$
$1\le k\le 5$

题目类型

数学推导

解题思路

第1,3,4步都是非常直观，一直模拟直接计算出来
主要难点就是第二步，根据a计算b
下面进行拆解分析：
粗暴地计算每个 $b_i$ 的复杂度非常高，由于n最大可能是$10^7$，需要找一个方式可以让 $b_i$ 的计算可以在 $O(1)$ 的时间内完成

tips:
一般这种情况都是寻找 $b_i$ 和 $b_{i-1}$的关系，使得 $b_i$ 可以通过 $b_{i-1}$ 快速得出
而找他们的关系，就是将他们展开，然后相减，大部分情况都适用

我们将其展开，然后相减来得出其关系
$$\begin{array}{clcrccccc}{b}_i& =& C_i^k\cdot a_1& +& C_{i-1}^k\cdot a_2& \dots & C_2^k\cdot a_{i-1}& +& C_1^k\cdot a_i\\ b_{i-1}& =& C_{i-1}^k\cdot a_1& +& C_{i-2}^k\cdot a_2& \dots & C_1^k\cdot a_{i-1}& & \\ b_i-b_{i-1}& =& (C_i^k-C_{i-1}^k)\cdot a_1& +& (C_{i-1}^k-C_{i-2}^k)\cdot a_2& \dots & (C_2^k-C_1^k)\cdot a_{i-1}& +& C_1^k\cdot a_i\end{array}$$
$\because$ 组合数有这个性质 $C_i^k=C_i^{k-1}+C_{i-1}^{k-1}$
$\therefore b_i-b_{i-1}=(C_{i-1}^{k-1}\cdot a_i+\dots C_1^{k-1}\cdot a_{i-1})+C_1^k\cdot a_i$
括号里边的内容是不是很眼熟，和$b_{i-1}$非常的像，就是当k为k-1时候的$b_{i-1}$
我们将b扩展一维，原本的$b_i$用$b_{k,i}$代替，那么就得到
$b_{k,i}-b_{k,i-1}=b_{k-1,i-1}+C_1^k\cdot a_i$
$\therefore b_{k,i}=b_{k,i-1}+b_{k-1,i-1}+C_1^k\cdot a_i$
所以，我们值需要计算出$b_{0,i}$和$b_{k,0}$，剩余的$b_{k,i}$就能通过前序已经计算出的结果得到
根据定义可得
$b_{0,i}=a_i+a_{i-1}+\dots a_1=b_{k,i-1}+a_i$
$b_{k,0}=0$
再根据 $b_{k,i}=b_{k,i-1}+b_{k-1,i-1}+C_1^k\cdot a_i$ (当k=1时，$C_1^k=1$，其他情况等于0) 就可以根据前置计算所得$O(1)$的复杂度计算得出
整体复杂度$O(n\cdot k)$

代码

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

const int mod = 998244353;

long long a[10000001];
long long b[6][10000001];

int main() {
  int n, x, y, m, K;
  cin >> n;
  cin >> a[1] >> x >> y >> m >> K;
  for (int i = 2; i <= n; i ++) {
    a[i] = (a[i - 1] * x + y) % m;
  }
  for (int i = 1; i <= n; i ++) {
    b[0][i] = b[0][i-1] + a[i];
  }
  for (int k = 1; k <= K; k ++) {
    for (int i = 1; i <= n; i ++) {
      b[k][i] = (b[k][i - 1] + b[k - 1][i - 1] + (k == 1 ? a[i] : 0)) % mod;
    }
  }
  long long ans = 0;
  for (int i = 1; i <= n; i ++) {
    ans ^= b[K][i] * i;
  }
  cout << ans << endl;
}