【一次过】Lintcode 1368. 相同数字

给一个数组，如果数组中存在相同数字，且相同数字的距离小于给定值k，输出YES，否则输出NO。

样例

样例1

输入: array = [1,2,3,1,5,9,3] 和 k = 4
输出: "YES"
解释:
index为3的1和index为0的1距离为3，满足题意输出YES。

样例2

输入: array = [1,2,3,5,7,1,5,1,3] 和 k = 4,
输出: "YES"
解释:
index为7的1和index为5的1距离为2，满足题意。

注意事项

• 输入的数组长度为n，保证n <= 100000。
• 数组元素的值为x，0 <= x <= 1e9。
• 输入的k满足 1 <= k < n。

---

解题思路：

建立HashMap存储数字与其下标的值，遍历数组时边存储map边检查当前值是否存在于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.get(nums[i]) == null)
                map.put(nums[i], i);
            else{       //在序列后面找到与nums[i]相同的数字
                if(i - map.get(nums[i]) < k)//如果这个相同的数字与前一个的距离小于k，则YES
                    return "YES";
                else    //否则更新这个相同数字的下标
                    map.put(nums[i], i);
            }
        }
        
        return "NO";
    }
}