十四届蓝桥杯CB 子串的简写

分析

找到所有长度>k的a.....b串：

统计所有b的个数，再遍历字符串，遇到a，就加上其后b的数量

1.设置一个int[]与string s对应，遇到s[i]=b，int[i] = 1

2.求出前缀和

3.遍历，遇到a，用sum(b数量总和)-a前b的数量

代码

#include<iostream>
#include<cstring>
using namespace std;

typedef long long ll;

const int N = 500010;
int a[N],s[N];
string c;

int main()
{
	int k;
	char c1,c2;
	cin >> k;
	cin >> c >> c1 >> c2;
	
	int sum = 0;
	for(int i = 0;i < c.size();i ++)
	{
		if(c[i] == c2)
		{
			a[i] = 1;
			sum += 1;
		}
	}
	
	s[0] = a[0];
	for(int i = 1;i < c.size();i ++)
	{
		s[i] = s[i-1]+a[i];
	}
	
	ll cnt = 0;
	for(int i = 0;i+k-1 < c.size();i ++)
	{
		if(c[i] == c1)
		{
			cnt += (sum - s[i+k-2]);
		}
	}
	cout << cnt << endl;
	return 0;
}