[HNOI2008]越狱

Description

监狱有连续编号为1...N的N个房间,每个房间关押一个犯人,有M种宗教,每个犯人可能信仰其中一种。如果相邻房间的犯人的宗教相同,就可能发生越狱,求有多少种状态可能发生越狱。

Sample Input

2 3

Sample Output

6

直接计算并不好做,我们从反面来考虑。不可能发生越狱的状态数怎么求呢?我们只要从前到后,依次保证每个房间的宗教都与之前的一个不同即可,总状态数为:
M(M-1)(M-1)(M-1)…(M-1)=M(M-1)^N-1

#include
#include
using namespace std;
typedef long long LL;
const LL C=LL(100003);
LL q_mod(LL a,LL b)
{
    LL ans=1%C;a%=C;
    while(b)
    {
        if(b%2==1)ans=ans*a%C;
        b=b>>1;a=a*a%C;
    }
    return ans;
}
int main()
{
    LL n,m;
    scanf("%lld%lld",&n,&m);
    LL s1=q_mod(n,m);
    LL s2=n*q_mod(n-1,m-1)%C;
    LL ss=(s1-s2+C)%C;
    printf("%lld\n",ss);
    return 0;
}

你可能感兴趣的:(数论,xgc的做题记录)