PAT 乙级 7-5 链表合并(博&人&)

PAT1078 字符串压缩与解压 (c++实现)—乙级真题

7-5 链表合并给定两个单链表 L1=a1 →a2 →⋯→ an−1→a​n ​​ 和 L2=b1→b2→⋯→bm−1→bm ​​ 。如果 n≥2m，你的任务是将比较短的那个链表逆序，然后将之并入比较长的那个链表，得到一个形如 a1→a2→b​m→a3→a4→bm−1 ​​ ⋯ 的结果。例如给定两个链表分别为 6→7 和 1→2→3→4→5，你应该输出 1→2→7→3→4→6→5。

输入格式:

输入首先在第一行中给出两个链表 L ​1 ​​ 和 L ​2 ​​ 的头结点的地址，以及正整数 N (≤10 ​5 ​​ )，即给定的结点总数。一个结点的地址是一个 5 位数的非负整数，空地址 NULL 用 -1 表示。随后 N 行，每行按以下格式给出一个结点的信息：Address Data Next其中 Address 是结点的地址，Data 是不超过10的5次方的正整数，Next 是下一个结点的地址。题目保证没有空链表，并且较长的链表至少是较短链表的两倍长。

输出格式:

按顺序输出结果链表，每个结点占一行，格式与输入相同。

分析 :

先将每组数据存到结构体数组中，然后通过输出的两个头结点，分别找到两个链表，分别放到两个vectort1,t2中，接着判断哪个向量的元素少，然后定义一个新的vectort3，进行合并。注意：合并的时候，要将插入的Next的值变为前一组数据的Next的值，合并完后，将下一组数据的Address赋值给上一组的Next(除了最后一组数据不用这样赋值) （Address1,Next1)---(Address2,Next2)这里的Next1==Address2 如果要将一组数据插入,如（Address3,Next3）,需要将Next1=Address3;Next3=Next1;还需要注意短链表的长度L1*2>=长链表的长度L2时，短链表的最后几条数据需要直接插在t3的后面

代码如下 :

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
struct List{
	int Address;
	int Data;
	int Next;
}list[100000];
vector<struct List>t1;
vector<struct List>t2;
vector<struct List>t3;
int main(){
	int a1,b1,N;
	cin>>a1>>b1>>N;
	for(int i=0;i<N;++i){
		struct List t;
		cin>>t.Address>>t.Data>>t.Next;
		list[t.Address]=t;
	}
	while(a1!=-1){
		t1.push_back(list[a1]);
		a1=list[a1].Next;
	}
	while(b1!=-1){
		t2.push_back(list[b1]);
		b1=list[b1].Next;
	}
	int sum=0;
	if(t1.size()<t2.size()){
		reverse(t1.begin(),t1.end());
		int flag=0;
		for(int i=0;i<t2.size();++i){
			if(flag!=0&&(flag%2==0)){
				if(sum<t1.size()){
					t1[sum].Next=t3[flag-1].Next;
					t3.push_back(t1[sum]);
					sum++;	
				}
				flag=0;
				i--;
			}else{
				t3.push_back(t2[i]);
				flag++;
			}
		}
		while(sum<t1.size()){
			t3.push_back(t1[sum]);
			sum++;
		} 
	}else{
		reverse(t2.begin(),t2.end());
		
		int flag=0;
		for(int i=0;i<t1.size();++i){
			if(flag!=0&&(flag%2==0)){
				if(sum<t2.size()){
					t2[sum].Next=t3[flag-1].Next;
					t3.push_back(t2[sum]);
					sum++;
				}
				flag=0;
				i--;
			}else{
				t3.push_back(t1[i]);
				flag++;
			}
		}
		while(sum<t2.size()){
			t3.push_back(t2[sum]);
			sum++;
		} 
	}
	int temp;
	for(int i=0;i<t3.size()-1;++i){
		t3[i].Next=t3[i+1].Address;
	}
	for(int i=0;i<t3.size();++i){
		if(i==t3.size()-1)
		printf("%05d %d -1",t3[i].Address,t3[i].Data);
		else
		printf("%05d %d %05d\n",t3[i].Address,t3[i].Data,t3[i].Next);
	}
	return 0;
}

运行截图 :