【Lintcode】1368. Same Number

题目地址：

https://www.lintcode.com/problem/same-number/description

给定一个数组$A$，问其是否存在两个相同的数的距离小于给定数$k$。如果存在则返回"YES"，否则返回"NO"。

遍历$A$，用一个哈希表存每个数出现的最新位置。如果发现相邻的相同数距离小于$k$了则立即返回”YES“。否则最后返回"NO"。代码如下：

import java.util.HashMap;
import java.util.Map;

public class Solution {
    /**
     * @param nums: the arrays
     * @param k:    the distance of the same number
     * @return: the ans of this question
     */
    public String sameNumber(int[] nums, int k) {
        // Write your code here
        Map<Integer, Integer> map = new HashMap<>();
        
        for (int i = 0; i < nums.length; i++) {
            if (map.containsKey(nums[i])) {
                int prev = map.get(nums[i]);
                if (i - prev < k) {
                    return "YES";
                }
            }
            
            map.put(nums[i], i);
        }
        
        return "NO";
    }
}

时空复杂度$O(l_A)$.