【PAT】B1075 链表元素分类（25 分）

这道题算有点难，心目中理想的难度。
不能前怕狼后怕虎，一会担心超时，一会又担心内存过大，直接撸
将三部分分别保存到vector
有意思的在于输出
分别输出第一个的add和num
中间输出nextadd ，换行，add，num
最后输出尾元素的next为-1

#include<iostream>
#include<stdio.h>
#include<map>
#include<string>
#include<algorithm>
#include<vector>
using namespace std;
struct number{
	int add,num,next;
}arr[100000];//原始输入
int main() {
	int head,N,K;scanf("%d %d %d",&head,&N,&K);
	for(int i=0;i<N;i++){			//输入数据
		int add;scanf("%d",&add);
		arr[add].add=add;
		scanf("%d %d",&arr[add].num,&arr[add].next);
	}
	vector<int> v[3];				//存储三种类型
	while(head!=-1){				//遍历链表,把链表分成三份
		if(arr[head].num<0)			//负数
			v[0].push_back(head);
		else if(arr[head].num<=K)	//【0-K】
			v[1].push_back(head);
		else v[2].push_back(head);	//大于K
		head=arr[head].next;
	}
	int flag=0;
	for(int i=0;i<3;i++){
		for(int j=0;j<v[i].size();j++){
            if(flag==0){							//找出第一个，只输出地址和值
				printf("%05d %d ",arr[v[i][j]].add,arr[v[i][j]].num);
				flag=1;
            }
			else								//其他的输出上一行的下一个元素，换行，当前元素地址，当前值
				printf("%05d\n%05d %d ",arr[v[i][j]].add,arr[v[i][j]].add,arr[v[i][j]].num);
		}
	}
	printf("-1");				//尾指针的next为-1
	return 0;
}