链接:https://ac.nowcoder.com/acm/contest/7412/A
来源:牛客网
牛牛和牛可乐的赌约
时间限制:C/C++ 2秒,其他语言4秒
空间限制:C/C++ 262144K,其他语言524288K
64bit IO Format: %lld
题目描述
牛可乐发明了一种n面骰子(点数分别从 1 到 n,掷出每面的概率为
1
n
\frac {1} {n}
n1 去给牛牛玩,因为牛牛是个欧皇,所以他想测试一下牛牛的人品,他告诉牛牛,让牛牛投
m
m{}
m次骰子,牛牛如果全部投出点数为
n
{}n
n的面就算牛牛赢,牛牛很相信自己的人品,就和牛可乐赌一包辣条,说自己肯定可以全部投出点数为
n
{}n
n点面,但是牛牛又有点害怕自己打赌输了,想让你提前帮他计算一下他输概率有多少?
输入描述:
有多组输入样例,第一行为样例组数
t
t
t
(
t
≤
1
×
1
0
6
)
(t\leq 1×10^6)
(t≤1×106)
接下来
t
t
t 行每行有一个整数 n 和 m ,分别表示骰子的面数和牛牛的投掷次数
(
n
,
m
<
=
1
×
1
0
9
)
(n,m<=1×10^9)
(n,m<=1×109)
输出描述:
输出t行,每行输出为分数 p/q mod 1e9+7的形式
示例1
输入
1
2 1
输出
500000004
备注:
数据较大,建议使用较快的输入输出
思路:就是求
1
−
1
n
m
1 -\frac{1} {n^{m}}
1−nm1 =
(
n
m
−
1
)
∗
1
n
m
(n^m-1)*\frac{1}{n^{m}}
(nm−1)∗nm1 =
(
n
m
−
1
)
∗
i
n
v
(
n
m
)
(n^m-1)*inv(n^{m})
(nm−1)∗inv(nm)
inv(a)指a的逆元
用两次快速幂,一次求
n
m
n^m
nm一次求inv(
n
m
n^m
nm)
用到费马小定理
p是质数,且 gcd(p,a) = 1, 1 = a p − 1 ( m o d p ) 1 = a^{p-1} (\bmod p) 1=ap−1(modp)
由这个定理可以推出 inv(a)*a =
1
=
a
p
−
1
(
m
o
d
p
)
1 = a^{p-1} (\bmod p)
1=ap−1(modp)
所以inv(a) =
a
p
−
2
(
m
o
d
p
)
a^{p-2} (\bmod p)
ap−2(modp),把
n
m
n^m
nm带入就行
AC代码:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<string>
#include<stack>
#include<queue>
#include<set>
#include<map>
#include<time.h>
#include<vector>
#define ll long long
using namespace std;
inline void read(ll &x)
{
char c;
x = 1;
while((c=getchar())<'0'||c>'9') if(c=='-')x=-1;
ll res = c-'0';
while((c=getchar())>='0'&&c<='9') res = res*10+c-'0';
x *= res;
}
inline ll qpow(ll a,ll n,ll p)
{
ll ans = 1;
while(n>0){
if(n & 1){
ans = ans % p * a % p;
}
a = a % p * a % p;
n >>= 1;
}
return ans;
}
inline ll inv(ll a,ll p)
{
return qpow(a,p-2,p);
}
int main()
{
const ll p = 1e9+7;
ll t,n,m;
read(t);
while(t--){
read(n);
read(m);
ll nm = qpow(n,m,p);
ll ans = (nm-1) % p * inv(nm,p) % p;
printf("%lld\n",ans);
}
return 0;
}