BZOJ[1008]越狱

1008: [HNOI2008]越狱
Time Limit: 1 Sec Memory Limit: 162 MB

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

这道题可以看成有n个盘子，m种芒果。（我喜欢吃芒果）
每个盘子放一个芒果，相邻的盘子芒果种类一样，求方案数。如果我们直接暴力的话，数据这么大，会TLE的。
所以我们可以想:第1个盘子可以放m种芒果，第2个盘子可以放m-1种芒果，第3个盘子仍然可以放m-1芒果。
我们就得出了num=m * (m-1)^(n-1) (这是合法的数量)。
sum=m^n ans=sum-num (这是总的数量)

但如果直接这样乘，也会TLE，所以我们就要用到快速幂了。

请欣赏代码：（注意快速幂部分）

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<ctime>
#include<set>
#include<map>
#include<sstream>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long LL;
LL q_mod(LL a,LL b,LL c)    //快速幂函数 
{
    LL ans=1%c;
    a=a%c;
    while(b) 
    {
        if(b%2==1) ans=ans*a%c;
        b/=2; a=a*a%c;
    }
    return ans;
}      //快速幂函数 
int main()
{
    LL n,m,ss=0;
    scanf("%lld%lld",&m,&n);
    ss=(q_mod(m,n,LL(100003))-m*q_mod(m-1,n-1,LL(100003))%LL(100003)+LL(100003))%LL(100003);
    printf("%lld\n",ss);
    return 0;
}

觉得好就给个赞吧~~~

版权声明：本文为博主原创文章，未经博主允许不得转载。