PAT乙级 1075 链表元素分类 （25 分）

给定一个单链表，请编写程序将链表元素进行分类排列，使得所有负值元素都排在非负值元素的前面，而 [0, K] 区间内的元素都排在大于 K 的元素前面。但每一类内部元素的顺序是不能改变的。例如：给定链表为 18→7→-4→0→5→-6→10→11→-2，K 为 10，则输出应该为 -4→-6→-2→7→0→5→10→18→11。

输入格式：

每个输入包含一个测试用例。每个测试用例第 1 行给出：第 1 个结点的地址；结点总个数，即正整数N (≤10⁵)；以及正整数K (≤10³)。结点的地址是 5 位非负整数，NULL 地址用 −1 表示。

接下来有 N 行，每行格式为：

Address Data Next

其中 Address 是结点地址；Data 是该结点保存的数据，为 [−10⁵,10⁵] 区间内的整数；Next 是下一结点的地址。题目保证给出的链表不为空。

输出格式：

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

输入样例：

00100 9 10
23333 10 27777
00000 0 99999
00100 18 12309
68237 -6 23333
33218 -4 00000
48652 -2 -1
99999 5 68237
27777 11 48652
12309 7 33218

输出样例：

33218 -4 68237
68237 -6 48652
48652 -2 12309
12309 7 00000
00000 0 99999
99999 5 23333
23333 10 00100
00100 18 27777
27777 11 -1

思路：

先将链表按地址排序，然后分三次再次按题目要求筛选并赋值。这种方法的结果正确但测试点4、5运行超时。

代码：

#include<stdio.h>
int main(){
	int first_address,N,K;
	scanf("%d %d %d",&first_address,&N,&K);
	int Address[N],Data[N],Next[N];
	for(int i=0;i<N;++i){
		scanf("%d %d %d",&Address[i],&Data[i],&Next[i]);
	}
	int k=0,temp=first_address,address[N],data[N],next[N];
	while(k<N){
		for(int i=0;i<N;++i){
			if(Address[i]==temp){
				address[k]=temp;
				data[k]=Data[i];
				next[k++]=Next[i];
				temp=Next[i];
			}
		}
	}
	k=0;
	int a[N],d[N],n[N];
	for(int i=0;i<N;++i){
		if(data[i]<0){
			a[k]=address[i];
			d[k++]=data[i];
		}
	}
	for(int i=0;i<N;++i){
		if(data[i]>=0&&data[i]<=K){
			a[k]=address[i];
			d[k++]=data[i];
		}
	}
	for(int i=0;i<N;++i){
		if(data[i]>K){
			a[k]=address[i];
			d[k++]=data[i];
		}
	}
	for(int i=0;i<N;++i){
		n[i]=a[i+1];
		if(i==N-1) n[N-1]=-1;
	}
	for(int i=0;i<N-1;++i){
		printf("%05d %d %05d\n",a[i],d[i],n[i]);
	}
	printf("%05d %d %d\n",a[N-1],d[N-1],n[N-1]);
	return 0;
}

第一次修改：

优化了链表的串联过程，同时要注意，有些结点是不在链表上的。

#include<stdio.h>
typedef struct{
	int data,next;
}linklist;
int main(){
	int first_address,N,K,Address;;
	scanf("%d %d %d",&first_address,&N,&K);
	linklist l[100000];
	for(int i=0;i<N;++i){
		scanf("%d",&Address);
		scanf("%d %d",&l[Address].data,&l[Address].next);
	}
	int k=0,temp=first_address,address[N],data[N],next[N];
	while(temp!=-1){
		address[k]=temp;
		data[k]=l[temp].data;
		next[k++]=l[temp].next;
		temp=l[temp].next;
	}
	N=k,k=0;
	int a[N],d[N],n[N];
	for(int i=0;i<N;++i){
		if(data[i]<0){
			a[k]=address[i];
			d[k++]=data[i];
		}
	}
	for(int i=0;i<N;++i){
		if(data[i]>=0&&data[i]<=K){
			a[k]=address[i];
			d[k++]=data[i];
		}
	}
	for(int i=0;i<N;++i){
		if(data[i]>K){
			a[k]=address[i];
			d[k++]=data[i];
		}
	}
	for(int i=0;i<N;++i){
		n[i]=a[i+1];
		if(i==N-1) n[i]=-1;
	}
	for(int i=0;i<N-1;++i){
		printf("%05d %d %05d\n",a[i],d[i],n[i]);
	}
	printf("%05d %d %d\n",a[N-1],d[N-1],n[N-1]);
	return 0;
}