wannafly 挑战赛9 D(卡特兰数)

链接: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);
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值