题目
给定 n 个 0 和 n 个 1,它们将按照某种顺序排成长度为 2n 的序列,求它们能排列成的所有序列中,能够满足任意前缀序列中 0 的个数都不少于 1 的个数的序列有多少个。
输出的答案对 109+7 取模。
输入格式
共一行,包含整数 n。
输出格式
共一行,包含一个整数,表示答案。
数据范围
1≤n≤105
输入样例:
3
输出样例:
5
思路
可以转化为二维坐标问题,0代表向右走一步,1代表向上走一步。
要求任意前缀序列中 0 的个数都不少于 1 ,则是说明这条路径上的所有点(x,y)都必须满足x<y,则为在对角线下方。
因此我们计算合法路径的数量方法为 :方法总数 - 在对角线上方的路径数(即为穿越红线的路径数)
因为任何一个路径如果他有一个点在绿色的对角线之上的话,那么我们可以通过它与红色线的交点对后面的部分进行轴对称,我们发现所有的路径最终都到达了(5,7)这个点
因此计算公式为 c(2n n) - c(2n n -1)
推公式得到 结果为C(2n n) / (n + 1)
代码
#include<iostream>
using namespace std;
typedef long long LL;
const int mod = 1e9 + 7;
int qmi(int a, int k, int p)
{
int res = 1;
while(k)
{
if(k & 1) res = (LL)res * a % p;
a = (LL)a * a % p;
k >>= 1;
}
return res;
}
int main()
{
int n;
cin >> n;
int a = 2 * n, b = n;
int res = 1;
for(int i = a; i >= a - b + 1; i --) res = (LL)res * i % mod;
for(int i = 1; i <= b; i ++) res = (LL)res * qmi(i, mod - 2, mod) % mod;
res = (LL)res * qmi(n + 1, mod - 2, mod) % mod;
cout << res;
return 0;
}