1. 程式人生 > >LeetCode219:存在重複元素II

LeetCode219:存在重複元素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

解析:

        本題和217不同的是,有一個下標差的約束,所以應該利用map的特點,key值為nums中的值,value為該值的下標,如果元素在前面出現過,且兩個下標只差小於k,則返回true,否則繼續執行,遍歷執行結束後仍然沒有返回true,則返回false。

程式碼:

bool containsNearbyDuplicate(vector<int>& nums, int k)
{
	map<int, int>mNums;
	for (int i = 0; i < nums.size(); i++)
	{
		if (mNums.find(nums[i]) != mNums.end() && i - mNums[nums[i]] <= k)
		{
			return true;
		}
		else
		{
			mNums[nums[i]] = i;
		}
	}
	return false;
}