XDU-1032 找规律II （Lucas定理）

1032: 找规律II

时间限制: 1 Sec 内存限制: 128 MB

http://acm.xidian.edu.cn/problem.php?id=1032

[提交][状态][讨论版]

题目描述

现有数阵如下：



求这个数阵的第n行m列是多少（行列标号从1开始）
结果对10007取模

输入

多组数据，每组数据一行，包含两个整数n,m（1<=n<=m<=10^18）

输出

每组数据输出一行，为数阵中第n行m列对10007取模后的值。

样例输入

1 1
1 2
1 3

样例输出

1
2
3

很容易就能看出是结果组合数，所以直接套用Lucas模版即可

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

typedef long long LL;

const LL MOD=10007;

LL n,m;

LL quick_pow(LL a, LL b) {
LL ans = 1;
a %= MOD;//防止指数为0时，返回未取模的结果
while(b!=0) {
if((b & 1)==1) {
ans = ans * a % MOD;
}
b >>= 1;
a = a * a % MOD;
}
return ans;
}

LL C(LL n, LL m) {
if(n < m) {
return 0;
}
if(n == m) {
return 1;
}
if(m > n - m) {//利用组合数性质，减少运算次数
m=n - m;
}
LL ans, a = 1, b = 1;
for(int i=0; i<m; ++i) {
a = (a * (n - i)) % MOD;
b = (b * (m - i)) % MOD;
}
ans = (a * quick_pow(b, MOD-2)) % MOD;
return ans;
}

LL Lucas(LL n, LL m) {
LL ans=1;
while(m>0) {
ans=(ans*C(n % MOD, m % MOD)) % MOD;
n/=MOD;
m/=MOD;
}
return ans;
}

int main() {
LL n,m;
while(2==scanf("%lld%lld",&n,&m)) {
printf("%lld\n",Lucas(m,m-n));
}
return 0;
}