51Nod 1119 机器人走方格 (扩展欧几里得+逆元+求组合数)

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;
}

猜你喜欢

转载自blog.csdn.net/qq_41410799/article/details/83780986