[JLOI2015]骗我呢

StaroForgin

于 2021-08-05 20:32:20 发布

阅读量219

点赞数 1

分类专栏： ------容斥------ # 组合

本文链接：https://blog.csdn.net/Tan_tan_tann/article/details/119425976

版权

组合同时被 2 个专栏收录

42 篇文章 0 订阅

订阅专栏

------容斥------

16 篇文章 0 订阅

订阅专栏

骗我呢

题解

首先我们可以将原题的模型转化一下。
对于矩阵上的每个点，它是比它右边与右上的两个点小的，而总共有 $m$ 列， $m + 1$ 个取值。
所以我们知道，对于每一行，它最多存在一个点使得这个点的右边的点比这个点大 $2$ ，其它地方都是大 $1$ 。
我们记第 $i$ 行的这个点的列标号为 $p_{i}$ ，很明显， $p$ 序列满足 $\forall\,i<j, p_{i}+i\leqslant p_{j}+j$ 。
我们相当于要求出满足这个条件的序列数。

这就很容易想到 $d p$ ，我们记 $dp_{i,j}$ 表示选到第 $i$ 个数， $p_{i}+i=j$ 时的合法方案数。
有转移式，
$dp_{i,j}=\sum_{k=i-1}^{j}dp_{i-1,k}$
很明显的前缀和结构，转移式再可以被表示为
$dp_{i,j}=dp_{i-1,j}+dp_{i,j-1}$

这不就是杨辉三角了吗？由于只有 $j\in[i,i+m]$ 的 $d p$ 才有值，所以我们可以联想到卡塔兰数的模型。
我们有两条直线 $x - y = 0$ 和 $x - y + m = 0$ ，我们走时不能超出这两条直线，也就是走到 $x - y - 1 = 0$ 与 $x - y + m + 1 = 0$ 上，我们的终点是 $(n + 1, n + m + 2)$ 。
我们分别记这两条直线为 $A$ 与 $B$ 。
由于我们可以从 $A$ 之间 $B$ 相互走，也就是从 $A$ 走到 $B$ 和从 $B$ 走到 $A$ ，我们不能直接求走到这两条直线的方案数，考虑容斥。
也就是说我们的答案为
$a n s = f - f (A) - f (B) + f (A B) + f (B A) - f (A B A) - f (B A B) + f (A B A B) + f (B A B A) - . . .$

但这些答案又该如何求呢？
$f$ 明显是全部从 $(1, 1)$ 走到 $(n + 1, n + m + 2)$ 的方案，也就是 $\binom{2n+m+1}{n}$ 。
而 $f (A)$ 与 $f (B)$ 只要我们将终点沿着 $A$ 与 $B$ 翻折即可。
但对于 $A B, B A, A B A, B A B . . .$ 这些呢？
如果我们现在再直线 $A$ 上，走到直线 $B$ 上，相当于我们 $x$ 要比 $y$ 多走 $m + 2$ 步。
假设我们原终点是 $(x, y)$ ，那么我们此时的新终点应是 $(y - m - 2, x + m + 2)$ 。
同理可得从 $B$ 走到 $A$ 的新终点是 $(y + 1, x - 1)$ 。
我们就一直枚举 $A B A B A . . .$ 与 $B A B A B . . .$ 的路径终点，直到走出去为止。
只需要先将组合数预处理后就可以得到答案了。
时间复杂度 $O\left(n+m\right)$ 。

源码

#include<bits/stdc++.h>
using namespace std;
#define MAXN 3000005
#define lowbit(x) (x&-x)
#define reg register
#define pb push_back
#define mkpr make_pair
#define fir first
#define sec second
typedef long long LL;
typedef unsigned long long uLL;
const int mo=1e9+7;
const int jzm=2333;
const int lim=10000000;
const int orG=3,invG=332748118;
const double Pi=acos(-1.0);
const double eps=1e-9;
typedef pair<int,int> pii;
template<typename _T>
_T Fabs(_T x){return x<0?-x:x;}
template<typename _T>
void read(_T &x){
	_T f=1;x=0;char s=getchar();
	while(s>'9'||s<'0'){if(s=='-')f=-1;s=getchar();}
	while('0'<=s&&s<='9'){x=(x<<3)+(x<<1)+(s^48);s=getchar();}
	x*=f;
}
template<typename _T>
void print(_T x){if(x<0){x=(~x)+1;putchar('-');}if(x>9)print(x/10);putchar(x%10+'0');}
LL gcd(LL a,LL b){return !b?a:gcd(b,a%b);}
int add(int x,int y,int p){return x+y<p?x+y:x+y-p;}
int qkpow(int a,int s,int p){int t=1;while(s){if(s&1LL)t=1ll*a*t%p;a=1ll*a*a%p;s>>=1LL;}return t;}
int n,m,dp[MAXN],ans,fac[MAXN],inv[MAXN],f[MAXN];
void init(){
	fac[0]=fac[1]=inv[0]=inv[1]=f[1]=1;
	for(int i=2;i<=n+n+m+1;i++){
		fac[i]=1ll*i*fac[i-1]%mo;
		f[i]=1ll*(mo-mo/i)*f[mo%i]%mo;
		inv[i]=1ll*f[i]*inv[i-1]%mo;
	}
}
int C(int x,int y){if(x<0||y<0||x<y)return 0;return 1ll*fac[x]*inv[y]%mo*inv[x-y]%mo;}
int sakura1(int x,int y);
int sakura2(int x,int y);
int sakura1(int x,int y){if(x<0||y<0)return 0;return add(C(x+y,x),mo-sakura2(y+1,x-1),mo);}
int sakura2(int x,int y){if(x<0||y<0)return 0;return add(C(x+y,x),mo-sakura1(y-m-2,x+m+2),mo);}
signed main(){
	read(n);read(m);init();int ans=C(n+n+m+1,n);
	ans=add(ans,mo-sakura1(n-1,n+m+2),mo);
	ans=add(ans,mo-sakura2(n+m+2,n-1),mo);
	printf("%d\n",ans);
	return 0;
}