背景
对于下面两个有序set,找到出他们的交集:
list1=[2,3,8,10,12]
list2=[3,10,12,22]
方法
1、计数器
对两个list分别遍历一遍;
#include <iostream>
using namespace std;
int main(){
int a[]={2,3,8,10,12};int alen=5;
int b[]={3,10,12,22};int blen=4;
int count[1000];
for(int i=0;i<alen;i++){
count[a[i]]=1;
}
for(int i=0;i<blen;i++){
if(count[b[i]]==1)
cout<<b[i]<<" ";
}
return 0;
}
复杂度:O(m+n)
如果数组元素值过大,则空间申请过大。
2、启发于归并排序
这个过程有点像归并排序中的合并过程。
#include <iostream>
using namespace std;
int main(){
int a[]={2,3,8,10,12};int alen=5;
int b[]={3,10,12,22};int blen=4;
int aindex=0,bindex=0;
while(aindex<alen && bindex<blen){
int tmp1=a[aindex],tmp2=b[bindex];
if(tmp1<tmp2) aindex++;
else if(tmp1>tmp2) bindex++;
else{
cout<<tmp1<<" ";
aindex++;
bindex++;
}
}
return 0;
}
复杂度:O(m+n);
基于二分的合并
在合并的基础上,想到了一种优化方式,不一定要一个个的比较头元素,我们可以固定一定头元素,在另一个数组中找到不小于他的第一个元素,而之前的元素就可以直接省略。
// 用来找到a中第一个不比t小的数的位置
int UpBsearch(int s, int e, int t, VI& a) {
int l = s, r = e, mid;
while (l < r) {
mid = (l + r) / 2;
if (a[mid] >= t) {
r = mid;
}
else {
l = mid + 1;
}
}
return l;
}
// 找到a,b数组的共同元素,并且保存在common中。
void Graph::commonWeight(VI &a, VI& b, VI& common) {
int sizeA = a.size(), sizeB = b.size();
int l1 = 0, r1 = sizeA, l2 = 0, r2 = sizeB, ans = 0;
if (a[r1 - 1] < b[0] || b[r2 - 1] < a[0]) return ;
while (l1 < r1 && l2 < r2) {
if (a[l1] >= b[l2]) {
int pos = UpBsearch(l2, r2, a[l1], b);
if (pos < r2) {
if (a[l1] == b[pos]) {
l1++;
l2 = pos + 1;
if (head[b[pos]].del == 0) {
common.push_back(b[pos]);
}
}
else {
l2 = pos;
}
}
else break;
}
else {
int pos = UpBsearch(l1, r1, b[l2], a);
if (pos < r1) {
if (b[l2] == a[pos]) {
l2++;
l1 = pos + 1;
if (head[a[pos]].del == 0) {
common.push_back(a[pos]);
}
}
else {
l1 = pos;
}
}
else break;
}
}
}