M * N的方格,一个机器人从左上走到右下,只能向右或向下走。有多少种不同的走法?由于方法数量可能很大,只需要输出Mod 10^9 + 7的结果。
收起
输入
第1行,2个数M,N,中间用空格隔开。(2 <= m,n <= 1000000)
输出
输出走法的数量 Mod 10^9 + 7。
输入样例
2 3
输出样例
3
第一次本着试试的想法写了个递归。
结果TLE。。。
看题解说是杨辉三角。一脸懵逼。
这里附上杨辉三角的重要性质链接:
说是求C(m+n-2)(m-1)%mod;
这里n>m;
先设N=m+n-2,M=m-1;
然后可以转化为N!/(N-M)!*M!
令a=N!%mod/(N-M)!%mod;
b=M!;
我们可以先求出a的值。
a/b则可以转化为求a*b';
这里b'为b的逆元。
b'可以通过扩展欧几里得求出。
然后求出a*b'%mod即可。
代码如下:
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
using namespace std;
typedef long long ll;
ll n,m;
ll N,M;
const ll Mod=1e9+7;
ll Cnm (ll nn,ll mm)
{
ll sum=1;
for (ll i=nn-mm+1;i<=nn;i++)
{
sum=sum*i%Mod;
}
return sum;
}
void ex_gcd (ll a,ll b,ll &x,ll &y)
{
if(b==0)
{
x=1;y=0;
return ;
}
else
{
ll temp;
ex_gcd(b,a%b,x,y);
temp=x;
x=y;
y=temp-a/b*y;
}
}
int main()
{
scanf("%lld%lld",&m,&n);
if(m>n) swap(m,n);
N=m+n-2;
M=m-1;
ll a=Cnm(N,M);
ll b=1;
for (ll i=2;i<=M;i++)
{
b=b*i%Mod;
}
ll x,y;
ex_gcd (b,Mod,x,y);
x=(x%Mod+Mod)%Mod;
printf("%lld\n",x*a%Mod);
return 0;
}