「KDOI-04」Pont des souvenirs（组合数学+思维）

「KDOI-04」Pont des souvenirs

题目背景

虽然这是一个 C，但是

题目描述

给定正整数 $n,k$，求有多少个长度为 $n$ 的正整数序列 $a$ 满足：

• $0<a_1\le a_2\le a_3\le \cdots \le a_n\le k$；
• $\forall i\ne j$，$a_i+a_j\le k+1$。

答案对 $10^9+7$ 取模。

输入格式

本题包含多组测试数据。

输入的第一行包含一个正整数 $T$，表示测试数据组数。

对于每组测试数据，输入包含一行两个正整数 $n,k$。

输出格式

对于每组测试数据，输出一行一个整数表示答案。

样例 #1

样例输入 #1

5
2 2
1 3
4 5
4030 218
1145 1419

样例输出 #1

2
3
20
571656908
172735629

提示

【样例解释】

对于第 $1$ 组测试数据，所有满足要求的序列为 $(1,1)$ 和 $(1,2)$。

对于第 $2$ 组测试数据，所有满足要求的序列为 $(1)$，$(2)$ 和 $(3)$。

【数据范围】

对于 $100\%$ 的数据，保证 $1\le T\le 2\times 10^5$，$1\le n,k\le 10^7$。

本题开启捆绑测试。

子任务编号	分值	$T\le$	$n\le$	$k$
$1$	$8$	$5$	$5$	$\le 5$
$2$	$3$	$10^5$	$10^7$	$=1$
$3$	$3$	$10^5$	$10^7$	$=2$
$4$	$8$	$10^5$	$10^7$	$=3$
$5$	$16$	$10$	$200$	$\le 200$
$6$	$16$	$10$	$3000$	$\le 3000$
$7$	$8$	$10^4$	$10^7$	$\le 5$
$8$	$8$	$100$	$10^7$	$\le 10^5$
$9$	$30$	$2\times 10^5$	$10^7$	$\le 10^7$

思路

• 首先，我们为了满足条件：$0<a_1\le a_2\le a_3\le \cdots \le a_n\le k$，对于这种单调的，我们可以采用隔板法来做，因此在没有条件 $2$ 的约束下，这样的方案数为：$$C_{n+k-1}^{k-1}$$

• 为什么是这样呢？因为我们发现：

  令 $x_i$ 表示 $i$ 在序列种出现的个数。

  那么 ${\sum }_{i=1}^n{x}_i=n(x_i\ge 0)$

  接着，我们再令 $y_i$ 为 $x_i+1$。

  此时：$\sum _{i=1}^k{y}_i=n+k(y_i\ge 1)$。

因此这就是隔板法的推导。

• 那么此时我们加上条件 $2$，那么我们就得优先分析 $a_n$ 了，我们就暴力枚举它，我们就可以得到： 0 ≤ a 1 ≤ a 2 ≤ ⋯ ≤ a n − 1 ≤ min ⁡ { a n , k + 1 − a n } 0\le a_1\le a_2\le\cdots\le a_{n-1}\le\min\{a_n, k+1-a_n\} 0≤a1​≤a2​≤⋯≤an−1​≤min{an​,k+1−an​}。那么此时的方案数为： ∑ i = 1 k C n − 1 + min ⁡ { i , k + 1 − i } − 1 min ⁡ { i , k + 1 − i } − 1 \sum_{i=1}^{k}C_{n-1+\min\{i, k+1-i\}-1}^{\min\{i,k+1-i\}-1} ∑i=1k​Cn−1+min{i,k+1−i}−1min{i,k+1−i}−1​。

• 但我们发现，直接枚举的话，时间复杂度很高，因此我们尝试化简上述式子：

• 拓展：（很重要）
  - $C_n^m=C_n^{n-m}$；
  - $\sum _{i=0}^m{C}_i^n=C_{m+1}^{n+1}$。

  • 我们考虑 min ⁡ { i , k + 1 − i } \min\{i, k+1-i\} min{i,k+1−i} 的取值范围。
    • 我们令 $m=k+1$。
    • 则变成了 min ⁡ { i , m − i } \min\{i, m-i\} min{i,m−i} 的取值范围。
    • 即 $1\sim \lceil \frac{m}{2}\rceil$ 和 $1\sim \lfloor \frac{m}{2}\rfloor$。
      <=> $ans=\sum _{i=1}^{\lfloor \frac{k+1}{2}\rfloor }{C}_{n+i-2}^{i-1}+\sum _{i=1}^{\lceil \frac{k+1}{2}\rceil }{C}_{n+i-2}^{i-1}$。
      <=>我们枚举 $i-1$$\sum _{i=0}^{\lfloor \frac{k+1}{2}\rfloor -1}{C}_{n+i-1}^i+\sum _{i=0}^{\lceil \frac{k+1}{2}\rceil -1}{C}_{n+i-1}^i$。
      <=> 根据 $C_n^m=C_n^{n-m}$，此时：$\sum _{i=0}^{\lfloor \frac{k+1}{2}\rfloor -1}{C}_{n+i-1}^n+\sum _{i=0}^{\lceil \frac{k+1}{2}\rceil -1}{C}_{n+i-1}^n$。
      <=> 根据 $\sum _{i=0}^m{C}_i^n=C_{m+1}^{n+1}$，柿子就变成了 $C_{n+\lfloor \frac{k+1}{2}\rfloor }^n+C_{n+\lceil \frac{k+1}{2}\rceil }^n$。
      以上思路参考

细节

• 细节1：开 long long。
• 细节2：因为这个数据范围在 $10^6$ 到 $10^7$ 之间，而且还有取模，我们就可以用逆元法求组合数的方法来做。
• 细节3：那个 $N$ 最好开：$15000000$ （经验，这个调了很久）

AC 代码

#include<iostream>
#include<algorithm>
#include<cstring>

typedef long long ll;

using namespace std;

const int N = 15100000 , mod = 1e9+7;

ll fac[N],infac[N];
int T,n,k;

ll qmi(ll a,ll b){
    ll res=1;
    
    while(b){
        if(b&1)res=res*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return res;
}

ll C(ll a,ll b){
    a=a+b-1,b=b-1;
    if(b>a)return 0;
    return fac[a]*infac[b]%mod*infac[a-b]%mod;
}

void init(){
    infac[0]=fac[0]=1;
    
    for(register int i=1;i<=N-10;i++){
        fac[i]=fac[i-1]*i%mod;
        infac[i]=infac[i-1]*qmi(i,mod-2)%mod;
    }
}

int main(){
    scanf("%d",&T);
    
    init();
    
    while(T--){
        scanf("%d%d",&n,&k);
        
        //隔板法
        
        if(k&1){
            int p=(k+1)>>1;
            printf("%d\n",(C(n,p)+C(n,p-1))%mod);
        }else{
            int p=(k+1)>>1;
            printf("%d\n",(C(n,p)+C(n,p))%mod);
        }
        
    }
    
    return 0;
}