M * N的方格,一个机器人从左上走到右下,只能向右或向下走。有多少种不同的走法?由于方法数量可能很大,只需要输出Mod 10^9 + 7的结果。
Input
第1行,2个数M,N,中间用空格隔开。(2 <= m,n <= 1000000)
Output
输出走法的数量 Mod 10^9 + 7。
Input示例
2 3
Output示例
3
思路:
从左上走到右下,一共需要走M-1 + N - 1步,需要向右走N-1步,因此所有的可能是C(M-1+N-1, N-1)。C(n, m) = n!/(m!(n-m)!),但由于数量太大,需要用到除法取模。(a/b) mod p = (a mod p) * (b的逆元 mod p)。 求逆元可根据费曼小定理:
假如p是质数,且gcd(a,p)=1,那么 a^(p-1)≡1(mod p),即:假如a是整数,p是质数,且a,p互质(即两者只有一个公约数1),那么a的(p-1)次方除以p的余数恒等于1。
即a*a^(p-2)≡1(mod p), a的p-2次方是a的逆元。
#include <iostream>
#include <cstring>
#include <algorithm>
#include <stdio.h>
using namespace std;
typedef long long int ll;
const ll MOD = 1e9 + 7;
ll pow(ll a, ll i)
{
ll result = 1;
while (i)
{
if (i & 1)
{
result = (result * a) % MOD;
}
i >>= 1;
a = (a * a) % MOD;
}
return result;
}
ll factorial(ll n)
{
ll result = 1;
for (ll i = 1; i <= n; i++)
{
result = (result * i) % MOD;
}
return result;
}
ll C(ll n,ll m)
{
ll result = 1;
result = (result * factorial(n)) % MOD;
result = result * pow(factorial(m), MOD-2) % MOD;
result = result * pow(factorial(n-m), MOD-2) % MOD;
return result;
}
int main()
{
int m, n;
cin >> m >> n;
cout << C(m-1+n-1, n-1) << endl;
return 0;
}