AtCoder Beginner Contest 205 E - White and Black Balls(卡特兰数变形)

最新推荐文章于 2022-04-30 13:06:07 发布

issue是fw

最新推荐文章于 2022-04-30 13:06:07 发布

阅读量409

点赞数 1

分类专栏： ABC题解

本文链接：https://blog.csdn.net/jziwjxjd/article/details/118574416

版权

ABC题解专栏收录该内容

15 篇文章 0 订阅

订阅专栏

这篇博客探讨了一种组合计数问题，涉及在坐标轴上从(0,0)出发，通过选择白球和黑球进行移动，最终到达特定点的合法路径数。文章利用了卡特兰数的概念，并给出了当白球和黑球数量已知时，如何计算合法路径数的公式。通过构造对称折线并分析边界条件，博主推导出了非法路径的数量，并通过C(n+m,n)减去非法路径数C(n+m,z)来求得合法路径总数。该问题适用于n≤mn≤mn≤m的情况。

摘要由CSDN通过智能技术生成

LINK

看到题目非常像卡特兰数,~~实际上就是~~

考虑以 $(0, 0)$ 为起点,选白球相当于往右下走一步,黑球相当于向右上走一步

白球 $n$ 个,黑球 $m$ 个,设 $x = m - n$ (先设 $m > = n$ )

放在坐标轴上,类似这样的一张图

不管怎么走,最后终点都是 $(n + m, x)$

总的方案是 $\binom{n+m}{n}$ ,我们需要知道不合法的方案是多少

容易看出,只要上面的折线越过了 $y = - k$ 这根直线就不合法

我们构造另一条折现,以 $y = - k - 1$ 为对称轴翻转得到

发现翻转后折线的终点变为 $(n + m, 2 k + 2 + x)$

也就是不合法的方案数相当于向下走 $z$ 步,满足 $z - (n + m - z) = 2 k + 2 + x$

于是真正的方案数为 $\binom{n+m}{n}-\binom{n+m}{z}$

当 $n < m$ 时同理

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int mod = 1e9+7;
const int maxn = 2e6+10;
int n,m,k,fac[maxn];
int quick(int x,int n)
{
	int ans = 1;
	for( ; n ; n>>=1,x=x*x%mod )
		if( n&1 )	ans = ans*x%mod;
	return ans;
}
int C(int n,int m)
{
	if( n<m )	return 0ll;
	return fac[n]*quick( fac[m]*fac[n-m]%mod,mod-2 )%mod;
}
signed main()
{
	fac[0] = 1;
	for(int i=1;i<=2000000;i++)	fac[i] = fac[i-1]*i%mod;
	cin >> n >> m >> k;
	if( m+k<n )	puts("0");
	else
	{
		//满足m+k>=n
		int ans = 0;
		if( m>=n )
		{
			int x = m-n;
			int z = (2*k+2+x+n+m)/2;
			ans = C(n+m,m)-C(n+m,z);
		}
		else
		{
			int x = n-m;
			int z = (2*k+2-x+n+m)/2;
			ans = C(n+m,m)-C(n+m,z);	
		}
		cout << ( ans%mod+mod )%mod;
	}
}