快速求出兩個數組的交集的算法,如果用循環遍歷的方法,其時間復雜度為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;
}
