Tony's Log

Algorithms, Distributed System, Machine Learning

  博客园 :: 首页 :: 博问 :: 闪存 :: 新随笔 :: 联系 :: 订阅 订阅 :: 管理 ::

A typical sliding window solution.

class Solution {
    int rec[26];    // for a-z
    int rec1[26];    // for A-Z
    int cnt;
    int &getCnt(char c)
    {
        if (c >= 'a' && c <= 'z')
            return rec[c - 'a'];
        if (c >= 'A' && c <= 'Z')
            return rec1[c - 'A'];
    }
public:
    int lengthOfLongestSubstringTwoDistinct(string s) 
    {
        std::fill(rec, rec + 26, 0);
        std::fill(rec1, rec1 + 26, 0);
        cnt = 0;

        int ret = 0;
        int start = 0;
        for (int i = 0; i < s.length(); i++)
        {
            char c = s[i];
            if (getCnt(c) == 0) cnt++;
            getCnt(c)++;

            int len = i - start + 1;
            if (cnt <= 2)
            {
                ret = std::max(ret, len);
            }
            else
            {
                while (start < i && cnt > 2)
                {
                    char nc = s[start];
                    if (getCnt(nc) == 1)
                        cnt--;
                    
                    getCnt(nc)--;
                    start++;
                }
            }
        }
        return ret;
    }
};
View Code
posted on 2015-08-21 13:16  Tonix  阅读(121)  评论(0编辑  收藏  举报