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

原创于 2024-04-02 19:29:25 发布 · 208 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#蓝桥杯 #职场和发展

蓝桥杯刷题专栏收录该内容

21 篇文章

订阅专栏

本文介绍了一种在给定字符串c中，分析a...b子串出现情况并计算a后面紧跟着b的计数的C++编程方法，涉及前缀和的计算和字符串操作。

分析

找到所有长度>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;
}