PTA——L2-022 重排链表(链表)

---

一、题目

L2-022 重排链表
分数 25
例如：给定L为1→2→3→4→5→6，则输出应该为6→1→5→2→4→3。

输入格式：
每个输入包含1个测试用例。每个测试用例第1行给出第1个结点的地址和结点总个数，即正整数N (≤10^5)。结点的地址是5位非负整数，NULL地址用−1表示。

接下来有N行，每行格式为：
Address Data Next
其中Address是结点地址；Data是该结点保存的数据，为不超过10
5
的正整数；Next是下一结点的地址。题目保证给出的链表上至少有两个结点。

输出格式：
对每个测试用例，顺序输出重排后的结果链表，其上每个结点占一行，格式与输入相同。

二、题解

1.基本思路：

• 这一题和之前的链表去重那到题非常相似，我们可以用map来构建链表
• 构建一个正的链表，然后根据正的链表dfs到最后一个节点，构建反向链表。
• 这样我们就可取交替取两个链表的节点，可以用双指针遍历两个链表，当两个指针指向同一个值时，说明结果链表构建完成。

2.代码：

#include<bits/stdc++.h>
using namespace std;

#define IOS ios::sync_with_stdio(0),cin.tie(0),cout.tie(0)
#define endl "\n"
#define int long long
#define fi first
#define se second
const int N = 1e3+10, INF = 0x3f3f3f3f;
int h,n,head=-2,tar=-2;
unordered_map<int,int> mp,w; //mp初始链表，w节点权值 
unordered_map<int,int> List,ans; //List反向链表，答案链表 

void dfs(int node){
	if(node==-1) return; //递归边界 
	dfs(mp[node]); 
	List[head]=node; //构建反向链表 
	head=node;
	if(node==h) List[node]=-1; //反向链表的对后一个节点指向空地址-1 
}

void solve() {
	cin>>h>>n;
	mp[head]=h;  
	
	//构建初始链表 
	for(int i=1;i<=n;i++){
		int a,b,c;
		cin>>a>>b>>c;
		mp[a]=c; w[a]=b;
	}
	
	//构建反向链表 
	int node=mp[head];
	dfs(node); head=-2; //构建反向链表,记得重置head 
	
	//构建答案链表 
	int node1=List[head],node2=mp[head],flag=0;
	while(node1!=node2){
//		cout<<node1<<' '<<node2<<endl;
		if(!flag)
		  ans[node1]=node2,node1=List[node1];
		else
		  ans[node2]=node1,node2=mp[node2];;
		flag^=1;
	}
	ans[node1]=-1; //答案链表最后一个节点指向空 
	
	//输出答案 
	node=List[head];
	while(node!=-1){
		if(ans[node]!=-1)
		  printf("%05d %d %05d\n",node,w[node],ans[node]);
		else
		  printf("%05d %d %d\n",node,w[node],ans[node]);
		node=ans[node];
	}
}

signed main() {
//	IOS;
	int T=1;
//	cin>>T;
	while(T--) {
		solve();
	}
	return 0;    
}
/*
00100 6
00000 4 99999
00100 1 12309
68237 6 -1
33218 3 00000
99999 5 68237
12309 2 33218

68237 6 00100
00100 1 99999
99999 5 12309
12309 2 00000
00000 4 33218
33218 3 -1
*/