越狱(组合数+快速幂)
监狱有连续编号为 1 到 n 的 n 个房间,每个房间关押一个犯人。有 m 种宗教,每个犯人可能信仰其中一种。如果相邻房间的犯人信仰的宗教相同,就可能发生越狱。求有多少种状态可能发生越狱。
输入格式
输入两个整数 m 和 n。
输出格式
可能越狱的状态数,对 100003 取余。
样例
Input
2 3
Output
6
所有可能的 6 种状态为:{0,0,0},{0,0,1},{0,1,1},{1,0,0},{1,1,0},{1,1,1}
数据范围与提示
对于全部数据,1≤m≤108,1≤n≤1012。
思路: 求有多少种情况可以越狱,要考虑两个及其以上宗教信仰相同,宗教信仰又有好几种,相同的位置又有很多,不好求,那我们就可以求不符合的情况用总共的情况减去即可,总共的情况是每一个位置都可以有m种宗教信仰,即mn, 不符合的情况就必须保证相邻位置的宗教信仰不同,那第一个位置就有m种选择,第二个位置有m-1种选择,第三个位置同有m-1种选择(除去和第二个位置相同的宗教),同理第n个位置也有m-1种选择,则不符合的情况有m * (m-1)(n-1) 种, 符合的就有mn-m * (m-1)(n-1)种,但数据较大要取模,为了防止总情况取模后的值比不符合取模后的值小,我们要在总情况取模后加上100003,防止答案为负。
取模常用公式:
(a + b)% mod = (a %mod + (b % mod)) % mod;
(a - b)% mod = (a %mod - (b % mod)) % mod;
(a * b)% mod = (a %mod * (b % mod)) % mod;
AC代码:
#include<stdio.h>
#define ll long long//数据太大要开long long
#include<string.h>
#include<algorithm>
#define mod 100003
using namespace std;
ll ksm(ll a,ll b)//快速幂
{
ll ans=1;
a=a%mod;
while(b)
{
if(b%2)
ans=ans*a%mod;
a=a*a%mod;
b/=2;
}
return ans;
}
int main()
{
ll n,m,i,j,k;
scanf("%lld%lld",&m,&n);
k=(ksm(m,n)+mod-(m%mod*(ksm(m-1,n-1)%mod)%mod))%mod;
//printf("%d+++\n",ksm(m-1,n-1));
printf("%lld\n",k);
return 0;
}