度度熊找子串(百度2017秋招真题)
题目描述
度度熊收到了一个只有小写字母的字符串S,他对S的子串产生了兴趣,S的子串为S中任意连续的一段。他发现,一些子串只由一种字母构成,他想知道在S中一共有多少种这样的子串。
例如在串”aaabbaa”中,度度熊想找的子串有”a”,”aa”,”aaa”,”b”,”bb”五种。
(本题只考虑子串的种数,相同的子串在多个位置出现只算一次)
解题思路:
最长子串包括了所有非最长子串的情况,如aaa包含了a,aa,aaa三种情况,即子串的种数等于最长子串的长度,因此对于每个小写字母,找出最长连续子串,相加即得结果。
#include<iostream>
#include<string>
using namespace std;
int main()
{
int number=0; //记录总数
int count[26]={0}; //记录每个字母的最长子串的长度
string s; //输入
cin>>s;
int len=s.length(); //字符串长度
for(int i=0;i<len;i++){
int k=1;
count[s[i]-'a']=(count[s[i]-'a']>k)?count[s[i]-'a']:k;
//更新最长子串的长度
while(s[i]==s[i+1])
{
i++;
k++;
}
count[s[i]-'a']=(count[s[i]-'a']>k)?count[s[i]-'a']:k;
}
for(int i=0;i<26;i++){
number+=count[i];
}
cout<<number;
return 0;
}