快速幂(排列组合) - 越狱 - AcWing 1290

快速幂(排列组合) - 越狱 - AcWing 1290

监狱有连续编号为 1 到 n 的 n 个房间，每个房间关押一个犯人。

有 m 种宗教，每个犯人可能信仰其中一种。

如果相邻房间的犯人信仰的宗教相同，就可能发生越狱。

求有多少种状态可能发生越狱。

输入格式

共一行，包含两个整数 m 和 n。

输出格式

可能越狱的状态数，对 100003 取余。

数据范围

$1\le m\le 10^8,1\le n\le 10^{12}$

输入样例：

2 3

输出样例：

6

样例解释

所有可能的 6 种状态为：(000)(001)(011)(100)(110)(111)。

---

分析：

$正难则反，我们反过来考虑。$

$每个房间都有m种情况，根据乘法原理，方案总数是m^n。$

$不存在两个相邻的方案总数是：$

$第一间有m种情况，第二件有m-1种情况，第三件也有m-1种情况，...，方案总数是m\times (m-1{)}^{(n-1)}。$

$最终结果：m^n-m\times (m-1{)}^{(n-1)}，注意取模防止出现负数。$

代码：

#include<iostream>

#define ll long long

using namespace std;

const int mod=100003;

ll quick_pow(ll a,ll b,int mod)
{
    ll res=1;
    while(b)
    {
        if(b&1) res=res*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return res;
}

int main()
{
    ll n,m;
    cin>>m>>n;
    ll ans=quick_pow(m,n,mod)-m*quick_pow(m-1,n-1,mod)%mod;
    cout<<(ans+mod)%mod<<endl;
    
    return 0;
}