[洛谷P4925][1007]Scarlet的字符串不可能这么可爱

题目大意:问字符集大小为$k$,长度为$L$的字符串,且没有长度超过$1$的回文段的个数。规定第$s(若为0则无限制)$位为$w$。

题解:懒得写了,根据是否有限制分类讨论

卡点:中途有个地方忘记取模

C++ Code:

#include <cstdio>
long long k, L, mod, s;
inline long long pw(long long base, long long p) {
	base %= mod;
	long long res = 1;
	for (; p; p >>= 1, base = base * base % mod) if (p & 1) res = res * base % mod;
	return res;
}
int main() {
	scanf("%lld%lld%lld%lld%*d", &k, &L, &mod, &s);
	if (s) {
		if (L == 1) puts("1");
		else if (L == 2) printf("%lld
", (k - 1) % mod);
		else printf("%lld
", (k - 1) % mod * pw(k - 2, L - 2) % mod);
	} else {
		if (L == 1) printf("%lld
", k % mod);
		else if (L == 2) printf("%lld
", k % mod * (k - 1) % mod);
		else printf("%lld
", k % mod * ((k - 1) % mod) % mod * pw(k - 2, L - 2) % mod);
	}
	return 0;
}

  

原文地址:https://www.cnblogs.com/Memory-of-winter/p/9755564.html