PAT 1133. Splitting A Linked List (25) 链表的拆分和合并

/*************************
题意:
拆分1个链表
把链表划分成 <0 ， 0-k ， 大于k三个链表（同一链表内相对顺序不变）
并对其进行重新连接。
并输出。
************************/
/***********************
解题思路：
搞3个vector，然后收集对应的节点
最后再合并，输出，即可。
*************************/
/***********************
注意
存无向图时，一定不要遗漏双向的存储！！
*********************/
#include<iostream>
#include<stdio.h>
#include<string>
#include<vector>
#include<queue>
#include<stdlib.h>
#include<algorithm>
#include<string.h>
#include<stack>
#include<map>
#include<set>
#include<unordered_map>
using namespace std;
#define M 100005
#define INF 0x7ffffff

struct Node{
int addr;
int Next;
int data;
};
Node node[M];

int main(){
int n, i, m;
int head,k;
cin>>head>>n>>k;
int s,e,data;
while(n--){
scanf("%d%d%d",&s ,&data, &e);
node[s].addr = s;
node[s].Next = e;
node[s].data = data;
}
vector<Node> v1,v2,v3;
int p = head;
while(p != -1){
if(node[p].data <0 )
v1.push_back(node[p]);
else if(node[p].data <= k)
v2.push_back(node[p]);
else v3.push_back(node[p]);
p = node[p].Next;
}

v1.insert(v1.end(),v2.begin(),v2.end());
v1.insert(v1.end(),v3.begin(),v3.end());

for(i=0;i<v1.size();i++){
if(i != v1.size()-1)
printf("%05d %d %05d\n",v1[i].addr,v1[i].data,v1[i+1].addr);
else printf("%05d %d -1\n",v1[i].addr,v1[i].data);
}

return 0;
}