leetcode349. 两个数组的交集
  TEZNKK3IfmPf 28天前 19 0

给定两个数组,编写一个函数来计算它们的交集。

示例:
 输入:nums1 = [1, 2, 2, 1], nums2 = [2, 2]
 输出:[2]

思路:
按照我们做数学题时求交集的方法就行了,但注意在求交集前先分别对两个数组的元素进行去重。

求两个数组的交集的步骤可分为以下三步:

  1. 对nums1当中的元素进行去重,得到序列s1。
  2. 对nums2当中的元素进行去重,得到序列s2。
  3. 遍历s1,依次判断s1中的每个元素是否在s2当中出现,若没有出现则为交集元素。

代码如下:

class Solution {
     
       
public:
	vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
     
       
		//1、对nums1当中的元素进行去重,得到序列s1
		set<int> s1;
		for (auto e : nums1)
		{
     
       
			s1.insert(e);
		}
		//2、对nums2当中的元素进行去重,得到序列s2
		set<int> s2;
		for (auto e : nums2)
		{
     
       
			s2.insert(e);
		}
		//3、遍历s1,依次判断s1中的每个元素是否在s2当中出现,若没有出现则为交集元素
		vector<int> vRet;
		for (auto e : s1)
		{
     
       
			if (s2.find(e) != s2.end())
			{
     
       
				vRet.push_back(e);
			}
		}
		return vRet;
	}
};

时间复杂度: O ( N l o g N ) O(NlogN) O(NlogN)

优化思路:
上述代码中,先使用set容器对两个数组进行去重,然后在s2序列中分别查找s1序列的元素是否存在。set容器的底层是用红黑树实现的,在set容器当中查找元素时的时间复杂度是 O ( l o g N ) O(logN) O(logN),因此上述代码整体的时间复杂度是 O ( N l o g N ) O(NlogN) O(NlogN)

由于set容器去重后的序列是有序的,因此在这里我们可以利用双指针的思想将时间复杂度优化到 O ( N ) O(N) O(N),在用set容器对两个数组的元素进行去重后,使用双指针从去重后两个序列的第一个元素开始比较:

  1. 若两个指针指向的元素不相等,则小的一个元素属于差集,小的指针往后走。
  2. 若两个指针指向的元素相等,则该元素属于交集,两个指针同时往后走。
  3. 若其中一个序列走完了,则另一个序列剩下的元素都属于差集。

使用双指针进行处理,在 O ( N ) O(N) O(N)的时间复杂度下既能求出交集,还能同时求出差集。

拓展:该算法常用于同步数据,求出服务器和客户端之间数据的差集,进而将服务器上多的数据下载到客户端,将客户端上多的数据上传到服务器,最终做到数据同步。

优化代码如下:

class Solution {
     
       
public:
	vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
     
       
		//1、对nums1当中的元素进行去重,得到序列s1
		set<int> s1;
		for (auto e : nums1)
		{
     
       
			s1.insert(e);
		}
		//2、对nums2当中的元素进行去重,得到序列s2
		set<int> s2;
		for (auto e : nums2)
		{
     
       
			s2.insert(e);
		}
		//3、使用双指针找出s1和s2的交集
		vector<int> vRet;
		set<int>::iterator it1 = s1.begin();
		set<int>::iterator it2 = s2.begin();
		while (it1 != s1.end() && it2 != s2.end())
		{
     
       
			if (*it1 < *it2) //两个指针指向的元素不相等
			{
     
       
				it1++; //小的往后走
			}
			else if (*it2 < *it1) //两个指针指向的元素不相等
			{
     
       
				it2++; //小的往后走
			}
			else //两个指针指向的元素相等
			{
     
       
				vRet.push_back(*it1); //该元素属于交集
				//两个指针一起往后走
				it1++;
				it2++;
			}
		}
		return vRet;
	}
};

时间复杂度: O ( N ) O(N) O(N)

【版权声明】本文内容来自摩杜云社区用户原创、第三方投稿、转载,内容版权归原作者所有。本网站的目的在于传递更多信息,不拥有版权,亦不承担相应法律责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@moduyun.com

  1. 分享:
最后一次编辑于 28天前 0

暂无评论

推荐阅读
TEZNKK3IfmPf