题目内容
Given an array A of positive integers, call a (contiguous, not necessarily distinct) subarray of A good if the number of different integers in that subarray is exactly K.
(For example, [1,2,3,1,2] has 3 different integers: 1, 2, and 3.)
Return the number of good subarrays of A.
Example:
Input: A = [1,2,1,2,3], K = 2
Output: 7
Explanation: Subarrays formed with exactly 2 different integers: [1,2], [2,1], [1,2], [2,3], [1,2,1], [2,1,2], [1,2,1,2].
Example 2:
Input: A = [1,2,1,3,4], K = 3
Output: 3
Explanation: Subarrays formed with exactly 3 different integers: [1,2,1,3], [2,1,3], [1,3,4].
分析过程
- 题目归类：
array - 题目分析：
ij 为字串首和尾
那hashmap 来保存每一个结果，对于hashmap的key数不能超过K个，设置flag 来保存当前k的值添加string.charAt(j)hashmap中,并判断目前的flag 和 K 的关系 1. 小于K j右移 2. 等于k sum ++； j++; 3.大于k hashmap中的s.charAt(i++)减一，如果减到0了，则直接删除这个key/value; - 边界分析：
- 空值分析
- 循环边界分析
j == s.length()时停止
- 方法分析：
- 数据结构分析
hashmap就是比较适合做这种题目 - 状态机
- 状态转移方程
- 最优解
- 数据结构分析
- 测试用例构建
[] 0,
[aaa],
代码实现 大数据会超时，所以学习别人的方法
import java.util.*;
class Solution {
public int subarraysWithKDistinct(int[] A, int K) {
HashMap<Integer,Integer> map = new HashMap<>();
int flag = 0;
int sum = 0;
int i = 0,j = 0;
int addFlag = 0;
while(j<A.length){
if(map.containsKey(A[j])&&addFlag == 0)
map.put(A[j],map.get(A[j])+1);
else if(addFlag == 0){
map.put(A[j],1);
flag++;
}
if(flag < K) {
j++;
addFlag=0;
}else if (flag == K){
sum++;
int m = i + 1;
while(m<=j){
int n = m;
HashMap<Integer,Integer> map1 = new HashMap<>();
while(n<=j){
if(map1.containsKey(A[n]))
map1.put(A[n],map.get(A[n])+1);
else{
map1.put(A[n],1);
}
n++;
}
if(map1.size()==K)
sum++;
m++;
}
j++;
addFlag=0;
}else{
map.put(A[i],map.get(A[i])-1);
if(map.containsKey(A[i])&&map.get(A[i])==0){
map.remove(A[i]);
flag--;
}
i++;
addFlag=1;
}
}
return sum;
}
}