题目链接:https://www.51nod.com/onlineJudge/questionCode.html#!problemId=1119
基准时间限制:1 秒 空间限制:131072 KB 分值: 10
难度:2级算法题
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
解析:求C(m+n-2, n - 1), 运用Lucas定理求
快速幂:
#include<bits/stdc++.h>
#define N 10009
using namespace std;
typedef long long LL;
LL q_mod(LL a, LL b, LL p)
{
LL ans = 1;
while(b)
{
if(b & 1) ans = ans * a % p;
b >>= 1;
a = a * a % p;
}
return ans % p;
}
LL c_mn(LL n, LL m, LL p)
{
if(n < m) return 0;
if(m == n) return 1;
LL ca, cb;
ca = cb = 1;
for(int i = 0; i < m; i++)
{
ca = ca * (n - i) % p;
cb = cb * (m - i) % p;
}
// ca / cb
return ca * q_mod(cb, p - 2, p) % p;
}
LL Lucas(int n, int m, int p)
{
LL ans = 1;
while(m && n && ans)
{
ans = ans * c_mn(n % p, m % p, p) % p;
n /= p;
m /= p;
}
return ans;
}
int main()
{
int m, n;
int p = 1e9 + 7;
cin >> m >> n;
LL ans = Lucas(m + n - 2, n - 1, p);
cout << ans << endl;
return 0;
}
扩展欧几里得:
#include<bits/stdc++.h>
#define N 10009
using namespace std;
typedef long long LL;
void e_gcd(LL a, LL b, LL &x, LL &y)
{
if(b == 0)
{
x = 1; y = 0;
return ;
}
e_gcd(b, a % b, x, y);
LL t = x; x = y;
y = t - a / b * y;
}
LL c_mn(LL n, LL m, LL p)
{
if(n < m) return 0;
if(m == n) return 1;
LL ca, cb, x, y;
ca = cb = 1;
for(int i = 0; i < m; i++)
{
ca = ca * (n - i) % p;
cb = cb * (m - i) % p;
}
// ca / cb
e_gcd(cb, p, x, y);
x = ((x % p) + p) % p;
return ca * x % p;
}
LL Lucas(int n, int m, int p)
{
LL ans = 1;
while(m && n && ans)
{
ans = ans * c_mn(n % p, m % p, p) % p;
n /= p;
m /= p;
}
return ans;
}
int main()
{
int m, n;
int p = 1e9 + 7;
cin >> m >> n;
LL ans = Lucas(m + n - 2, n - 1, p);
cout << ans << endl;
return 0;
}