气球消消乐

题目描述

蒜头君有 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;
}