leetcode-992. K 个不同整数的子数组之滑动窗口

这个题目的示例其实是有误导性的，假如题目以：12，121，21，1212，212，12，23的方式给出，我们马上就可以发现其实就是求以A[i]结尾的且满足不同整数个数为K的子数组的个数！因此，我们其实只要知道使得左侧满足不同整数个数为k的最远的位置low，和使得左侧满足不同整数个数为k的最远的位置high，high-low就是我们要的结果！将每个位置的high-low求和，就是题解！
这里如何维护high和low的位置有技巧，题目已经告诉了

所以可以不用map，直接用长度为A.size()+1的vector数组代替！

class Solution {
public:
    int subarraysWithKDistinct(vector<int>& A, int K) {
        vector<int> k_minus_one(A.size(),0);
        vector<int> k(A.size(),0);
        find(k_minus_one, K-1, A);
        find(k,K,A);
        int res = 0;
        for(int i=0;i<A.size();i++){
            res += (k_minus_one[i]-k[i]);
        }
        return res;
    }

    void find(vector<int>& arr, int num, vector<int>& A){
        vector<int> m(A.size()+1, 0);
        int sum = 0; // sum表示的是从i到j之间，数字的种类数量！
        
        for(int i=0,j=0;i<A.size();i++){
        	// m[A[i]] == 0，说明i这个位置是新的数字种类
            if(m[A[i]] == 0)
                sum++;
            m[A[i]]++;
            
            // 数字种类要是超过了限制，就需要收缩左边界，直到sum == num
            while(sum>num){
                m[A[j]]--;
                if(m[A[j]] == 0)
                    sum--;
                j++;
            }
            arr[i] = j;
        
        }
    }
};