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

最新推荐文章于 2022-05-14 17:31:35 发布

_er

最新推荐文章于 2022-05-14 17:31:35 发布

阅读量246

点赞数

本文链接：https://blog.csdn.net/Estia_/article/details/82961976

版权

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

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

#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cctype>
#include<cstring>
#include<iostream>
#include<algorithm>
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;
}