蓝桥杯2023年第十四届省赛真题-子串简写
时间限制: 1s 内存限制: 256MB 提交: 2460 解决: 557
题目描述
程序猿圈子里正在流行一种很新的简写方法:对于一个字符串,只保留首尾字符,将首尾字符之间的所有字符用这部分的长度代替。例如 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 的长度。
图片的题
个人解法
看到题解有人用dp的,我个人在考场上用的后缀和,然后就又在写了一遍,现在学起来可以用前缀和也行。
代码如下
#include<iostream>
#include<cmath>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 5e5+10;
int k;
int cntb[N];
string a;
char x,y;
int main()
{
int cnta = 0;
cin >> k;
cin >> a >> x >> y;
cntb[a.size()] = 0;
for(int i = a.size()-1; i >= 0; i--)
{
cntb[i] = cntb[i+1]; //cntb[i]来统计包括下标i及之后给出字母y的个数
if(a[i] == y)
{
cntb[i] ++;
}
}
long long ans = 0;
for(int i = 0; i < a.size(); i++)
{
if(a[i] == x && i+k-1 <= a.size()-1 ) //如果这个a[i]==x 那就看它i+k-1及后面有多少个y,及是cntb[i+k-1]。
{
ans += cntb[i+k-1];
} //还可以统计一下x的个数,如果没有x了,就可以直接跳出循环。
}
cout << ans <<endl; //输出结果
return 0;
}
总结
大致思路就是这样了,当然也可以前缀和统计第i位x的个数,方法是一样的,个人这个复杂度算是o(n)了感觉,想记录一下我这第一次打蓝桥杯的思路,这题还是算简单的,我个人感觉是大题里面最简单的,(哈哈,观点不同勿喷,个人看法)。