leetcode 219 Contains Duplicate II

題目詳情

Given an array of integers and an integer k, find out whether there are two distinct indices i and j in the array such that nums[i] = nums[j] and the absolute difference between i and j is at most k.

這道題目的意思是：輸入一個整數數組和一個整數k，若是數組中存在相等的兩個數，並且他們的位置差不超過k，那麼返回true，不然返回false

思路

• 這道題比較容易想到的想法就是用hashmap來存儲不一樣值的元素的值（key）和位置信息（value）。而後在每次遍歷的時候進行比較。
• 但上面這個想法並非最簡單的，若是咱們對任何索引大於k的元素進行遍歷的時候，同時刪除hashset中和當前元素的位置差已經超過k的對應元素。這樣就能夠減小後序查找的時間。
• 這樣只要新遍歷到的元素的值已經存在於hashset之中，咱們就能夠判定，這兩個元素的位置差必定是小於k的了。

解法

int length = nums.length;
        if(length<=1) return false;
        Set<Integer> count = new HashSet<Integer>();
        
        for(int i=0;i<length;i++){
            if(i > k){
                count.remove(nums[i-k-1]);
            }
            if(!count.add(nums[i])){
                return true;
            }
        }
        return false;