力扣350(java&python)-两个数组的交集 II(简单)

题目:

给你两个整数数组 nums1 和 nums2 ,请你以数组形式返回两数组的交集。返回结果中每个元素出现的次数,应与元素在两个数组中都出现的次数一致(如果出现次数不一致,则考虑取较小值)。可以不考虑输出结果的顺序。

 

示例 1:

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

输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出:[4,9]
 

提示:

1 <= nums1.length, nums2.length <= 1000
0 <= nums1[i], nums2[i] <= 1000
 

进阶:

如果给定的数组已经排好序呢?你将如何优化你的算法?
如果 nums1 的大小比 nums2 小,哪种方法更优?
如果 nums2 的元素存储在磁盘上,内存是有限的,并且你不能一次加载所有的元素到内存中,你该怎么办?

来源:力扣(LeetCode
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解题思路:

双指针:先将两个数组进行排序,然后定义两个指针index1, index2 分别指向两个数组的首元素,并创建一个数组res来存放两个数组的交集,然后就开始遍历两数组的元素,并对当前index1和index2所指的元素大小进行比较:

  • 如果nums1[index1] < num2[index2]:则数值小的指针index1,向右移一位,即index1++;
  • 如果nums1[index1] > num2[index2]:则数值小的指针index2,向右移一位,即index2++;
  • 如果nums1[index1] = num2[index2]:则将交集元素,存入res中,index1,index2,index均向右移一位,即index1++;index2++; index++

如果有任何一个数组遍历结束,则另一个数组的剩余元素均为唯一元素,不会产生交集。

举例:

nums1 = [4,9,5,9]
nums2 = [9,4,9,8,4]

①初始将两个数组排序,并定义指针位置指向两个数组的起始元素;

②上一步中初始4 == 4,则将4加入到res[index]中,双指针均向右移动一位,res的索引index也向后移动一位;

③上一步移动后5 != 4,即让数值小的指针往后移动一位即 index2++;

④上一步移动后5 != 8,即让数值小的指针往后移动一位即 index1++;

⑥上一步移动后9 != 8,即让数值小的指针往后移动一位即 index2++;

 ⑦上一步移动后9==9,则将9加入到res[index]中,双指针均向右移动一位,res的索引index也向后移动一位;

 ⑧上一步移动后9==9,则将9加入到res[index]中,双指针均向右移动一位,res的索引index也向后移动一位;

 ⑨这时候两个指针已经超出数组长度,截取res中的元素进行返回即可。

java代码:

 1 class Solution {
 2     public int[] intersect(int[] nums1, int[] nums2) {
 3         int m = nums1.length, n = nums2.length;
 4         int[] res = new int[Math.min(m, n)];
 5         Arrays.sort(nums1);
 6         Arrays.sort(nums2);
 7         int index = 0, index1 = 0, index2 = 0;
 8         while(index1 < m && index2 < n){
 9             if(nums1[index1] < nums2[index2]){
10                 index1++;
11             }else if(nums1[index1] > nums2[index2]){
12                 index2++;
13             }else{
14                 res[index] = nums1[index1];
15                 index++;
16                 index1++;
17                 index2++;
18             }
19         }
20         return Arrays.copyOfRange(res, 0, index);
21 
22     }
23 }

 python3代码:

 1 class Solution:
 2     def intersect(self, nums1: List[int], nums2: List[int]) -> List[int]:
 3         m, n = len(nums1), len(nums2)
 4         nums1.sort()
 5         nums2.sort()
 6         res = list()
 7         index1, index2 =  0, 0
 8         while index1 < m and index2 < n:
 9             if nums1[index1] < nums2[index2]:
10                 index1 += 1
11             elif nums1[index1] > nums2[index2]:
12                 index2 += 1
13             else:
14                 res.append(nums1[index1])
15                 index1 += 1
16                 index2 += 1
17         return res

 2023-05-09

补充:java-哈希表法:

 1 class Solution {
 2     public int[] intersect(int[] nums1, int[] nums2) {
 3         HashMap<Integer, Integer> map = new HashMap<>();
 4         int m = nums1.length, n = nums2.length;
 5         ArrayList<Integer> result = new ArrayList<>();
 6         for (Integer num : nums1){
 7             map.put(num, map.getOrDefault(num, 0) + 1);
 8         }
 9         for (Integer num : nums2){
10             if (map.containsKey(num)){
11                 result.add(num);
12                 map.put(num, map.get(num) - 1);
13                 if (map.get(num) == 0){
14                     map.remove(num);
15                 }
16             }
17         }
18         int[] ans = new int[result.size()];
19         int index = 0;
20         for (Integer temp : result){
21             ans[index++] = temp;
22         }
23         return ans;
24     }
25 }

 小知识:

 Arrays提供了一个copyOfRange方法进行数组复制。Arrays.copyOfRange()的格式如下:

copyOfRange(int[] original, int from, int to):

第一个参数表示源数组
第二个参数表示开始位置(取得到)
第三个参数表示结束位置(取不到)
posted on 2022-10-12 10:47  我不想一直当菜鸟  阅读(53)  评论(0编辑  收藏  举报