2023年蓝桥杯省赛C++B组——字串简写

题目描述

解题代码

#include<bits/stdc++.h> 
using namespace std;
#define ll long long int
int main(){
	string s,first,end;
	ll k=0,ans=0;
	cin>>k;
	cin>>s>>first>>end;
	ll count[s.length()]={};
	for(int i=0;i<s.length();i++){
		if(s.substr(i,1)==first){
			if(i==0){
				count[i]=1;
			}else{
				count[i]=count[i-1]+1;	
			}
		}else{
			if(i==0){
				count[i]=0;	
			}else{
				count[i]=count[i-1];	
			}
		}
		if(s.substr(i,1)==end&&i+1>=k){
			ans+=count[i-k+1];
		}
	}
	cout<<ans;
}