第十四届省赛大学B组(C/C++)子串简写

原题链接:子串简写

程序猿圈子里正在流行一种很新的简写方法:

对于一个字符串,只保留首尾字符,将首尾字符之间的所有字符用这部分的长度代替。

例如 internationalization 简写成 i18nKubernetes 简写成 K8sLanqiao 简写成 L5o 等。

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

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

输入格式

第一行包含一个整数 K。

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

输出格式

一个整数代表答案。

数据范围

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

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

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

输入样例:

4
abababdb a b

输出样例:

6

样例解释

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

[abab]abdb
[ababab]db
[abababdb]
ab[abab]db
ab[ababdb]
abab[abdb]

解题思路:

此题主要是考察前缀和,此题暴力的话只会过部分测试点,会超时。我们定义前缀和presumb[i]表示第i个位置之前(包括第i个位置)有多少个b,这样确定了尾,我们再遍历一遍只要是头,那么就可以利用前缀和求出方案,以下为样例为例。

那么我们怎么求方案数,当我们遍历到第i个位置上的首字符时,题目中要求最短字串为N,那么我们再往后走N-1个,第N-1个及到最后凡是有尾字符的都是一种情况,那么方案数就是presumb[s.zize()-1]-presumb[i+n-2],下面以i=2为例。


AC代码:

cpp 复制代码
#include <iostream>
#include<cstring>
using namespace std;
typedef long long ll;//注意开longlong
const int N=5e5+5;
string s;
char a,b;
ll n,ans,res;
ll presumb[N];
int main()
{
  cin>>n>>s>>a>>b;
  for(int i=0;i<s.size();i++){
    if(s[i]==b) presumb[i]=presumb[i-1]+1;
    else presumb[i]=presumb[i-1];
  }
  for(int i=0;i<s.size();i++){
    if(s[i]==a&&i+n<=s.size()){
      res+=presumb[s.size()-1]-presumb[i+n-2];
    }
  }
  cout<<res<<endl;
  return 0;
}
相关推荐
痛&快乐着7 分钟前
C++高精度计时方法总结(测试函数运行时间)
c++
Ddddddd_1587 分钟前
C++ | Leetcode C++题解之第419题棋盘上的战舰
c++·leetcode·题解
CrazyZ1268 分钟前
C++第九章编程练习答案
开发语言·c++·笔记·学习·算法
小周的C语言学习笔记11 分钟前
鹏哥C语言44---函数的声明和定义
c语言·c++·算法
XXXJessie23 分钟前
c++278函数指针
java·c++·算法
小胡不加班26 分钟前
2024.09.22 leetcode 每日一题
算法·leetcode·职场和发展
Parasoft中国26 分钟前
Parasoft助力Joby Aviation符合DO-178B标准
c++·代码规范·测试覆盖率
DdddJMs__13529 分钟前
C语言 | Leetcode C语言题解之第419题棋盘上的战舰
c语言·leetcode·题解
GGBondlctrl30 分钟前
【后端开发】JavaEE初阶—Theard类及常见方法—线程的操作(超详解)
java·开发语言·thread类的构造方法·thread类的属性·启动线程·终止线程·等待线程
机器不会学习CL32 分钟前
时序预测|基于灰狼优化LightGBM的时间序列预测Matlab程序GWO-LightGBM 单变量和多变量 含基础模型
开发语言·神经网络·matlab·数据挖掘·回归·时间序列·智能优化