快速求出两个数组的交集的算法,如果用循环遍历的方法,其时间复杂度为O(N^N),在面试中一般不考虑这种方法。
这里提供一种快速算法,算法实现步骤如下:
1. 找到arr1的最大数max,创建一个max+1大小的数组result。
2. 以arr1中的值作为result的下标,该索引处的值+1。
3. 在result数组中遍历arr2中的值为下标处的值,如果该索引值不为0,则代表该值是两者的交集,保留。
程序实现如下:
/**
* 求两个数组的交集
* @param arr1
* @param arr2
* @return 数组交集
*/
private static int[] intersection(int[] arr1,int[] arr2){
int[] result = null;
ArrayList<Integer> list = new ArrayList<>();
Arrays.sort(arr1);
result = new int[arr1[arr1.length-1]+1];
for (int i = 0; i < arr1.length; i++) {
result[arr1[i]]++;
}
for (int i = 0; i < arr2.length; i++) {
if (arr2[i]<=result.length&&result[arr2[i]]!=0) {
list.add(arr2[i]);
}
}
int[] returnarr = new int[list.size()];
for (int i = 0; i < list.size(); i++) {
returnarr[i] = list.get(i);
}
return returnarr;
}