描述
给定字符串S和整数K.
计算长度为K且包含K个不同字符的子串数
样例
String: “abcabc”
K: 3
Answer: 3
substrings: [“abc”, “bca”, “cab”]
String: “abacab”
K: 3
Answer: 2
substrings: [“bac”, “cab”]
思考
维护一个数组temp和一个值cnt,记录当前迭代值i之前k个长度的子串中包含的每个不同字符的个数和个数和。如果cnt==k,是符合要求的子串。每次i向后移动时,都要对第i-K个字符的数量和cnt进行更新。把得到的子串放到set中,最后返回set的大小。
代码
class Solution {
public:
/**
* @param stringIn: The original string.
* @param K: The length of substrings.
* @return: return the count of substring of length K and exactly K distinct characters.
*/
int KSubstring(string &stringIn, int K) {
// Write your code here
if (stringIn.length() < K)
return 0;
set<string> res;
int cnt = 0;
vector<int> temp(256,0);
for (int i = 0; i < stringIn.length();i++) {
temp[stringIn[i]]++;
if (temp[stringIn[i]] == 1)
cnt++;
if (i >= K) {
if (temp[stringIn[i-K]] == 1)
cnt--;
temp[stringIn[i-K]]--;
}
if (cnt == K)
res.insert(stringIn.substr(i-K+1,K));
}
return res.size();
}
};