7-52 两个有序链表序列的交集 (20 分)
已知两个非降序链表序列S1与S2,设计函数构造出S1与S2的交集新链表S3。
输入格式:
输入分两行,分别在每行给出由若干个正整数构成的非降序序列,用−1表示序列的结尾(−1不属于这个序列)。数字用空格间隔。
输出格式:
在一行中输出两个输入序列的交集序列,数字间用空格分开,结尾不能有多余空格;若新链表为空,输出NULL。
输入样例:
1 2 5 -1
2 4 5 8 10 -1
输出样例:
2 5
思路:当两个数组有相同的一对数时,消耗掉这一对数字
1 2 2 3 3 4
0 2 2 3 5 6
输出为 2 2 3
代码如下
#include<bits/stdc++.h>
using namespace std;
int main()
{
vector<int> a, b, c;
while (1)
{
int x;
cin >> x;
if (x == -1)
break;
a.push_back(x);
}
while (1)
{
int x;
cin >> x;
if (x == -1)
break;
b.push_back(x);
}
for (int i = 0, j = 0; i < a.size() && j < b.size(); )
{
if (a[i] < b[j])
i++;
else if (a[i] > b[j])
j++;
else
{
c.push_back(a[i]);
i++;
j++;
}
}
if (c.size() == 0)
cout << "NULL";
else
for (int i = 0; i < c.size(); i++)
{
if (i != 0)
cout << " ";
cout << c[i];
}
}