【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;
}