Intersection of Two Arrays II--求交集

题意:
求两集合的交集

本来第一个想法是:用两个map存两个数组的数据,key存放下标i,value存放数组值,再将其中一个map1迭代与看另一个map2中是否包含相同的value数据,有相同的则去除map2中的这条数据(这里的删除操作就有一次循环了),再将这条数据放入另建的一个list中,继续迭代。这种方法确实是效率低下,光存放数据的两个map就要占很大的内存空间了,要是两个数组数据量比较大都会内存溢出了,而且时间复杂度也是n*n,比较大了。

还是下面这种解法比较靠谱:
先将数组一存进一个map中,key存放数组数据值,value存放该数据在数组中出现的次数,循环数组二,判断数组二中的每个数据在map的key中是否存在,存在则将该数据添加进一个list中,对应key的value值减1,当value值为1时,就不减1直接remove掉该数据。这样的解法空间复杂度直接减少了一半,时间复杂度也降低为了n。解法如下:

public int[] intersect(int[] nums1, int[] nums2) {
        int i = 0, len1 = nums1.length, len2 = nums2.length;
        List<Integer> list = new ArrayList<>();
        Map<Integer, Integer> map = new HashMap<>();
        for (i = 0; i < len1; i++) {
            if (map.containsKey(nums1[i])) {
                map.put(nums1[i], map.get(nums1[i]) + 1);
            } else {
                map.put(nums1[i], 1);
            }
        }

        for (i = 0; i < len2; i++) {
            if (map.containsKey(nums2[i])) {
                list.add(nums2[i]);
                if (map.get(nums2[i]) > 1) {
                    map.put(nums2[i], map.get(nums2[i]) - 1);
                } else {
                    map.remove(nums2[i]);
                }

            }
        }

        int size = list.size();
        int[] intersection = new int[size];
        for (i = 0; i < size; i++) {
            intersection[i] = list.get(i);
        }
        return intersection;

    }

猜你喜欢

转载自blog.csdn.net/yabaj/article/details/77164709