原题链接
题意:
要求用m种字母构造一个长度为n的字符串,使得该字符串的所有长度为k的子串均为回文串,求方案数。
思路:
分情况讨论
case 1 : k = 1, 方案数为 m^n
case 2: k = n 且 n 为奇数, 方案数为m^(n/2 + 1)
case 3 : k = n 且 n 为偶数, 方案数为 m^(n/2)
case 4 : k < n 且 k为奇数, 方案数为m*m
case 5 : k < n 且 k为偶数,方案数为m
case 6(坑点) : k > n,方案数为m^n
理解不了case 6
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 1e7 + 10;
typedef long long ll;
ll n, m, k, mod = 1e9 + 7;
int main()
{
cin >> n >> m >> k;
ll res = 1;
if(k == 1 || k > n) {
for(int i = 1;i <= n; ++ i) {
res *= m;
res %= mod;
}
} else if(k == n) {
for(int i = 1;i <= n/2; ++ i) {
res *= m;
res %= mod;
}
if(n&1) res *= m, res %= mod;
} else {
if(k&1) {
res = m * m % mod;
} else {
res = m;
}
}
printf("%lld\n", res);
return 0;
}