[LGR053B] Scarlet的字符串不可能这么可爱 [结论]

题意：https://www.luogu.org/problemnew/show/T48749

开始还在想是不是要矩阵快速幂
打月赛的时候没特判好
简单的结论，推一推就出来了
显然不能有$AA$或者$ABA$ 。放在一起考虑就是每个位置都得是$CBA$这样的。
$s$的限制就是去掉了一个$k$的选择，答案全部相比$s=0$要除掉一个$k$

#include
#include
#include
#include
#include
#include
#include
using namespace std;
#define ll long long
ll k,L,p,s,w,Ann;
ll qpow(ll base,ll cnt)
{
    ll ans=1;
    while(cnt)
    {
        if(cnt&1)
        {
            ans*=base;
            ans%=p;
        }
        cnt>>=1;
        base*=base;
        base%=p;
    }
    return ans;
}
int main()
{
    scanf("%lld%lld%lld%lld%lld",&k,&L,&p,&s,&w); k%=p;
    Ann=qpow(k-2,L-2);
    if(!s)
    {
    	if(L==1)printf("%lld",k);
    	else if(L==2)printf("%lld",k*(k-1)%p);
    	else printf("%lld",(k-1)*k%p*Ann%p);
    }
    else
    {
    	if(L==1)printf("1");
    	else if(L==2)printf("%lld",k-1);
    	else printf("%lld",(k-1)*Ann%p);
    }
    return 0;
}