[CSP-S模拟测试]:题(DP+数学)
题目描述
出个题就好了.这就是出题人没有写题目背景的原因。
你在平面直角坐标系上。
你一开始位于$(0,0)$。
每次可以在上/下/左/右四个方向中选一个走一步。
即:从$(x,y)$走到$(x,y+1),(x,y-1),(x-1,y),(x+1,y)$四个位置中的其中一个。
允许你走的步数已经确定为$n$,现在你想走$n$步之后回到$(0,0)$。
但这太简单了,你希望知道有多少种不同的方案能够使你在n步之后回到$(0,0)$,当且仅当两种方案至少有一步走的方向不同,这两种方案被认为是不同的.
答案可能很大所以只需要输出答案对${10}^9+7$取模后的结果。
这还是太简单了,所以你给能够到达的格点加上了一些限制,一共有三种限制,加上没有限制的情况,一共有四种情况,用$0,1,2,3$标号:
$0.$没有任何限制,可以到达坐标系上所有的点。即能到达的点集为$\{ (x,y)|x,y$为整数$\}$
$1.$只允许到达x轴非负半轴上的点。即能到达的点集为$\{ (x,y)\|x$为非负数$,y=0\}$
$2.$只允许到达坐标轴上的点。即能到达的点集为$\{ (x,y)\|x=0$或$y=0\}$
$3.$只允许到达$x$轴非负半轴上的点,$y$轴非负半轴上的点以及第$1$象限的点。即能到达的点集为$\{ (x,y)\| x\geqslant 0,y\geqslant 0\}$
输入格式
一行两个整数(空格隔开)n和typ,分别表示你必须恰好走的步数和限制的种类。
typ的含义见【题目描述】
输出格式
一行一个整数ans,表示不同的方案数对${10}^9+7$取模后的结果。
样例
样例输入0:
100 0
样例输出0:
383726909
样例输入1:
100 1
样例输出1:
265470434
样例输入2:
100 2
样例输出2:
376611634
样例输入3:
100 3
样例输出3:
627595255
数据范围与提示
$10\%$的数据,$typ=0,n\leqslant 100$
$10\%$的数据,$typ=0,n\leqslant 1,000$
$ 5\%$的数据,$typ=0,n\leqslant 100,000$
$10\%$的数据,$typ=1,n\leqslant 100$
$10\%$的数据,$typ=1,n\leqslant 1,000$
$ 5\%$的数据,$typ=1,n\leqslant 100,000$
$10\%$的数据,$typ=2,n\leqslant 100$
$15\%$的数据,$typ=2,n\leqslant 1,000$
$10\%$的数据,$typ=3,n\leqslant 100$
$10\%$的数据,$typ=3,n\leqslant 1,000$
$ 5\%$的数据,$typ=3,n\leqslant 100,000$
以上$11$部分数据没有交集。
$100%$的数据,保证$n$为偶数,$2\leqslant n\leqslant 100,000$,$0\leqslant typ\leqslant 3$。
题解
一个数学和$DP$的大结合。
首先,这四问都可以用$\Theta (n^3)$的$DP$拿到部分分。
定义$dp[t][i][j]$表示$t$时刻在点$(i,j)$的方案数。
$DP$式子为:
$dp[t][i][j]=dp[t-1][i-1][j]+dp[t-1][i][j-1]+dp[t-1][i+1][j]+dp[t-1][i][j+1]$。
注意边界就好了。
期望得分:$40$分。
$typ=0$的情况,显然让我们想到了visit那道题,25分就轻松拿到了。
$typ=1$的情况,我们可以理解为向右走为进栈,想左走为出栈,那么就是求卡特兰数的第$\frac{n}{2}$项,直接用公式解决就可以了,附带一个卡特兰数的求法。
$typ=2$的情况,注意数据范围$n\leqslant 1,000$,所以我们可以考虑$\Theta (n^2)$的做法,定义$dp[t][i]$表示在$t$时刻位于距离原点为$i$的方案数,现在我们只考虑它在$x$轴上的情况,那么$DP$式子为:$dp[t][i]=dp[t-1][i-1]+dp[t-1][i+1]$,现在放回在坐标系上的情况,那么只有在原点的时候才会有四个方向向它贡献答案,那么$DP$式子只在$i=0$时发生改变,为:$dp[t][0]=4\times dp[t-1][1]$,也可以用滚动数组节省空间。注:正解为卡特兰数,在此介绍另一种做法。
$typ=3$的情况,只是在$typ=1$的情况上加一个枚举横向走的步数$i$纵向的步数即为$n-i$,需要注意的是横向的步数肯定为偶数步,那么方案数就是:$C_n^i\times catalan\frac{i}{2}\times catalan\frac{n-i}{2}$。
代码时刻
#include<bits/stdc++.h>
using namespace std;
int n,typ;
long long inv[200001],jget_C[200001];
long long dp[1001][1001];
long long qpow(long long x,long long y)
{
long long res=1;
while(y)
{
if(y%2)res=(res*x)%1000000007;
y>>=1;
x=(x*x)%1000000007;
}
return res;
}
void pre_work()
{
jget_C[0]=1; jget_C[1]=1;
for(int i=2;i<=n;i++)jget_C[i]=1LL*jget_C[i-1]*i%1000000007;
inv[0]=1; inv[1]=1;
for(int i=2;i<=n;i++)inv[i]=1LL*(1000000007-1000000007/i)*inv[1000000007%i]%1000000007;
for(int i=1;i<=n;i++)inv[i]=1LL*inv[i]*inv[i-1]%1000000007;
}
long long get_C(int n,int m)
{
if(n<m)return 0;
if(m==0)return 1;
return (jget_C[n]%1000000007*inv[m]%1000000007*inv[n-m]%1000000007)%1000000007;
}
long long get_Catalan(int x){return get_C(2*x,x)%1000000007*qpow((long long)(x+1),1000000005)%1000000007;}
int main()
{
scanf("%d%d",&n,&typ);
pre_work();
switch(typ)
{
case 0:
{
long long ans=0;
for(int i=0;i<=n;i+=2)
ans=(ans+get_C(n,i)%1000000007*get_C(i,i/2)%1000000007*get_C(n-i,(n-i)/2)%1000000007)%1000000007;
printf("%lld",ans%1000000007);
break;
}
case 1:
{
printf("%lld",get_Catalan(n/2));
break;
}
case 2:
{
dp[0][0]=1;
for(int i=1;i<=n;i++)
{
dp[i][0]=dp[i-1][1]*4%1000000007;
for(int j=1;j<=n/2;j++)
dp[i][j]=(dp[i-1][j-1]+dp[i-1][j+1])%1000000007;
}
printf("%lld\n",dp[n][0]);
break;
}
case 3:
{
long long ans=0;
for(int i=0;i<=n;i+=2)
ans=(ans+get_C(n,i)%1000000007*get_Catalan(i/2)%1000000007*get_Catalan((n-i)/2)%1000000007)%1000000007;
printf("%lld",ans);
break;
}
}
return 0;
}
rp++