已知两个非降序链表序列S1与S2,设计函数构造出S1与S2的交集新链表S3。
输入格式:
输入分两行,分别在每行给出由若干个正整数构成的非降序序列,用−1表示序列的结尾(−1不属于这个序列)。数字用空格间隔。
输出格式:
在一行中输出两个输入序列的交集序列,数字间用空格分开,结尾不能有多余空格;若新链表为空,输出NULL
。
输入样例:
1 2 5 -1
2 4 5 8 10 -1
输出样例:
2 5
题解
这题的关键就在于怎么找交集元素。
我来说说我的尝试。
由于链表掌握不太好(看着心累),故试图使用数组来求解。
注:下面分析时,为了方便考虑,就将a数组和b数组大小都看作n,实际上这并不一定,效率会受两个数组大小影响。不过此处还是简单想吧。
1.直接暴力
for(int i=1;i<=top1;i++)
for(int j=1;j<=top2;j++)
if(a[i]==b[j]) c[++top] = a[i];
显然时间复杂度为O(n*2),样例还行,只是最后一个点被卡住了。
于是尝试转变方法。
2.二分法查找
main里的部分:
for(int i=1;i<=top1;i++) //遍历数组a
{
if(ha(i)==true) //判断数组下标i对应a中元素在b中
{
c[++top3] = a[i];
}
}
写了一个判断函数ha
bool ha(int i)
{
item = a[i]; //当前下标对应值
long long low=1,high=top2,mid;
if(item<b[1]||item>b[top2]) return false; //试图细微减少时间,但价值不大
while(low<=high) //标准二分查找
{
mid = (low+high)/2;
if(item<b[mid])
{
high = mid-1;
}
else if(item>b[mid])
{
low = mid+1;
}
else
return true;
}
return false;
}
时间复杂度为O(n*log n) 。还是没过最后一个点。
3.flag数组标记法
longlong flag[1000005];
for(int i=1;i<=top1;i++)
{
flag[a[i]]=1; //标记a中元素出现情况
}
for(int i=1;i<=top2;i++) //遍历b数组
{
if(flag[b[i]]==1) //b[i]就对应b中元素,如果它在flag数组中已然为1,自然是a的功劳
c[++top3] = b[i]; //说明数据为二者共有
}
这应该是O(n)吧。不过还是没有过。
4.你追我赶的双指针移动法
int p1=1,p2=1; //两个数组停留于开头的移动指针
while(p1<=top1&&p2<=top2) //执行条件
{
if(a[p1]==b[p2]) //相等,齐头并进
{
c[++top3] = a[p1];
p1++;
p2++;
}
else if(a[p1]<b[p2]) p1++; //哪个数小,它的指针就下移使数据变大(利用了有序)
else p2++;
}
复杂度为O(n)。
相比标记法,当复杂度的量级相同时,就要考虑n的系数。标记法近于2n,而此法却近于n,故此法更为优越。
前面都是可以处理无序数组求交集的算法,而此算法确是针对有序的,要求严格了,当真正可用时,时间自然也剩下来了。
所以过了。
ac代码
#include<bits/stdc++.h>
using namespace std;
long long a[1000005],b[1000005],c[1000005];
long long top1=0,top2=0,top3=0; //使用了栈的思想来装数据
long long item;
int main()
{
while(cin>>item) //存a数组
{
if(item==-1) break;
else
a[++top1] = item;
}
while(cin>>item) //存b数组
{
if(item==-1) break;
else
b[++top2] = item;
}
int p1=1,p2=1; //双指针移动法
while(p1<=top1&&p2<=top2)
{
if(a[p1]==b[p2])
{
c[++top3] = a[p1];
p1++;
p2++;
}
else if(a[p1]<b[p2]) p1++;
else p2++;
}
for(int i=1;i<=top3;i++) //输出交集数组c
{
if(i==1) cout<<c[i];
else cout<<" "<<c[i];
}
if(top3==0) cout<<"NULL";
return 0;
}