PAT Basic 1075 链表元素分类

PAT Basic 1075 链表元素分类

题设

给定一个单链表，请编写程序将链表元素进行分类排列，使得所有负值元素都排在非负值元素的前面，而 [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 (≤105)；以及正整数K (≤103)。结点的地址是 5 位非负整数，NULL 地址用 −1 表示。

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

Address Data Next

其中 Address 是结点地址；Data 是该结点保存的数据，为 [−105,105] 区间内的整数；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

由题意可知链表元素应分成三类，

1. 负数
2. 节点的值在区间 [0,K]，注意闭区间
3. 节点的值大于K

还要注意要保持三个部分内部的顺序。

可见只要分出来三类，然后把三部分按照顺序输出就行了，链表的存储方法和输出方法和 1025反换链表类似。

可以一次遍历，分三类把地址分别存储到三个数组，最后再分别输出。也可以分三次遍历，每次都把同一类元素存到同一个数组里，最后直接顺序输出。前者省时间，后者省空间，差别不大。但是第一种写法要注意三个部分有可能某个部分为空，所以衔接处的处理比较麻烦，容易错。

#include <stdio.h>
#include <stdlib.h>

int main(){
	int head,N,K;
    int list[100001]={-1},data[100001]={0},output[100001]={-1},cnt=0;
	scanf("%d %d %d",&head,&N,&K);
	for(int i=0;i<N;i++){
		int addr,d,next;
		scanf("%d %d %d",&addr,&d,&next);
		list[addr] = next;
		data[addr] = d;
	}
	for(int i=head;i != -1;i = list[i]){
		if(data[i] < 0) output[cnt++] = i;
	}
	for(int i=head;i != -1;i = list[i]){
		if(data[i]>=0 && data[i] <= K) output[cnt++] = i;
	}
	for(int i=head;i != -1;i = list[i]){
		if(data[i] > K) output[cnt++] = i;
	}	
	for(int i=0;i < cnt;i++){
		printf("%05d %d ",output[i],data[output[i]]);
		if(i != cnt-1){
			printf("%05d\n",output[i+1]);
		}else{
			printf("-1");
		}
	}
	return 0;		
}

History

一开始做的一次遍历分三类的写法，整体的变量太多，逻辑比较混乱

#include <stdio.h>
#include <stdlib.h>

int main(){
	int head,N,K,list[100001]={-1},data[100001]={0};
	int cls1[100001]={-1},cls2[100001]={-1},cls3[100001]={-1},cnt1=0,cnt2=0,cnt3=0;
	scanf("%d %d %d",&head,&N,&K);
	for(int i=0;i<N;i++){
		int addr,d,next;
		scanf("%d %d %d",&addr,&d,&next);
		list[addr] = next;
		data[addr] = d;
	}
	for(int i=head;i!=-1;i = list[i]){
		if(data[i] < 0){
			cls1[cnt1++] = i;
		}else if(data[i] <= K){
			cls2[cnt2++] = i;
		}else{
			cls3[cnt3++] = i;
		}
	}
	
	for(int i=0;i<cnt1;i++){
		printf("%05d %d ",cls1[i],data[cls1[i]]);
		if(i != cnt1-1){
			printf("%05d\n",cls1[i+1]);
		}else{
			if(cnt2 != 0){
				printf("%05d\n",cls2[0]);
			}else if(cnt3 != 0){
				printf("%05d\n",cls3[0]);
			}else{
				printf("-1");
			}
		}
	}
	for(int i=0;i<cnt2;i++){
		printf("%05d %d ",cls2[i],data[cls2[i]]);
		if(i != cnt2-1){
			printf("%05d\n",cls2[i+1]);
		}else{
			if(cnt3 != 0){
				printf("%05d\n",cls3[0]);
			}else{
				printf("-1");
			}
		}
	}
	for(int i=0;i<cnt3;i++){
		printf("%05d %d ",cls3[i],data[cls3[i]]);
		if(i != cnt3-1){
			printf("%05d\n",cls3[i+1]);
		}else{
			printf("-1");
		}
	}
	return 0;		
}