(LeetCode 每日一题)2799. 统计完全子数组的数目(哈希表、滑动窗口)

题目：2799. 统计完全子数组的数目

思路：对于一个符合要求的字数组，其左端点，可以无限扩展到最左边。时间复杂度0(n)，细节看注释。

C++版本：

class Solution {
public:
    int countCompleteSubarrays(vector<int>& nums) {
    	//集合，获得数组nums中不同元素的个数
        unordered_set<int> st(nums.begin(),nums.end());
        //哈希表，维护滑动窗口中元素出现的情况
        unordered_map<int,int> mp;
        //答案answer
        int ans=0;
        //left表示不符合要求时的最左端
        for(int left=0,right=0;right<nums.size();right++){
            mp[nums[right]]++;
            //维护left
            while(mp.size()>=st.size()){
                mp[nums[left]]--;
                if(mp[nums[left]]==0) mp.erase(nums[left]);
                left++;
            }
            //下标从0开始的，0~ans-1都是符合要求的
            ans+=left;
        }
        return ans;
    }
};

JAVA版本：

class Solution {
    public int countCompleteSubarrays(int[] nums) {
        Set<Integer> st= new HashSet<>();
        for(var x:nums){
            st.add(x);
        }
        int k=st.size();
        int ans=0;
        Map<Integer,Integer> mp=new HashMap<>();
        for(int left=0,right=0;right<nums.length;right++){
            mp.merge(nums[right],1,Integer::sum);
            while(mp.size()>=k){
                //mp.merge(nums[left],-1,Integer::sum);
                mp.put(nums[left],mp.get(nums[left])-1);
                if(mp.get(nums[left])==0){
                    mp.remove(nums[left]);
                }
                left++;
            }
            ans+=left;
        }
        return ans;
    }
}

Go版本：

func countCompleteSubarrays(nums []int) int {
    st := map[int]bool{}
    for _,x :=range nums {
        st[x]=true
    }
    k:=len(st)
    mp := map[int]int{}
    left :=0
    ans :=0
    for _,right := range nums {
        mp[right]++;
        for len(mp)>=k {
            mp[nums[left]]--;
            if mp[nums[left]]==0 {
                delete(mp,nums[left])
            }
            left++
        }
        ans+=left
    }
    return ans
}