目录
集合技术在解题中主要用于处理有数据重复出现的问题。这里直接看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;
}
参考链接: