给定多个数组,请输出所有数组的交集。例如有三个数组 [1,2, 3, 4, 5],[1, 4, 5, 7],[2, 4, 5, 8],则输出交集为[4,5]
利用遍历+二分查找的方法
选定一个一个数组,假设选第一个数组,我们依次遍历数组,然后在其他几个数组中查找每一个元素,如果其他数组都有该元素,则输出,否则继续查找下一个元素。
这里我们可以优化点是:用一个数组来维护查找的下标,充分利用有序数组的特点,对于以后数据量庞大的输入十分有效,就是每次查找我们将找到的位置保存在该数组中,下次查找的数一定是大于等于该位置的数,所以可以该位置之前的数据不用再重复查找,代码如下:
int BinaryFind(int val, vector<int>&vec, int idx) { //二分查找,idx是查找的起点,提高效率
int lf = idx, rg = vec.size() - 1;
while (lf <= rg) {
int mid = (rg + lf) / 2;
if (vec[mid] == val) return mid;
else if (vec[mid] < val)
lf = mid + 1;
else
rg = mid - 1;
}
return -1;
}
void getValue(vector<vector<int>>& num) {
int n = num[0].size();
vector<int>vec(num[0]);
vector<int>index(n, 0); //保存在每个数组中查找到的元素的位置
bool flag;
for (int i = 0; i < n; i++) {
flag = true;
for (int j = 1; j < num.size(); j++) {
if (index[j] > num[j].size()) return;
int location = BinaryFind(vec[i], num[j], index[j]);
if (location != -1)
index[j] = location; //将在该数组中查找到的下标索引保存
else {
flag = false; //该元素不是交集
break;
//一旦有一个数组没有查到,说明vec[i]不是交集,跳出循环,不用继续查后面的数组
}
}
if (flag) cout << vec[i] << " "; //在所有数组中都查找完了,flag还为true,输出该元素
}
}