查找两个整型数组的公共元素

一，问题描述

给定两个整型数组，假设一个长度为M，另一个长度为N。请找出(打印出)这两个数组中的公共元素。

 

二，算法分析

有两种思路求解这个问题。

①使用一个HashSet保存第一个数组中的所有元素，然后遍历第二个数组中的每个元素，判断该元素是否在HashSet中。如果在，就表明这个元素是公共元素。

此方法的时间复杂度为O(M+N)，空间复杂度为O(M)[假设第一个数组长度为M，保存在HashSet中]。

 

②首先对两个数组进行排序。然后分别设置两个指针 i, j   初始时，分别指向两个数组的第一个元素，依次比较这两个指针指向的元素，指向较小的元素的那个指针向后移一位。如果，两个元素相同，则两个指针同时向后移一位。

此方法，首先需要对数组排序，时间复杂度为O(MlogM + NlogN)，空间复杂度为O(1)---不考虑排序算法的空间复杂度。

 

三，完整代码实现

import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;

public class CommonElement {
    
    //排序时间复杂度为O(NlogN+MlogM),排序之后寻找公共元素的时间复杂度为O(M+N)
    public static void printCommonEle(int[] arr1, int[] arr2){
        if(arr1 == null || arr2 == null || arr1.length == 0 || arr2.length == 0)
            return;
        
        Arrays.sort(arr1);
        Arrays.sort(arr2);
        
        int i = 0, j = 0;
        while(i < arr1.length && j < arr2.length)
        {
            if(arr1[i] > arr2[j])
                j++;
            else if(arr1[i] < arr2[j])
                i++;
            else{
                System.out.print(arr1[i] + " ");
                i++;
                j++;
            }
        }
    }
    
    //时间复杂度为O(M),空间复杂度为O(N)
    public static void findCommon(int[] arr1, int[] arr2){
        if(arr1 == null || arr2 == null || arr1.length == 0 || arr2.length == 0)
            return;
        
        Set<Integer> intSets = new HashSet<Integer>();
        
        for (int i : arr1)
            intSets.add(i);
        
        for(int i : arr2)
            if(intSets.contains(i))
                System.out.print(i + " ");
    }
    
    public static void main(String[] args) {
        int[] arr1 = {3,5,1,7,8};
        int[] arr2 = {4,7,3,8,5,2,9,0};

        printCommonEle(arr1, arr2);
        System.out.println();
        findCommon(arr1, arr2);
    }
}

View Code