1075 链表元素分类

#include<bits/stdc++.h>
using namespace std;
struct node{
    int data,next;
}ldst[100000];
vector<int> v[3];
int main(){
    int n,k,sum=0,temp,first;
    cin>>first>>n>>k;
    for(int i=0;i<n;i++){
        cin>>temp;
        cin>>ldst[temp].data>>ldst[temp].next;
    }
    int p=first;
    while(p!=-1){
        int data=ldst[p].data;
        if(data<0) v[0].push_back(p);
        else if(data>=0&&data<=k) v[1].push_back(p);
        else v[2].push_back(p); 
        p=ldst[p].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 ",v[i][j],ldst[v[i][j]].data);
                flag=1;
            }else{
                printf("%05d\n%05d %d ",v[i][j],v[i][j],ldst[v[i][j]].data);
            }
        }
    }
    printf("-1");
    return 0;
}