这次还是动态规划的题:Unique Substrings in Wraparound String
467 | Unique Substrings in Wraparound String | 31.5% | Medium |
Consider the string s
to be the infinite wraparound string of "abcdefghijklmnopqrstuvwxyz", so s
will look like this: "...zabcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyzabcd....".
Now we have another string p
. Your job is to find out how many unique non-empty substrings of p
are present in s
. In particular, your input is the string p
and you need to output the number of different non-empty substrings of p
in the string s
.
Note: p
consists of only lowercase English letters and the size of p might be over 10000.
Example 1:
Input: "a" Output: 1 Explanation: Only the substring "a" of string "a" is in the string s.
Example 2:
Input: "cac" Output: 2 Explanation: There are two substrings "a", "c" of string "cac" in the string s.
Example 3:
Input: "zab" Output: 6 Explanation: There are six substrings "z", "a", "b", "za", "ab", "zab" of string "zab" in the string s.
给了一个string s是这样的:abcdefg…xyzabcdefg...xyz这样无限循环,然后呢要求在新的string p里面招数有多少个非空的substring是在s里面的,也就是说要找出p里面有多少个类似于abc...或者是ghi...之类的substring。而且!!!xyzab...这样的也算(划重点!!!)
这个问题的动态规划的重点在于,如果你发现了一个新的substring,你是无法确定这个substring里面的substring是否之前有被统计过了,如果算法设计不好有可能出现重复统计或者是漏算了的情况。
26个字母,我用了一个vector来存放对应的各个字母的check,这个check表示这个字母在之前如果出现并且被统计过的话,它在那个substring是在第几位,因为我们知道,对于一个长度为n的substring来说,它的substring个数为1+2+3+...+n,因为每增加一位,个数就增加n。
但是对于一个存在substring已经出现过的substring来说,比如bcabcd,那么我们可以得到在遍历到a的时候,check[1]和check[2]分别是1和2,因为在第一个substring中b和c分别位于第一位和第二位,那么在第二次遍历到b、c的时候,此时b、c是分别位于第二位和第三位,那bc这个位于abcd之中substring前面已经出现过了怎么办呢,我们知道没增加1位就会增加n个substring,所以当b在第m位时和b在第n位时相比较会增加m-n个substring(m<n)。所以我们利用的是字符在substring中的第几位来判断增加的substring的个数。
听起来可能很绕,还是上代码吧
int findSubstringInWraproundString(string p)
{
vector<int> check;
for(int i=0;i<26;i++)
{
check.push_back(0);
}
int sub=0;
int length=0;
int temp=0;
for(int i=0;i<p.size();i++)
{
if(i>0)
{
if((p[i]-p[i-1]+26)%26!=1)
{
length=0;
}
}
length++;
if(check[p[i]-'a']<length)
{
sub+=(length-check[p[i]-'a']);
check[p[i]-'a']=length;
}
}
return sub;
}