题目描述
给两个只包含小写字母的字符串s和t,求字符串s中长度为k的本质不同子串在字符串t中的出现次数总和。
两个长度为k的子串,它们对应位置的字母存在不同,则它们是本质不同的。
输入
第一行一个整数k
接下来两行,两个字符串s和t
数据范围:
s和t的长度<=105
k <= len(s)
输出
输出一个整数,表示答案
样例输入
2
aaa
aaaaa
样例输出
4
提示
样例解释:s为"aaa",t为"aaaaa",s中长度为2的本质不同子串为"aa",它在t中的出现次数为4,所以答案是4。
思路
本来想用kmp,发现不会写,就用了string容器中自带的substr函数进行求解
使用set求不同本质子串
#include<iostream>
#include<cstring>
#include<set>
#include<string>
using namespace std;
const int maxn=100010;
string str,subst;
int k;
//‘\0’是字符串的结束符号,由编译器自动加上。
//使用set求不同本质子串,因为set去重
set<string> st;
void push(){
for(int i=0;i<subst.length()-k;i++){
int j=i+k;
string s=subst.substr(i,j);
st.insert(s);
}
}
//统计s2在s1中的出现次数
int tot=0; //次数
void count(string s1,string s2,int pos){ //从pos位开始找
if(s1.find(s2,pos)==string::npos) return;//未找到
if(s1.find(s2,pos)!=string::npos){
tot++;
count(s1,s2,pos+1);
}
}
int main(){
cin>>k;
cin>>subst;
cin>>str;
push();
int ans=0;//字符串s中长度为k的本质不同子串在字符串t中的出现次数总和
for(set<string>::iterator it=st.begin();it!=st.end();it++){
count(str,*it,0);
ans=ans+tot;
tot=0;
}
cout<<ans<<endl;
return 0;
}
//之前用char[]数组写的计算个数算法
//strstr()找到返回位置,未找到返回NULL
/*int tot=0;
void count(char s[],char sub[]){
char *p=strstr(s,sub);
if(p==NULL) return;
if(p!=NULL) {
tot++;
p=p+1;
count(p,sub); //从p位置再次查找
}
} */