题目描述:
监狱有连续编号为1...N的N个房间,每个房间关押一个犯人,有M种宗教,每个犯人可能信仰其中一种。如果相邻房间的犯人的宗教相同,就可能发生越狱,求有多少种状态可能发生越狱
大致思路:
这个题目一眼下去好像是动态规划,可以通过n-1,m-1转移过来,不过换种想法,其实就是一个组合计数,用所有的方案减去不会越狱的方案,就是越狱的状态数。那么这样问题就简单许多了,变成了一个快速幂的问题了。
代码:
#include <cstdio>
#include <iostream>
using namespace std;
const long long M = 100003;
long long n,m;
long long f(long long a, long long b) {
if (b == 1) return a%M;
else if (b == 0) return 1;
long long ans = f(a,b/2)%M;
if (b%2) return (long long)ans*ans%M*a%M;
else return (long long)ans*ans%M;
}
int main() {
while (cin>>m>>n) {
long long ans = f(m%M,n);
long long mul = f((m-1)%M,n-1)*m%M;
ans = ans - mul;
if (ans<0) ans += M;
cout<<ans<<endl;
}
}