【纪中oj】B组 n染色

Description
WYF画了一个极为不规则的n边形，画面太美简直不看，没有任意两条边长度是相等的。因为形状太难看了，做他同桌的CWQ看不下去了，趁着WYF上厕所的时间准备用他书包里的m种颜色的彩笔给n边形的边上色。但由于WYF画的实在太大，CWQ不知如何下手，他想知道他有多少种染色方法，能够使得每两条相邻边不同色。你只需输出答案模10^9+7的结果。
Input
一行，仅包含两个正整数n和m。
Output
一个正整数，表示答案模10^9+7的结果
Sample Input
3 3
Sample Output
6
Data Constraint
对于50%的数据，3≤n≤100000，3≤m≤10；
对于100%的数据，3≤n≤10^18，3≤m≤50。

找规律
可以先写一个暴力的程序，打表找规律（有时不得不承认打表是个好东西）

方案数=(m-1)^{n+(m-1)×(-1)}n。（快速幂）
这题DP只能拿50分

#include<iostream>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
using namespace std;
unsigned long long mod=1000000007;
unsigned long long sum=0,b;
unsigned long long f(unsigned long long x){//快速幂
    if(x==0) return 1;
    unsigned long long ans=f(x/2)%mod;
    ans=ans*ans%mod;
    if(x%2==1) ans=ans*b%mod;
    return ans;
}
int main(){
    //freopen("color.in","r",stdin);
    //freopen("color.out","w",stdout);
    unsigned long long n,p;
    cin>>p>>b;
    if(p==1){//特判一下
        cout<<b;
        return 0;
    }
    b--;
    sum+=f(p);
    if(p%2==1) cout<<(sum-b)%1000000007;
    else cout<<(sum+b)%1000000007;
}

注：很早以前上学期 的纪中老题