题目描述
监狱有连续编号为 1…N 的 N 个房间,每个房间关押一个犯人,有 M 种宗教,每个犯人可能信仰其中一种。如果相邻房间的犯人的宗教相同,就可能发生越狱,求有多少种状态可能发生越狱。
思路
首先考虑到这是一道数学题,根据乘法原理可以知道如果不考虑相邻的情况,即有多少种关押情况,情况总数为 S1 = M^N.
而考虑不相邻的情况总数为 S2 = M*M^(N-1)种,答案 ans = S1-S2.
由于题目数据比较庞大。一般的求幂运算的时间过长,所以需要快速幂。
代码:
#include <iostream>
#include <cstdio>
using namespace std;
long long int M,N;
int Mod = 100003;
long long int Power(long long int,long long int);
int main(void)
{
cin >> M >> N;
printf("%lld",(Power(M,N)-M*Power(M-1,N-1)%Mod+Mod)%Mod); //防止负数的出现
return 0;
}
long long int Power(long long int a,long long int b)
{
long long int ans = 1;
long long int base = a;
while(b)
{
if(b&1)
{
ans = (ans*base)%Mod;
}
base = (base*base)%Mod;
b >>= 1;
}
return ans;
}