leetcode 992. K 个不同整数的子数组 (求有多少个区间满足不同数个数为K 简单)

题目

算不上尺取吧，看一眼就会了。

class Solution {
public:
    int subarraysWithKDistinct(vector<int>& a, int K) {
        int n=a.size(),vis[n+1],las[n];
        memset(vis,0,sizeof vis);
        vis[a[n-1]]=n-1;
        for(int i=n-2;i>=0;--i){
            if(!vis[a[i]]) las[i]=-1;
            else las[i]=vis[a[i]];
            vis[a[i]]=i;
        }
        memset(vis,0,sizeof vis);
        int ans=0,l=0,r=0,cnt=0,firstok=-1;//cnt是不同数的个数
        while(l<=n-1){
            while(r<=n-1&&(cnt+(vis[a[r]]?0:1))<=K) {
                cnt+=(vis[a[r]]?0:1),++vis[a[r]];
                if(cnt==K) ++ans,((firstok==-1)?(firstok=r):(firstok+=0));
                ++r;
            }
            if(--vis[a[l]]&&firstok!=-1) ans+=r-max(firstok,las[l]),firstok=max(las[l],firstok);
            else --cnt,firstok=-1;
            ++l;
        }
        return ans;
    }
};