题目描述
牛牛在书上看到一种字符串叫做回文串,当一个字符串从左到右和从右到左读都是一样的,就称这个字符串为回文串。牛牛又从好朋友羊羊那里了解到一种被称为优美的回文串的字符串,考虑一个长度为N只包含大写字母的字符串,写出它所有长度为M的连续子串(包含所有可能的起始位置的子串,相同的子串也要计入),如果这个字符串至少有K个子串都是回文串,我们就叫这个字符串为优美的回文串。现在给出一个N,牛牛希望你能帮他计算出长度为N的字符串有多少个是优美的回文串(每个位置都可以是’A’~'Z’的一个。)
解题思路
这道题一开始没有想到很好的思路能够较小枚举的数量,后来在牛客网的讨论区上看到了一个同学的代码,不得不感叹思路的巧妙。
总结来说,就是将所有可能的情况模式化,比如回文串ABBBA,ACCCA,BCCCB等都满足有两个m=2的回文子串,可以看出其实他们的模式都是01110,而如果我们逐个去枚举的话,数量达到26*25,所以找出满足条件的模式,然后根据模式进行排列组合就会大大的减少数据量。
比如n=3的模式,可以有{0,0,0}(表示三个都完全相同),{0,0,1}(前两个和后一个不同){0,1,0}{1,3相同,2不同},等等,每找到一个满足条件的模式,就加上对应的排列数量(可以根据不同的个数找到对应的排列)。
实现代码
#include <iostream>
#define N 12
using namespace std;
int pattern[N];
int n,m,k;
unsigned long nums[N];
unsigned long ans=0;
bool check(int from){
for(int i=0;i<m/2;i++)
if(pattern[from+i]!=pattern[from+m-i-1])
return false;
return true;
}
bool check(){
int cnt=0;
for(int i=0;i<=n-m;i++)
if(check(i))
cnt++;
return (cnt>=k);
}
void dfs(int pos,int num){
if(pos==n){
if(check())
ans+=nums[num];
return;
}
for(int i=0;i<num;i++){
pattern[pos]=i;
dfs(pos+1,num);
}
pattern[pos]=num;
dfs(pos+1,num+1);
}
int main(){
cin>>n>>m>>k;
nums[0]=1;
for(int i=1;i<N;i++)
nums[i]=nums[i-1]*(27-i);
dfs(0,0);
cout<<ans<<endl;
return 0;
}