BZOJ 4403 序列统计 题解

题面

传送门

分析

我们首先考虑长度为 $l$ ，元素大小在 $L$ 到 $R$ 的==单调不降序列==的个数。设元素大小为 $L$ 到 $R$ 的元素的个数分别为 $x_1, x_2, x_3,\cdots, x_{R-L+1}$ 则题目转化为求方程 $\sum_{i=1}^{R-L+1}\,x_i=l$ 的非负整数解的个数。

设 $y_i=x_i+1$ ，则转化为求方程$\sum_{i=1}^{R-L+1}\,y_i=l+R-L+1$ 的正整数解得个数。

设 $len=R-L+1$ ，隔板法，得到答案:

$$\binom{l+len-1}{len-1}$$
题目要求的是长度为 $1$ 到 $N$ ，元素大小在 $L$ 到 $R$ 的单调不降序列的个数。设答案为 $Ans$ ，则
$$Ans=\binom{1+len-1}{len-1}+\binom{2+len-1}{len-1}+\binom{3+len-1}{len-1}+\cdots+\binom{N+len-1}{len-1}$$
等号两边同时加上 $\binom{len}{len}$ ，得到
$$Ans+\binom{len}{len}\\=\binom{len}{len}+\binom{1+len-1}{len-1}+\binom{2+len-1}{len-1}+\cdots+\binom{N+len-1}{len-1}$$
化简， 得
$$Ans=\binom{N+len}{len}-1$$

到现在为止思路已经讲完了。然而 $1\leq N,L,R\leq 10^9$ ，阶乘直接上天。

Lucas定理

对于非负整数m和n和素数p， 同余式:

$$\binom{m}{n}\equiv\prod_{i=0}^{k}\,\binom{m_i}{n_i}\,\,\,\,\,\,\left(\bmod p\right),$$
成立，其中：
$$m=m_kp^k+m_{k-1}p^{k-1}+\cdots+m_1p+m_0,$$
并且
$$n=n_kp^k+n_{k-1}p^{k-1}+\cdots+n_1p+n_0$$
是m和n的p进制展开。 当 $m<n$ 时，二项式系数 $\binom{m}{n}=0$ 。

由于结果要对 $10^6+3$ 取模，我们只需要将 $N+len$ 和 $len$ 转化为 $10^6+3$ 进制，再将他们对应的每一位用一遍组合公式。因此我们需要预处理出 $0$ 到 $10^6+2$ 的阶乘和它们在模 $10^6+2$ 意义下的逆。阶乘可以从0递推得到，然后使用费马小定理，用快速幂计算出 $10^6+2$ 的逆，然后反着推回去。

代码实现

#include <cstdio>

typedef long long ll;
const int ha=1000003;

int ksm(int x, int a) {
    int ret=1;
    for(; a; ret=(a%2)?(ll)ret*x%ha:ret, x=(ll)x*x%ha, a/=2);
    return ret;
}

int jc[ha], ni[ha];

void init() {
    jc[0]=1;
    for(int i=1; i<ha; i++) jc[i]=(ll)jc[i-1]*i%ha;
    ni[ha-1] = ksm(jc[ha-1], ha-2);
    for(int i=ha-1; i; i--) ni[i-1]=(ll)ni[i]*i%ha;
}

int c(int n, int m) {
    if(n<m) return 0;
    return (ll)jc[n]*ni[m]%ha*ni[n-m]%ha;
}

int C(int n, int m) {
    int ret=1;
    for(; n&&ret; ret=(ll)ret*c(n%ha,m%ha)%ha, n/=ha, m/=ha);
    return ret;
}

int T, n, L, R;

int main() {
    scanf("%d", &T);
    init();
    while(T--) {
        scanf("%d%d%d", &n, &L, &R);
        int len=R-L+1;
        printf("%d\n", (C(n+len, len)-1+ha)%ha);
    }
    return 0;
}