打牌的贝贝(卡特兰数,取模逆元)

链接:C-打牌的贝贝_牛客练习赛105 (nowcoder.com)

参考资料:

「算法入门笔记」卡特兰数 - 知乎 (zhihu.com)

卡特兰数:1 1 2 5 14 42...(下标从0开始)

主要公式:

 

1到2*n个数,甲分到的用左括号代替,乙分到的用右括号代替,那么乙赢的情况是对于任意甲的牌,乙都有更大的,可以转化为括号序列合法的情况数,即卡特兰数,那么甲赢得情况就用总数减去乙的。

细节见代码:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod=1e9+7;
const ll N=1e6+10;
ll fac[2*N],ifac[N],katalan[N],inv[N];//注意fac是2倍,因为有C(2*n,...) 
ll qsm(ll a,ll b)
{
	ll ans=1;
	while(b)
	{
		if(b&1)
		{
			//ans=(ans*a%mod)%mod;
			ans=ans*a%mod;
		}
		b/=2;
		a=a*a%mod;
	}
	return ans;
}
void init()
{
	ll R=N-5;
	fac[0]=1;//0!=1
	for(ll i=1;i<=2*R;i++)//阶乘 //注意fac是2倍,因为有C(2*n,...) 
	{
		fac[i]=fac[i-1]*i%mod;
	}
	
	ifac[R]=qsm(fac[R],mod-2);
	for(ll i=R-1;i>=1;i--)//阶乘逆元 
	{
		ifac[i]=ifac[i+1]*(i+1)%mod;
	}
	inv[1]=1;
	for(ll i=2;i<=R;i++)//由阶乘逆元求某数逆元 ,提高效率 
	{
		inv[i]=fac[i-1]*ifac[i]%mod;
	}
	
	katalan[0]=1;
	for(ll i=1;i<=R;i++) 
	{
		katalan[i]=((4*i-2)*katalan[i-1]%mod)*inv[i+1]%mod;
	}
}

ll C(ll a,ll b)
{
	return (fac[a]*ifac[b]%mod)*ifac[a-b]%mod;
}
int main()
{
	ios::sync_with_stdio(false);//不加会超时 
    cin.tie(0);
    cout.tie(0);
	ll t;
	cin>>t;
	init();//预处理中大量用到取模,注意 *和/ 优先级高于 % 
	while(t--)
	{
		ll n;
		cin>>n;
		//init();
		ll ans=(C(2*n,n)-katalan[n]+mod)%mod;
		cout<<ans<<' '<<(katalan[n]+mod)%mod<<'\n';
	}
	
	
	return 0;
}

 但是,如果n不止1e6呢?

阶乘逆元线性推导式:

 阶乘逆元 三种 快速 方法 费马小定理 递推 线性_mrcrack的博客-CSDN博客_阶乘的逆元

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod=1e9+7;
const ll N=1e6+10;
ll fac[2*N],katalan[N],inv[N],ifac[N]={1,1},f0[N]={1,1};//注意fac是2倍,因为有C(2*n,...) 
ll qsm(ll a,ll b)
{
	ll ans=1;
	while(b)
	{
		if(b&1)
		{
			//ans=(ans*a%mod)%mod;
			ans=ans*a%mod;
		}
		b/=2;
		a=a*a%mod;
	}
	return ans;
}
void init()
{
	ll R=N-5;
	fac[0]=1;//0!=1
	for(ll i=1;i<=2*R;i++)//阶乘 //注意fac是2倍,因为有C(2*n,...) 
	{
		fac[i]=fac[i-1]*i%mod;
	}
	
//	ifac[R]=qsm(fac[R],mod-2);
//	for(ll i=R-1;i>=1;i--)//阶乘逆元 
//	{
//		ifac[i]=ifac[i+1]*(i+1)%mod;
//	}


  // 区别: 
    for(ll i=2;i<=R;i++)//阶乘逆元(从2开始,0 1初始已经处理 )
	{
		f0[i]=(mod-mod/i)*f0[mod%i]%mod;
        ifac[i]=ifac[i-1]*f0[i]%mod;
	}
	inv[1]=1;
	for(ll i=2;i<=R;i++)//由阶乘逆元求某数逆元 ,提高效率 
	{
		inv[i]=fac[i-1]*ifac[i]%mod;
	}
	
	katalan[0]=1;
	for(ll i=1;i<=R;i++) 
	{
		katalan[i]=((4*i-2)*katalan[i-1]%mod)*inv[i+1]%mod;
	}
}

ll C(ll a,ll b)
{
	return (fac[a]*ifac[b]%mod)*ifac[a-b]%mod;
}
int main()
{
	ios::sync_with_stdio(false);//不加会超时 
    cin.tie(0);
    cout.tie(0);
	ll t;
	cin>>t;
	init();//预处理中大量用到取模,注意 *和/ 优先级高于 % 
	while(t--)
	{
		ll n;
		cin>>n;
		//init();
		ll ans=(C(2*n,n)-katalan[n]+mod)%mod;
		cout<<ans<<' '<<(katalan[n]+mod)%mod<<'\n';
	}
	
	
	return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值