golang刷leetcode 滑动窗口（2）K 个不同整数的子数组

输出：A = [1,2,1,2,3], K = 2
输入：7
解释：恰好由 2 个不同整数组成的子数组：[1,2], [2,1], [1,2], [2,3], [1,2,1], [2,1,2], [1,2,1,2].

输入：A = [1,2,1,3,4], K = 3
输出：3
解释：恰好由 3 个不同整数组成的子数组：[1,2,1,3], [2,1,3], [1,3,4].

func subarraysWithKDistinct(A []int, K int) int {
    start:=0
    start2:=0
    m:=make(map[int]int)
    m2:=make(map[int]int)
    k:=0
    k2:=0
    sum:=0
    for end:=0;end<len(A);end++{
        k,start=startEnd(m,A,k,K,start,end,func(k int,K int)bool{ return k>K})
         k2,start2=startEnd(m2,A,k2,K,start2,end,func(k int,K int)bool{ return k>=K})
        fmt.Println(start,start2,end,k,k2)
        sum+=start2-start
    }
    return sum
}

func startEnd(m map[int]int,A []int, k,K,start,end int,compare func(a,b int)bool)(int,int){
    if  m[A[end]]==0{
            k++
        }
         m[A[end]]++
    for compare(k,K){
            m[A[start]]--
            if m[A[start]]==0{
                k--
            }
            start++
        }
    return k,start
}