【LeetCode & 剑指offer刷题】数组题10:Intersection of Two Arrays(系列)
【LeetCode & 剑指offer 刷题笔记】目录(持续更新中...)
349. Intersection of Two Arrays
Given two arrays, write a function to compute their intersection.
Example:
Given
nums1
=
[1, 2, 2, 1]
,
nums2
=
[2, 2]
, return
[2]
.
Note:
-
Each element in the result must be unique .
-
The result can be in any order.
/*
问题:求两个数组的交集(不能有重复的数)
方法一:
hash
表法,用
set
避免重复
O(n) ,O(n)
*/
class
Solution
{
public
:
vector
<
int
>
intersection
(
vector
<
int
>&
nums1
,
vector
<
int
>&
nums2
)
{
if
(
nums1
.
empty
()
||
nums2
.
empty
())
return
{};
unordered_set
<int> m(nums1.begin(), nums1.end()); //vector转set,避免重复
vector
<
int
>
res
;
for
(
auto
a
:
nums2
)
{
if
(
m
.
count
(
a
))
{
res
.
push_back
(
a
);
m
.
erase
(
a
);
//去除该数
}
}
return
res
;
}
};
/*
方法二:排序之后再找
用时间来换空间
O(nlogn),O(1)
*/
class
Solution
{
public
:
vector
<
int
>
intersection
(
vector
<
int
>&
nums1
,
vector
<
int
>&
nums2
)
{
if
(
nums1
.
empty
()
||
nums2
.
empty
())
return
{};
vector
<
int
>
res
;
int
i
=
0
,
j
=
0
;
sort
(
nums1
.
begin
(),
nums1
.
end
());
sort
(
nums2
.
begin
(),
nums2
.
end
());
while
(
i
<
nums1
.
size
()
&&
j
<
nums2
.
size
())
//
从起始位置开始分别扫描两个数组
{
if (nums1[i] < nums2[j]) i++; //递增第一个数组的指针
else if (nums1[i] > nums2[j]) j++; //递增第二个数组的指针
else
{
if
(
res
.
empty
()
||
res
.
back
()
!=
nums1
[
i
])
//如果为空或者不重复时才push交集元素
{
res
.
push_back
(
nums1
[
i
]);
}
i
++; j++; //同时递增两个数组指针
}
}
return
res
;
}
};
Intersection of Two Arrays II
Given two arrays, write a function to compute their intersection.
Example:
Given
nums1
=
[1, 2, 2, 1]
,
nums2
=
[2, 2]
, return
[2, 2]
.
Note:
-
Each element in the result should appear as many times as it shows in both arrays.
-
The result can be in any order.
Follow up:
-
What if the given array is already sorted? How would you optimize your algorithm?
-
What if nums1 's size is small compared to nums2 's size? Which algorithm is better?
-
What if elements of nums2 are stored on disk, and the memory is limited such that you cannot load all elements into the memory at once?
/*
问题:计算两个数组的交集
(
交集中元素与顺序无关,不是指公共子数组
)
可以返回重复的数
思考题
3
:
If only nums2 cannot fit in memory, put all elements of nums1 into a HashMap, read chunks of array that fit into the memory, and record the intersections.
If both nums1 and nums2 are so huge that neither fit into the memory, sort them individually (external sort), then read 2 elements from each array at a time in memory, record intersections.
*/
/*
*
方法一:
hash
表
*
过程:构建
hash
表计数器,依
nums2
在计数器中出现的次数,如果出现了,就
push
元素,计数器对应值减一
*
分析:
O(m+n),O(m)
*/
#include <unordered_map>
class
Solution
{
public
:
vector
<
int
>
intersect
(
vector
<
int
>&
nums1
,
vector
<
int
>&
nums2
)
{
unordered_map
<
int
,
int
>
counter
;
vector
<
int
>
res
;
for
(
int
num
:
nums1
)
counter
[
num
]++;
//
统计每个数的出现次数(首次调用
operator[]
以零初始化计数器
for
(
int
num
:
nums2
)
{
if
(
counter
.
find
(
num
)
!=
counter
.
end
()
&&
counter
[
num
]>=
1
)
//
如果在
nums1
中找得到对应元素而且计数值大于
1
时,说明是交集的一部分
{
counter
[
num
]--;
res
.
push_back
(
num
);
}
}
return
res
;
}
};
/*
* 方法二:排序之后再找
以时间换空间
*
分析:
O
(
mlogm +nlogn+m+n
),O(1)
*/
#include <algorithm>
class
Solution
{
public
:
vector
<
int
>
intersect
(
vector
<
int
>&
nums1
,
vector
<
int
>&
nums2
)
{
if(nums1.empty() || nums2.empty()) return {};
vector<int> res;
int
i
=
0
,
j
=
0
;
sort
(
nums1
.
begin
(),
nums1
.
end
());
sort
(
nums2
.
begin
(),
nums2
.
end
());
while
(
i
<
nums1
.
size
()
&&
j
<
nums2
.
size
())
//i
,
j
分别用来遍历两个数组
{
if
(
nums1
[
i
]
<
nums2
[
j
])
i
++;
//
递增第一个数组的指针
else
if
(
nums1
[
i
]
>
nums2
[
j
])
j
++;
//
递增第二个数组的指针
else
{
res
.
push_back
(
nums1
[
i
]);
i
++;
j
++;
//
同时递增两个数组指针
}
}
return
res
;
}
};