Leetcode P901水果成篮 Java双指针

Leetcode P901水果成篮 Java双指针

ideas

首先创建一个map用于记录当前背包，

        //背包
        HashMap<Integer,Integer> map = new HashMap<>();

创建2个指针为l，r。l 代表我们从哪里开始，r代表我们拿到了哪里。

     for (int r = 0,l=0; r < n; r++) {
     		....
     }

遇见水果我们直接放入背包，如果发现背包的元素大于2，那么就从l除开始依次丢下水果，直到当前这个品种的水果数量为0那么我们删除这个水果，这就保证了我们背包永远只有2个水果。

            //查看是否可以直接放
            map.put(fruits[r],map.getOrDefault(fruits[r],0)+1);

            //超出背包范围
            if (map.size() > 2) {
                map.put(fruits[l], map.get(fruits[l]) - 1);
                if (map.get(fruits[l]) == 0) {
                    map.remove(fruits[l]);
                }
                l++;
            }

code

class Solution {
    public int totalFruit(int[] fruits) {
        //背包
        HashMap<Integer,Integer> map = new HashMap<>();

        int res = 0;
        int n = fruits.length;
        for (int r = 0,l=0; r < n; r++) {
            //查看是否可以直接放
            map.put(fruits[r],map.getOrDefault(fruits[r],0)+1);

            //超出背包范围
            if (map.size() > 2) {
                map.put(fruits[l], map.get(fruits[l]) - 1);
                if (map.get(fruits[l]) == 0) {
                    map.remove(fruits[l]);
                }
                l++;
            }
            //System.out.println(r+"and"+l);
            res = Math.max(res,r-l+1);
        }

        return res;
    }
}