已经排好序了,所以我们可以借此同时对三个数组进行搜索来找到共同元素
下面是java实现代码:
- import java.util.HashSet;
- import java.util.Set;
- public class threeListComm {
-
- /**
- * 3 soted array, find common elements
- */
- static int [] a = {1,3,4,5,6,7,9};
- static int [] b = {1,5,8,9,11};
- static int [] c = {2,3,5,6};
- static Set<Integer> s = new HashSet<Integer>();
- public static void main(String[] args) {
- for(int i=0,j=0,k=0; i<6 && j<5 && k<4;){
- int [] status = min_status(a[i], b[j], c[k]);
- if(status[0]+status[1]+status[2]==3)
- s.add(a[i]);
- i += status[0];
- j += status[1];
- k += status[2];
- }
- System.out.println(s);
- }
- public static int [] min_status(int a, int b ,int c){
- int [] status = {0,0,0};
- int min = c<((a<b)?a:b)?c:((a<b)?a:b);
- if(a == min) status[0] = 1;
- if(b == min) status[1] = 1;
- if(c == min) status[2] = 1;
- return status;
- }
- }
大家可以看到,我实现了一个min_status函数,他的主要作用就是找到当前三个元素中的最小值,并且决定将最小的元素下标+1,来进行该数组的下一步搜索。然而跟两个数组的情形不同的是:有时候最小的元素不一定是一个,可能是两个或者三个,这样我们需要用一个status数组来标记各个数组下一步是否应该+1。
同样,如果要求从N个已排序数组中找到common elements,用这种方式同样可以保证时间复杂度为o(n)。不过需要经历一个寻找最小值和设置状态位的过程,这两个过程可以优化,优化方法就是对n个当前元素进行排序,找到m(m<<n)个最小的,仅把这m个元素的状态为修改为1即可。当m==n时,该元素就是我们所要的common element。