Contains Duplicate III 存在重複元素 III
阿新 • • 發佈:2018-12-15
給定一個整數陣列,判斷陣列中是否有兩個不同的索引 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
思路:我們需要維護一個長度為k的滑動視窗,然後在滑動窗口裡尋找滿足條件的等式:
-t<=|nums[i]-x|<=t => x>=nums[i]-t && x<=(nums[i]+t)
這裡的x表示滑動視窗中存在的數字,就是我們需要找的數字,那麼只要找到這個數字就可以了,我們可以呼叫自帶排序功能的set或者map的lower_bound函式,找到下標後觀察是否上限滿足要求,如果滿足直接返回true即可。
參考程式碼:
class Solution { public: bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) { set<int> window; for (int i = 0; i < nums.size(); i++) { if (i > k) window.erase(nums[i-k-1]); long long lower =((long long)nums[i]- (long long)t)<INT_MIN?INT_MIN:(nums[i] - t) ; // -t<=|nums[i]-x|<=t => x>=nums[i]-t && x<=(nums[i]+t) auto pos = window.lower_bound(lower); long long uper = ((long long)nums[i] + (long long)t) > INT_MAX ? INT_MAX : (nums[i] + t); if (pos != window.end() && *pos <= uper) return true; window.insert(nums[i]); } return false; } };