程式碼

#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);

	node nodeArray[MAX];
	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;
	}

	//original order
	vector<node> v;
	int now = start;
	for(int i=0; i<n; i++) {
		if(now!=-1) { //這個if判斷條件很關鍵！ 
			v.push_back(nodeArray[now]);
			now = nodeArray[now].next;
		}
	}
	
	//classify order
	vector<node> v2;
	for(int i=0; i<v.size(); i++){
		node tmp = v.at(i);
		if(tmp.data<0){
			v2.push_back(tmp);
		}
	}
	for(int i=0; i<v.size(); i++){
		node tmp = v.at(i);
		if(tmp.data>=0 && tmp.data<=k){
			v2.push_back(tmp);
		}
	}
	for(int i=0; i<v.size(); i++){
		node tmp = v.at(i);
		if(tmp.data>k){
			v2.push_back(tmp);
		}
	}

	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); //直接輸出的下一個節點的id，偷懶了。 
	}
	printf("%05d %d -1\n", v2.at(v2.size()-1).id, v2.at(v2.size()-1).data);

	return 0;
}
 

註解

1、此題的資料結構與乙級1025相同。 2、按連結串列的原始順序push到v中。 3、按題目給定的順序將v中的元素push到v2中。 4、按v2的順序輸出元素。主要是注意next.id如何輸出？這裡直接輸出的下一個節點的id，而不是真正改變了node.next值，偷懶了。後續看看如何優化？

結果