PAT-乙/甲-1075/1133 链表元素分类

 Splitting A Linked List 

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

实验思路

根据题意，一共分三个区间，分别为

然后在各区间内按照原顺序排列.

1. 按链表的原始顺序push到v中
2. 按题目给定的顺序将v中的元素push到v2中。
   3.按v2的顺序输出元素。

#include <iostream>
#include <vector>

using namespace std;

const int MAX = 100000;

struct node {
	int id;//结点地址 
	int data;//数据域 
	int next; // 下个结点的地址
};

int main() {

	int start, n, k;
	scanf("%d %d %d", &start, &n, &k);// 第 1 个结点的地址；结点总个数n ；以及正整数k (≤103)。 

	node nodeArray[MAX];//定义一个node类型的结点nodeArray 
	for(int i=0; i<n; i++) {
		int id, data, next;
		scanf("%d %d %d", &id, &data, &next);
		nodeArray[id].id = id; //初始化定义 
		nodeArray[id].data = data;
		nodeArray[id].next = next;
	}

	//按链表的原始顺序push到v中
	vector<node> v;//初始化一个node型的链表 v
	int now = start; 
	for(int i=0; i<n; i++) {
		if(now!=-1) { 
			v.push_back(nodeArray[now]);  
			now = nodeArray[now].next; 
		}
	}
	
	//按题目给定的顺序将v中的元素push到v2中。
	vector<node> v2;
	for(int i=0; i<v.size(); i++){
		node tmp = v.at(i);
		if(tmp.data<0){  //若为负数，把它们按原顺序push到v2 
			v2.push_back(tmp);
		}
	}
	for(int i=0; i<v.size(); i++){
		node tmp = v.at(i);
		if(tmp.data>=0 && tmp.data<=k){ //若为正数，且小于K，把他们按原顺序push到v2中 
			v2.push_back(tmp);
		}
	}
	for(int i=0; i<v.size(); i++){
		node tmp = v.at(i);
		if(tmp.data>k){  //若为正数，且大于K，把他们按原顺序push到v2中 
			v2.push_back(tmp);
		}
	}
     
    printf("以下为输出结果\n");
	for(int i=0; i<v2.size()-1; i++) {
		node t = v2.at(i);
		node tNext = v2.at(i+1);
		printf("%05d %d %05d\n", t.id, t.data, tNext.id); //输出地址，数据，下一个结点地址 
	}
	printf("%05d %d -1\n", v2.at(v2.size()-1).id, v2.at(v2.size()-1).data);//最后一个结点不能用直接输出下个结点地址这种偷懒的方法，单独输出尾节点的数据域和下一个结点的地址为NULL 

	return 0;
}