[LeetCode] 340. Longest Substring with At Most K Distinct Characters

Given a string, find the length of the longest substring T that contains at most k distinct characters.

Example 1:

Input: s = "eceba", k = 2
Output: 3
Explanation: T is "ece" which its length is 3.

Example 2:

Input: s = "aa", k = 1
Output: 2
Explanation: T is "aa" which its length is 2.

这个题跟159题一模一样，无非是把最多两个字母改成了最多K个字母。还是sliding window的思路做。不熟悉的话可以先做159题。

时间O(n)

空间O(1) - 只是一个256位长度的数组

Java实现

class Solution {
    public int lengthOfLongestSubstringKDistinct(String s, int k) {
        // corner case
        if (s == null || s.length() == 0) {
            return 0;
        }

        // normal case
        int[] map = new int[256];
        int start = 0;
        int end = 0;
        int maxLen = Integer.MIN_VALUE;
        int counter = 0;
        while (end < s.length()) {
            char c1 = s.charAt(end);
            if (map[c1] == 0) {
                counter++;
            }
            map[c1]++;
            end++;
            while (counter > k) {
                char c2 = s.charAt(start);
                if (map[c2] == 1) {
                    counter--;
                }
                map[c2]--;
                start++;
            }
            maxLen = Math.max(maxLen, end - start);
        }
        return maxLen;
    }
}

JavaScript实现

/**
 * @param {string} s
 * @param {number} k
 * @return {number}
 */
var lengthOfLongestSubstringKDistinct = function(s, k) {
    // corner case
    if (s == null || s.length == 0) {
        return 0;
    }

    // normal case
    let start = 0;
    let end = 0;
    let maxLen = -Infinity;
    let counter = 0;
    let map = {};
    while (end < s.length) {
        let c1 = s.charAt(end);
        if (map[c1] == null || map[c1] <= 0) {
            counter++;
        }
        map[c1] = map[c1] + 1 || 1;
        end++;
        while (counter > k) {
            let c2 = s.charAt(start);
            if (map[c2] == 1) {
                counter--;
            }
            map[c2]--;
            start++;
        }
        maxLen = Math.max(maxLen, end - start);
    }
    return maxLen;
};

sliding window相关题目

LeetCode 题目总结