【算法系列（五）】：集合

目录

一、算法应用

349. 两个数组的交集

217. 存在重复元素

160. 相交链表

136. 只出现一次的数字

1042. 不邻接植花

---

集合技术在解题中主要用于处理有数据重复出现的问题。这里直接看leetcode题库的实例来感受下吧。。。

一、算法应用

349. 两个数组的交集

• 题目描述

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

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

示例 2：
输入：nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出：[9,4]

• 解题思路

直接利用集合这种结构

• c++代码实现

vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
    set<int> set1(nums1.begin(),nums1.end());
    set<int> set2(nums2.begin(),nums2.end());
 
    vector<int> res;
 
    for(auto i=set1.begin();i!=set1.end();++i){
        if(set2.find(*i)!=set2.end()){
            res.push_back(*i);
        }
    }
 
    return res;
}

217. 存在重复元素

• 题目描述

给定一个整数数组，判断是否存在重复元素。

如果任意一值在数组中出现至少两次，函数返回 true 。如果数组中每个元素都不相同，则返回 false 。

示例 1:
输入: [1,2,3,1]
输出: true

示例 2:
输入: [1,2,3,4]
输出: false

• 解题思路

通过集合的方法

• c++代码实现

 bool containsDuplicate(vector<int>& nums) {
	 map<int, int> dict;
	 for (int i = 0;i < nums.size();i++) {
		 if (dict.find(nums[i]) != dict.end()) {
			 return true;
		 }
		 else {
			 dict.insert(pair<int, int>(nums[i], 1));
		 }
	 }

	 return false;
 }

160. 相交链表

• 题目描述

编写一个程序，找到两个单链表相交的起始节点。

如下面的两个链表：

在节点 c1 开始相交。

示例 1：

输入：intersectVal = 8, listA = [4,1,8,4,5], listB = [5,0,1,8,4,5], skipA = 2, skipB = 3
输出：Reference of the node with value = 8
输入解释：相交节点的值为 8 （注意，如果两个链表相交则不能为 0）。从各自的表头开始算起，链表 A 为 
[4,1,8,4,5]，链表 B 为 [5,0,1,8,4,5]。在 A 中，相交节点前有 2 个节点；在 B 中，相交节点前有 3 
个节点。

• 解题思路

通过集合的方法

• c++代码实现

 ListNode *getIntersectionNode(ListNode *headA, ListNode *headB) {
	 set<ListNode *> hash;
	 ListNode *tmp = headA;
	 while (tmp != NULL) {
		 hash.insert(tmp);
		 tmp = tmp->next;
	 }

	 tmp = headB;
	 while (tmp != NULL) {
		 if (hash.find(tmp) != hash.end()) {
			 return tmp;
		 }
		 tmp = tmp->next;
	 }

	 return NULL;
 }

136. 只出现一次的数字

• 题目描述

给定一个非空整数数组，除了某个元素只出现一次以外，其余每个元素均出现两次。找出那个只出现了一次的元素。

示例 1:
输入: [2,2,1]
输出: 1

示例 2:
输入: [4,1,2,1,2]
输出: 4

• 解题思路

通过集合的方法

• c++代码实现

int singleNumber(vector<int>& nums) {
	 set<int> sets;
	 for (int i = 0;i < nums.size();++i) {
		 if (sets.find(nums[i]) != sets.end()) {
			 sets.erase(nums[i]);
			 continue;
		 }
		 else {
			 sets.insert(nums[i]);
		 }
	 }

	 return *sets.begin();
 }

1042. 不邻接植花

• 题目描述

有 N 个花园，按从 1 到 N 标记。在每个花园中，你打算种下四种花之一。

paths[i] = [x, y] 描述了花园 x 到花园 y 的双向路径。

另外，没有花园有 3 条以上的路径可以进入或者离开。

你需要为每个花园选择一种花，使得通过路径相连的任何两个花园中的花的种类互不相同。

以数组形式返回选择的方案作为答案 answer，其中 answer[i] 为在第 (i+1) 个花园中种植的花的种类。花的种类用  1, 2, 3, 4 表示。保证存在答案。

示例 1：
输入：N = 3, paths = [[1,2],[2,3],[3,1]]
输出：[1,2,3]

示例 2：
输入：N = 4, paths = [[1,2],[3,4]]
输出：[1,2,1,2]

示例 3：
输入：N = 4, paths = [[1,2],[2,3],[3,4],[4,1],[1,3],[2,4]]
输出：[1,2,3,4]

• 解题思路

利用 字典 + 集合 构造图的邻接表。

• c++代码实现

bool judge(int j, vector<int>& res, vector<int>& vec)
    {
        for(int k=0;k<vec.size();k++)
        {
            if(res[vec[k]]==j)
            {
                return false;
            }
        }
        return true;
    }
 
    vector<int> gardenNoAdj(int N, vector<vector<int>>& paths) 
    {
        vector<int> res(N,0);
        int m=paths.size();
        vector<vector<int>> tmp(N);
        for(int i=0;i<m;i++)
        {
            tmp[paths[i][0]-1].push_back(paths[i][1]-1);
            tmp[paths[i][1]-1].push_back(paths[i][0]-1);
        }
        for(int i=0;i<N;i++)
        {
            for(int j=1;j<=4;j++)
            {
                if(judge(j,res,tmp[i]))
                {
                    res[i]=j;
                    break;
                }
            }
        }
        return res;
    }

参考链接：

集合技术在求解算法题中的应用