第十四届蓝桥杯 C++组省B 题目子串简写

原创已于 2024-04-20 19:01:27 修改 · 554 阅读

10 ·

本内容遵循CC 4.0 BY-SA版权协议

GEO检测

于 2023-12-08 20:26:16 首次发布

文章讨论了一种在程序猿圈子流行的字符串简写方法，通过统计给定字符串中以特定字符c1开头和c2结尾且长度大于等于K的子串数量。提供了两种解题思路：前缀和数组维护和双指针技巧。

题干

时间限制: 1s 内存限制: 256MB 提交: 4968 解决: 1119

题目描述

程序猿圈子里正在流行一种很新的简写方法：对于一个字符串，只保留首尾字符，将首尾字符之间的所有字符用这部分的长度代替。例如 internation-alization 简写成 i18n，Kubernetes （注意连字符不是字符串的一部分）简写成 K8s, Lanqiao 简写成 L5o 等。

在本题中，我们规定长度大于等于 K 的字符串都可以采用这种简写方法（长度小于 K 的字符串不配使用这种简写）。

给定一个字符串 S 和两个字符 c1 和 c2，请你计算 S 有多少个以 c1 开头c2 结尾的子串可以采用这种简写？

输入

第一行包含一个整数 K。

第二行包含一个字符串 S 和两个字符 c1 和 c2。

输出

一个整数代表答案。

样例输入

4

abababdb

a

b

样例输出

6

提示

符合条件的子串如下所示，中括号内是该子串：

[abab]abdb

[ababab]db

[abababdb]

ab[abab]db

ab[ababdb]

abab[abdb]

对于 20% 的数据，2 ≤ K ≤ |S | ≤ 10000。

对于 100% 的数据，2 ≤ K ≤ |S | ≤ 5 × 105。S 只包含小写字母。c1 和 c2 都是小写字母。

|S | 代表字符串 S 的长度。

题意分析

分析一下题目可得知，对于每一个c2，我们只需要知道在k个字符之前，出现过多少次c1，就有多少个子串可以被简写，所以用一个类似前缀和的数组维护c1在每次c2出现之前k个位置时出现的次数就行了

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

typedef long long LL;
const int N = 5e5 + 10;

int s[N],k;
LL ans;
char str[N];
char st,ed;

int main(){
	cin >> k >> str+1 >> st >> ed;
	int n = strlen(str+1);
	for (int i = 1, t = 0; i <= n; i++)
	{
		if (str[i] == st) 
			t++;
	    s[i] = t;
		if (i >= k && str[i] == ed) 
			ans += s[i - k + 1];
	}
	
	cout << ans;
	return 0;
}

前缀和思路
s[i]表示前缀中给定首字母(即st)出现的次数（用t变量维护），遍历的同时当出现尾字母(即ed)时，答案累加s[i - k + 1]
s[i - k + 1]表示在k个字符以前st出现过的次数，即为题意中可被简写的字符串的个数

双指针思路
把字符串中等于st的下标都存起来，把等于ed的下标也都存起来
然后判断两者下标之间有无大于等于k，直接加上 ed的个数-当前ed的下标 因为cnt2等于ed的个数加1，所以直接减就好，不用加1

#include<iostream>
#include<cstring>
using namespace std;
  
typedef long long LL;
const int N=5e5+10;
  
int l[N],r[N];
int k;
char str[N];
char st,ed;
int cnt1 = 1,cnt2 = 1,idx = 1;
LL ans;
  
int main()
{
	cin >> k >> str >> st >> ed;
	for(int i=0;i<strlen(str);i++){
		if(str[i]==st){
			l[cnt1++] = i;
		}
		if(str[i]==ed){
			r[cnt2++] = i;
		}
	}
	for(int i = 1;i < cnt1;i++){
		for(int j = idx;j < cnt2;j++){
			if(r[j] - l[i] < k - 1) 
			  continue;
			else{
				ans += (cnt2-j);
				idx = j;
				break;
			}
		}
	}
	cout<<ans<<endl; 
	return 0;
}