题目地址:
https://www.lintcode.com/problem/longest-substring-with-at-most-k-distinct-characters/description
给定一个字符串 s s s,和一个整数 k k k,求 s s s中最多只含 k k k个不同字符的最长子串的长度。
思路是,用一个哈希表维护当前区间内的每个字符出现的次数。当当前维护的区间满足条件,则扩展右边界(显然左指针右移不会产生更优解),否则右移左指针缩小区间。代码如下:
class Solution {
public:
/**
* @param s: A string
* @param k: An integer
* @return: An integer
*/
int lengthOfLongestSubstringKDistinct(string &s, int k) {
// write your code here
unordered_map<char, int> mp;
int res = 0;
for (int i = 0, j = 0; i < s.size(); i++) {
mp[s[i]]++;
while (mp.size() > k) {
mp[s[j]]--;
if (!mp[s[j]]) mp.erase(s[j]);
j++;
}
res = max(res, i - j + 1);
}
return res;
}
};
时间复杂度 O ( n ) O(n) O(n),空间 O ( min { n , k } ) O(\min\{n,k\}) O(min{n,k})。