package com.leetcode.leetcode.setmap;

import com.sun.javafx.scene.control.skin.VirtualFlow.ArrayLinkedList;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;

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

    示例 1:

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

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

    输出结果中每个元素出现的次数,应与元素在两个数组中出现的次数一致。
    我们可以不考虑输出结果的顺序。
    进阶:

    如果给定的数组已经排好序呢?你将如何优化你的算法?
    如果 nums1 的大小比 nums2 小很多,哪种方法更优?
    如果 nums2 的元素存储在磁盘上,磁盘内存是有限的,并且你不能一次加载所有的元素到内存中,你该怎么办?
*/
/**
用哈希表记录数组1中的数和其对应频次。遍历数组2,如果当前索引的数在哈希表中存在,则把它丢进结果list中,并把它在哈希表中的频次-1,如果频次为0就移除它。完成遍历后,list中就是我们希望得到的结果。
*/
public class Intersection2 {

  public int[] intersect2(int[] nums1, int[] nums2) {
    HashMap<Integer, Integer> map = new HashMap<>();
    List<Integer> list = new ArrayList<>();
    for(int num : nums1) {
       map.put(num, map.getOrDefault(num,0) + 1);
    }
    for(int num : nums2) {
      if(map.containsKey(num)) {
        map.put(num, map.get(num) - 1);
        if(map.get(num) == 0) map.remove(num);
        list.add(num);
      }
    }
    int[] res = new int[list.size()];
    for(int i = 0; i < list.size(); i++) {
      res[i] = list.get(i);
    }
    return res;
  }

}

github同步

You Might Also Like

No Comments

Leave a Reply