题目:
Given an array of integers,
1 ≤ a[i] ≤ n
(n
= size of array), some elements appear twice and others appear once.
Find all the elements that appear twice in this array.
Could you do it without extra space and in O(n) runtime?
Example:Input: [4,3,2,7,8,2,3,1] Output: [2,3]
解释:
注意所有数字范围都在1~n中,这是一个可以利用的特点。
不用额外的空间,用set或者dict会有额外的空间 。
条件1 ≤ a[i] ≤ n
是一个值得注意的点,可以用正负号标记法
遍历nums
,记当前数字为n
(取绝对值,因为在后面乘以-1
了),将数字n
视为下标(因为a[i]∈[1, n])
当n
首次出现时,nums[n - 1]
乘以-1
当n
再次出现时,则nums[n - 1]
一定<0
,将n
加入答案
python代码:
class Solution(object):
def findDuplicates(self, nums):
"""
:type nums: List[int]
:rtype: List[int]
"""
result=[]
for n in nums:
n=abs(n)
if nums[n-1]<0:
result.append(n)
else:
nums[n-1]*=-1
return result
c++代码:
#include <cmath>
using namespace std;
class Solution {
public:
vector<int> findDuplicates(vector<int>& nums) {
vector<int>result;
for(auto n:nums)
{
n=abs(n);
if (nums[n-1]<0)
result.push_back(n);
else
nums[n-1]*=-1;
}
return result;
}
};
总结:
正负号标记法是一个经常用的方法,用字典或者集合确实速度快,但是会用额外空间。