题目描述
蒜头君有 n 只气球,蒜头君把气球排成一排。初始时,气球都是白色,现在蒜头君想用 m 种颜色给气球涂色,如果相邻的气球的颜色相同,这 2 个气球会发生消消乐,蒜头君希望你求出会发生消消乐的涂色方法有多少种。最后答案对 10^9+7取模。
输入格式
输入两个整数n(1≤n≤10^12 ),m(1≤m≤10^8 )
输出格式
输出一行表示答案。
样例输入
2 2(3 4)
样例输出
2(28)
思路
总的涂法:d = m^n;
不发生消消乐的:c = m*(m-1)^(n-1),第一个有m种涂法,后面的每个是m-1种涂法。
所以发生消消乐的 d-c
要对负数进行处理。
AC代码
#include <iostream>
using namespace std;
long long int mod = 1e9+7;
long long int pow_mod(long long int a,long long int b,long long int mod){
if(b==0){
return 1%mod;
}
long long int temp = pow_mod(a,b/2,mod);
temp = temp*temp%mod;
if(b%2==1){
temp = temp*a%mod;
}
return temp;
}
int main(){
long long int n,m;
cin>>n>>m;
long long int d = pow_mod(m,n,mod);
long long int c = pow_mod(m-1,n-1,mod)*m;
cout<<((d-c)%mod+mod)%mod;
return 0;
}