一,问题描述
给定两个整型数组,假设一个长度为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); } }