链接:https://www.nowcoder.com/acm/contest/71/D
来源:牛客网
题目描述
WYF正试图用一个栈来构造一棵树,现在他已经构造了n个元素作为树的节点,只要将这n个元素依次入栈出栈就可以形成一棵树了。当然,这个问题与树并没有关系,所以它叫做WYF的栈。每次你可以入栈一个新元素或者当栈非空时出栈一个元素,n个元素必须依次入栈,而WYF希望其中第m个元素入栈之后,栈中恰好有k个元素,现在他想知道一共有多少种入栈出栈顺序满足这个条件。
输入描述:
第一行一个正整数T,表示数据组数。(1<=T<=10000)
对于每组数据包含一行三个正整数n,m,k。
输出描述:
对于每组数据输出一个正整数表示答案。
由于答案可能过大,所以只需要输出对109+7取模后的答案
示例1
输入
2
3 3 3
3 3 2
输出
1
2
示例2
输入
5
10 3 2
10 2 2
10 7 5
10 6 2
10 7 6
输出
6864
11934
2200
3780
924
示例3
输入
2
5 4 4
5 2 1
输出
5
14
备注:
1<=n,m,k<=10^6
思路:
卡特兰数参考这个大神的博客这里写链接内容
剩下的就很简单了。。。
accode
#include<bits/stdc++.h>
#define LL long long
#define INF 0x3f3f3f3f
using namespace std;
const int maxn = 1e6+2;
const LL mod = 1e9+7;
LL jie[maxn*2];
int n;
int m;
int k;
int T;
LL ksm(LL a,LL b)
{
LL ret = 1;
while(b){
if(b&1) ret = ret%mod*a%mod%mod;
a = a%mod*a%mod%mod;
b>>=1;
}
return ret;
}
void init()
{
jie[0] = jie[1] = 1;
for(int i = 1;i<=2*maxn;i++){
jie[i] = jie[i-1]%mod*i%mod%mod;
}
}
LL C(LL n,LL m)
{
if(m<0) return 0;
return (jie[n]%mod*ksm(jie[n-m]%mod*jie[m]%mod%mod,mod-2)%mod)%mod;
}
LL Cal(LL n,LL m)
{
// cout<<"fdwfw"<<' '<<n+m<<' '<<m<<endl;
// cout<<C(n+m,m)<<' '<<C(n+m,m-1)<<endl;
return (C(n+m,n)%mod-C(n+m,m-1)%mod+mod)%mod;
}
int main()
{
scanf("%d",&T);
init();
while(T--){
scanf("%d%d%d",&n,&m,&k);
if(m<k){
puts("0");
continue;
}
printf("%lld\n",Cal(n-m+k,n-m)%mod*Cal(m-1,m-k)%mod%mod);
}
}