刷题力扣349-两个数组的交集

这道题代码随想录用的是哈希数据结构,什么时候用哈希表,哈希表都是用来快速判断一个元素是否出现在集合里,相对于枚举的话,哈希表的时间复杂度只有O(1)。

常见的三种哈希结构

数组

set(集合)

map(映射)

当出现快速判断一个元素是否出现在集合里的时候,就要考虑哈希法,牺牲空间换取时间。

题目描述:

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

 

说明: 输出结果中的每个元素一定是唯一的。 我们可以不考虑输出结果的顺序。

在不考虑数量的情况下,假设数据有无穷大,我们可以用set的哈希结构去解这道题。在java中使用set,可以得到一个不包含重复元素的集合,而且set无序,且允许出现null元素。

set方法:

boolean add(E e) //如果 set 中尚未存在指定的元素,则添加此元素(可选操作)。
boolean addAll(Collection<? extends E> c) //如果 set 中没有指定 collection 中的所有元素,则将其添加到此 set 中(可选操作)。
void   clear() //移除此 set 中的所有元素(可选操作)。
boolean contains(Object o) //如果 set 包含指定的元素,则返回 true。
boolean containsAll(Collection<?> c) //如果此 set 包含指定 collection 的所有元素,则返回 true。
boolean equals(Object o) //比较指定对象与此 set 的相等性。
int    hashCode() //返回 set 的哈希码值。
boolean isEmpty() //如果 set 不包含元素,则返回 true。
Iterator<E> iterator() //返回在此 set 中的元素上进行迭代的迭代器。
boolean remove(Object o) //如果 set 中存在指定的元素,则将其移除(可选操作)。
boolean removeAll(Collection<?> c) //移除 set 中那些包含在指定 collection 中的元素(可选操作)。
boolean retainAll(Collection<?> c) //仅保留 set 中那些包含在指定 collection 中的元素(可选操作)。
int    size() //返回 set 中的元素数(其容量)。
Object[] toArray() //返回一个包含 set 中所有元素的数组。
<T> T[] toArray(T[] a) //返回一个包含此 set 中所有元素的数组;返回数组的运行时类型是指定数组的类型。

解题思路:

将nums1的集合元素放到一个set集合里,然后去遍历nums2集合元素,如果nums2集合元素在set集合里出现,我们就可以把该元素的下标放到结果的set集合里。set集合存放的其实就是一个下标与元素的映射关系。

代码:

class Solution {
public int[] intersection(int[] nums1, int[] nums2) {
if(nums1 == null || nums1.length == 0 || nums2 == null || nums2.length == 0){
return new int[0];
}
Set<Integer> set1 = new HashSet<>(); //用于存放nums1的元素
Set<Integer> resSet = new HashSet<>(); //用于存放交集的元素
for(int i : nums1){
set1.add(i);
}
for(int i : nums2){
if(set1.contains(i)){ //如果nums2的元素在set1中出现,把该元素下标放到resSet集合中
resSet.add(i);
}
}
//将结果集合转为数组-法一
return resSet.stream().mapToInt(x->x).toArray();

//将结果集合转为数组-法二
int[] resArr = new int[resSet.size()];
int index = 0;
for(int i : resSet1){
resArr[index++] = i;
}

return resArr;
}

力扣这道题后来把数据范围限制在1000以内,那么可以用数组来写,java如果用数组做哈希结构的话,代码方法如下:

class Solution {
public int[] intersection(int[] nums1, int[] nums2) {
if(nums1 == null || nums1.length == 0 || nums2 == null || nums2.length == 0){
return new int[0];
}
int[] hash = new int[1005];
Set<Integer> resSet = new HashSet<>(); //存放结果是为了去重
for(int i : nums1){ //对于nums1中出现过的元素,在hash数组中做记录
hash[i] = 1;
}
for(int i : nums2){
if(hash[i] == 1){
resSet.add(i);
}
}
int[] result = new int[resSet.size()];
int index = 0;
for(int i : resSet){
result[index++] = i;
}
return result;
}
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
请问您需要的是两个数组交集元素还是交集元素的数量? 如果是交集元素,可以采用双指针的方式,先将两个数组排序,然后设定两个指针 i 和 j,分别指向两个数组的开头,比较两个指针所指向的元素的大小,如果相等,则说明找到了一个交集元素,将其存到结果数组中;如果不相等,则将指向小的元素的指针向后移动一位,直到其中一个指针越界为止。 代码实现如下: ```c int* intersect(int* nums1, int nums1Size, int* nums2, int nums2Size, int* returnSize){ int i = 0, j = 0, k = 0; int* res = (int*)malloc(sizeof(int) * (nums1Size > nums2Size ? nums2Size : nums1Size)); // 排序 qsort(nums1, nums1Size, sizeof(int), cmp); qsort(nums2, nums2Size, sizeof(int), cmp); while (i < nums1Size && j < nums2Size) { if (nums1[i] == nums2[j]) { res[k++] = nums1[i]; i++; j++; } else if (nums1[i] < nums2[j]) { i++; } else { j++; } } *returnSize = k; return res; } int cmp(const void* a, const void* b) { return *(int*)a - *(int*)b; } ``` 如果是交集元素的数量,可以采用哈希表的方式,用一个哈希表来记录第一个数组中的元素出现的次数,然后遍历第二个数组,在哈希表中查找是否存在相同的元素,如果存在,则说明找到了一个交集元素,将其计数并从哈希表中删除该元素,最后返回计数结果即可。 代码实现如下: ```c int* intersect(int* nums1, int nums1Size, int* nums2, int nums2Size, int* returnSize){ int i = 0, j = 0, k = 0; int* res = (int*)malloc(sizeof(int) * (nums1Size > nums2Size ? nums2Size : nums1Size)); int* hash = (int*)malloc(sizeof(int) * 1001); // 由于题目条件限制在 [0, 1000] 范围内,所以哈希表可以开得比较小 memset(hash, 0, sizeof(hash)); for (i = 0; i < nums1Size; i++) { hash[nums1[i]]++; } for (j = 0; j < nums2Size; j++) { if (hash[nums2[j]] > 0) { res[k++] = nums2[j]; hash[nums2[j]]--; } } *returnSize = k; return res; } ``` 希望这些代码能够帮到您,如有疑问请随时问我~

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值