LeetCode第219號題–存在重複元素II
題目如下
給定一個整數數組和一個整數 k,判斷數組中是否存在兩個不同的索引 i 和 j,使得 nums [i] = nums [j],並且 i 和 j 的差的絕對值最大爲 k。
示例 1:
輸入: nums = [1,2,3,1], k = 3
輸出: true
示例 2:
輸入: nums = [1,0,1,1], k = 1
輸出: true
示例 3:
輸入: nums = [1,2,3,1,2,3], k = 2
輸出: false
解答
這道題我們使用Set和Map都可以,但是使用Set的空間複雜度只需要爲O(k),而Map需要O(n),兩者的時間複雜度都是爲O(n)
方法一:Set,使用HashSet
我們這裏也要有滑動窗口的思想去解決此問題
// 使用HashSet
// 時間複雜度: O(n)
// 空間複雜度: O(k)
public boolean containsNearbyDuplicate(int[] nums, int k) {
if (nums == null || nums.length <= 1)
return false;
if (k <= 0)
return false;
HashSet<Integer> record = new HashSet<>();
for (int i = 0; i < nums.length; i++) {
if (record.contains(nums[i]))
return true;
record.add(nums[i]);
//我們確保數據不超過k+1,使得兩數的差的絕對值不超過 k
if (record.size() == k + 1)
record.remove(nums[i - k]);
}
return false;
}
方法一:Map,使用HashMap
// 使用HashMap
// 時間複雜度: O(n)
// 空間複雜度: O(n)
public boolean containsNearbyDuplicate(int[] nums, int k) {
if (nums == null || nums.length <= 1)
return false;
if (k <= 0)
return false;
HashMap<Integer, Integer> record = new HashMap<>();
record.put(nums[0], 0);
for (int i = 1; i < nums.length; i++) {
if (record.containsKey(nums[i]) && i - record.get(nums[i]) <= k)
return true;
record.put(nums[i], i);
}
return false;
}
還有一道更簡單的題,是在本題中減去了一個條件,實現起來更簡單:
LeetCode第217號題–存在重複元素
還有此題的升級版
LeetCode第220號題–存在重複元素III