L2-022 重排链表 (25 分) 数组模拟链表

L2-022 重排链表 (25 分)

给定一个单链表 L
1
​
→L
2
​
→⋯→L
n−1
​
→L
n
​
，请编写程序将链表重新排列为 L
n
​
→L
1
​
→L
n−1
​
→L
2
​
→⋯。例如：给定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是下一结点的地址。题目保证给出的链表上至少有两个结点。

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

输入样例：
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

一开始第四个样例点总是过不去，只能得22，后来发现是有多余的结点，即链表中实际的结点个数可能小于n,题目的说明也跟谜语人一样，感觉很不严谨。

#include <bits/stdc++.h>

using namespace std;

int w[100005],ne[100005],m[100005];
int s,n;

int main()
{
	cin>>s>>n;
	for(int i=0;i<n;++i)
	{
		int ad,x,nad;
		cin>>ad>>x>>nad;
		w[ad]=x;
		ne[ad]=nad;
	}
	int p=1,nn=0;
	for(int i=s;i!=-1;i=ne[i])
	{
		nn++;
		m[p++]=i;
	}
	int a,b,x,last=-1;
	if(nn<n)n=nn;
	a=1;b=n;
	int t=0;
	while(t++<n)
	{
		if(t%2==1)
		{
			x=b;b--;
		}
		else
		{
			x=a;a++;
		}
		if(last==-1)
		{
			last=m[x];
			s=last;
			continue;
		}
		ne[last]=m[x];
		last=m[x];
		if(t==n)ne[last]=-1;
	}
	for(int i=s;i!=-1;i=ne[i])
	{
		if(ne[i]!=-1)
			printf("%05d %d %05d\n",i,w[i],ne[i]);
		else
			printf("%05d %d %d\n",i,w[i],ne[i]);
	}
	return 0;
}

更新：更加易懂的代码：

建立双向链表，重排时相当于新建一个链表，每次分别从左右两端取节点插入到新链表的表尾

#include <bits/stdc++.h>
using namespace std;
const int N=1000005;
int s,n,d,s2,p;
int w[N],ne[N],fo[N];
int main(){
	cin>>s>>n;
	for(int i=0;i<n;++i)
	{
		int x,y,z;cin>>x>>y>>z;
		w[x]=y;
		ne[x]=z;
		if(z!=-1)fo[z]=x;
		if(z==-1)d=x;
	}
	int l=s,r=fo[d];
	s2=d;p=s2;
	for(int i=0;;++i)
	{
		if(i%2==0){
			ne[p]=l;
			l=ne[l];
		}
		else {
			ne[p]=r;
			r=fo[r];
		}
		p=ne[p];
		if(l==ne[r]){//r==fo[l]
			ne[p]=-1;break;
		} 
	}
	for(int i=s2;i!=-1;i=ne[i])
	{
		if(ne[i]!=-1)printf("%05d %d %05d\n",i,w[i],ne[i]);
		else printf("%05d %d %d\n",i,w[i],ne[i]);
	}
	return 0;
}

二刷：

#include<bits/stdc++.h>
using namespace std;
const int N=1000005;
int w[N],ne[N],fo[N],s,d,s2,d2;
int main()
{
	int s,n;cin>>s>>n;
	for(int i=0;i<n;++i){
		int x,y,z;cin>>x>>y>>z;
		w[x]=y;
		ne[x]=z;
		if(z!=-1)fo[z]=x;
		if(z==-1)d=x;
	} 
	int l=s,r=d;
	for(int i=0;;++i){
		if(i%2==0){
			if(i==0)s2=d2=r;
			else {
				ne[d2]=r;
				d2=r;
			}
			r=fo[r];
		}
		else{
			ne[d2]=l;
			d2=l;
			l=ne[l];
		}
		if(l==ne[r]){
			ne[d2]=-1;
			break;
		}
	}
	for(int i=s2;i!=-1;i=ne[i]){
		if(ne[i]!=-1)printf("%05d %d %05d\n",i,w[i],ne[i]);
		else printf("%05d %d %d\n",i,w[i],ne[i]);
	}
	return 0;
}