18730 涂色问题

18730 涂色问题

时间限制:1000MS 代码长度限制:10KB
提交次数:0 通过次数:0

题型: 编程题 语言: 不限定
Description
在某大学的农场里，n间牛舍住着n头奶牛。现在你需要为n间牛舍的外墙涂色，有m种可选颜色。
我们已经知道当相邻两间牛舍颜色相同时，奶牛们会集体发疯。
请问有多少种涂色方案会让奶牛们发疯，由于答案可能较大，输出对1000000007求余的结果。

输入格式
仅一行，两个整数n和m，代表牛舍数量和颜色数量。(1<=n<=1e12),(1<=m<=1e12)

输出格式
仅一样，一个整数代表答案。

输入样例
3 2

输出样例
6

提示
3牛舍2颜色方案有(1,1,1),（1,1,2）,（1,2,2）,（2,1,1）,（2,2,1),(2,2,2)，共6种方案会让奶牛发疯。

分析：

本题有两个考点，一个是反向思维，发疯的方案个数不好求就求总数减掉不发疯的为m的n次方-m*（m-1）的（n-1）次方，其次是考察公式 C=(A%mod-B%mod+mod)%mod,如果不使用该公式可能在某一步会导致long long类型超出而使得结果与预期结果有偏差。

代码：

#include <iostream>

using namespace std;

typedef long long ll; //数据量大 需要用long long类型 
#define mod 1000000007

ll qmi(ll x,ll y)
{
	ll sum=0;
	if(y==0)
	return 1;
	ll t=qmi(x,y/2);
	sum=t*t%mod;
	if(y%2==1)
	sum=sum*x%mod;
	return sum;
} 

int main(void)
{
	ll n,m;
	cin>>n>>m;
	cout<<(qmi(m,n)%mod-m*qmi(m-1,n-1)%mod+mod)%mod; 
	return 0;
}