Description
监狱有连续编号为1...N的N个房间,每个房间关押一个犯人,有M种宗教,每个犯人可能信仰其中一种。如果相邻房间的犯人的宗教相同,就可能发生越狱,求有多少种状态可能发生越狱
Input
输入两个整数M,N.1<=M<=10^8,1<=N<=10^12
Output
可能越狱的状态数,模100003取余
Sample Input
2 3
Sample Output
6
HINT
6种状态为(000)(001)(011)(100)(110)(111)
Source
题解:(终于有拿手的题目了)
首先,范围还是比较大的,一开始想到动规(但显然是超时的):
an表示M,N的解,那么an=M*an-1+(M^(n-1)-an-1)
如果之前n-1的时候已经可以了,那么接下来那一格可以随便放,那么乘以M,然后剩下不可以的只需要在新增的一格放上配对的颜色(宗教)就可以了。
接下来就是如何改进了,两边同减去M^n,然后就发现等比数列了,于是快速幂解之。aN=M^N-M*(M-1)^(n-1)
#include<cstdio>
#include<iostream>
using namespace std;
long long m,n;
long long _pow(long long a,long long b){
long long base=a%100003,ans=1;
while(b>0){
if(b&1){
ans*=base;
ans%=100003;
}
base*=base;
base%=100003;
b>>=1;
}
return ans;
}
int main()
{
cin>>m>>n;
long long ans=_pow(m,n)-(_pow(m-1,n-1)*(m%100003));
while(ans<0)ans+=100003;
cout<<ans;
return 0;
}