思路一:暴力做法(能过80%)
从前向后依次遍历当找到一个以字符c1开头的字符,有多少c2能与之组成合法的子字符串,然后累加起来;
C++代码:
//暴力做法,能过80%
#include <iostream>
using namespace std;
typedef long long ll;
const int N=5e5+10;
int k;
char vis[N];
ll ans;
int a[N],b[N];
int main()
{
// 请在此输入您的代码
cin>>k;
string s;
cin>>s;
char c1,c2;
cin>>c1>>c2;
int a1=0,b1=0;
for(int i=0;i<s.size();i++){
if(s[i]==c1) a[++a1]=i+1;
if(s[i]==c2) b[++b1]=i+1;
}
for(int i=1;i<=a1;i++){
for(int j=1;j<=b1;j++){
if(b[j]-a[i]+1>=k)
ans++;
}
}
cout<<ans;
// for(int i=1;i<=cut;i++){
// cout<<vis[i]<<' ';
// }
return 0;
}
思路二:将每一个与c2相同的字符前面与它相隔距离大于等于k的所有与c1相同的字符数加起来(我语文表达能力不太行,如果没看懂多看几遍应该可以理解)
C++代码:
//ACcode
#include <iostream>
using namespace std;
typedef long long ll;
const int N=5e5+10;
ll k;
ll tmp,ans;
int main(){
cin>>k;
string s;
char c1,c2;
cin>>s>>c1>>c2;
for(int i=0,j=k-1;i<s.size();i++,j++){
if(s[i]==c1) tmp++;//统计某一个c2前面有多少比它小k的c1
if(s[j]==c2) ans+=tmp;//将每一个c2前面的c1的个数累加起来,即为答案
}
cout<<ans;
return 0;
}