蓝桥杯2023年第十四届省赛真题-子串简写题解

小菜niao第一次写题解，希望各位dalao勿喷。

大家一开始的思路如果是直接暴力两层for循环，但由于此题的数据范围较大，如果不进行优化必定会超时！

解题思路：

虽然我的方法也是两层for循环，但是我在两处地方进行了优化，最后也是AC了。

如何优化请看如下思路：

1、用数组a1，a2分别存储字符c1，c2出现在字符串s的下标。a1，a2的下标表示字符c1，c2出现的顺序，如a1[i]=j，表示第 i 个c1字符出现在字符串s的下标位置为 j。

2、通过两层for循环，遍历数组a1、a2，当出现 a1[i]<=a2[j]-k+1这个情况，,则表示c2与c1之间的长度大于等于k，即出现了符合的情况，并且是第一次出现符合的情况。此时需要注意【优化的地方来了】：由于 a[j] 后面的c2字符的下标（a[j]…a[k2]）肯定比此时的大，即与此时c1字符之间的长度肯定也比k大，所以后面的 j 就不需要再进行遍历了，直接通过计算便可知道此时有几个 c2字符是符合条件的。

3、【优化第二处】：通过一个 辅助变量index记录每次遍历 i 时，第一次出现满足 a1[i]<=a2[j]-k+1这个情况的 j 的值，于是在下一次重新遍历 a2[j] 时，就可以直接让 j 一开始就等于index。

原因：因为当出现了 符号情况的 a2[j] 后，接下来的a1[i]会增大，意味着 下一个字符c1出现的位置会变大，但由于 a[j] 之前 位置已经知道是不符合情况的了，所以就不再需要遍历，直接从上一次符合情况的 位置开始遍历。

注意：记得类型是long long

如果思路看的还是有点不懂的话，那就看看代码吧！

#include<bits/stdc++.h>
 using namespace std;
 #define ll long long
 ​
 ll k;
 string s;
 char c1,c2;
 ll ans;
 //记录c1,c2出现在字符串s的下标，我是从1位置开始存储的
 ll a1[500005],a2[500005];
 ​
 int main() 
 {
     cin>>k;
     cin>>s>>c1>>c2;
     
     ll l=s.length();//字符串s的长度   
     
     ll k1=0,k2=0;
     for(int i=0;i<l;i++)
     {   
         if(s[i]==c1)//该字符为c1，则加到a1数组中 
         {
             k1++;
             a1[k1]=i;
         }
         if(s[i]==c2)//该字符为c2，则加到a2数组中 
         {
             k2++;
             a2[k2]=i;
         }
     }
     
     //上面将字符串中所有c1,c2的位置记录到了数组中，
     //k1,k2也代表c1,c2的个数
     
     ll index=1;//因为j是从1开始，所以初始为1
     for(int i=1;i<=k1;i++)//遍历a1数组的下标
     {
         //注意：j从index开始
         for(int j=index;j<=k2;j++)//遍历a2数组的下标
         {
             if(a1[i]<=a2[j]-k+1){
                 ans+=k2-j+1;//计算此时有几个符合情况的c2字符
                 index=j;//更新index
                 break;//后面的不需要再遍历，所以直接break第二层循环
             }
         }
     }
     
     cout<<ans;
     
     return 0;
 }

 ​