编程题—leetcode220:存在重复元素 III

题目链接

题目描述

给定一个整数数组，判断数组中是否有两个不同的索引 i 和 j，使得 nums [i] 和 nums [j] 的差的绝对值最大为 t，并且 i 和 j 之间的差的绝对值最大为 ķ。

示例 1:

输入: nums = [1,2,3,1], k = 3, t = 0
输出: true
示例 2:

输入: nums = [1,0,1,1], k = 1, t = 2
输出: true
示例 3:

输入: nums = [1,5,9,1,5,9], k = 2, t = 3
输出: false

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/contains-duplicate-iii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

1、使用hash桶解题，首先将其分为桶的大小为t的hash桶，这样桶内的元素的差值必定满足条件，不相邻的桶必定不满足条件
2、循环每次先获取到每个元素的桶id，然后判断桶内是否有元素，如果有，满足条件，如果没有，则判断左右两桶

通过代码

import java.util.HashMap;

class Solution {
    public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
        if (t < 0) return false;
        HashMap<Long,Long> data = new HashMap<Long, Long>();
        long w = t+1;
        for(int i=0;i<nums.length;i++){
            long id = getID(nums[i], w);
            if (i > k) {
                data.remove(getID(nums[i - k - 1], w));
            }
            if (data.containsKey(id)){
                return true;
            }else if (data.containsKey(id-1)&&nums[i]-data.get(id-1)<w){
                return true;
            }else if (data.containsKey(id+1)&&data.get(id+1)-nums[i]<w){
                return true;
            }
            data.put(id,(long)nums[i]);
        }
        return false;
    }
    private long getID(long num, long w) {
        return num < 0 ? (num + 1) / w - 1 : num / w;
    }
}