PAT乙级-1075 链表元素分类 (25 分)

题目链接

#include<iostream>
#include<vector>
using namespace std;

struct node{
    int data,next;
}list[100000];

int main(){
    int start,n,k;
    cin>>start>>n>>k;
    int address;
    for(int i=0;i<n;i++){
        cin>>address;
        cin>>list[address].data>>list[address].next;
    }
    vector<int> v[3]; //分成3类
    int q=start;
    while(q!=-1){
        if(list[q].data<0) v[0].push_back(q);
        else if(list[q].data>=0&&list[q].data<=k) v[1].push_back(q);
        else v[2].push_back(q);
        q=list[q].next;
    }
    bool flag=true;
    for(int i=0;i<3;i++){
        for(int j=0;j<v[i].size();j++){
            if(flag){
                printf("%05d %d ",v[i][j],list[v[i][j]].data);
                flag=false;
            }else{
                printf("%05d\n%05d %d ",v[i][j],v[i][j],list[v[i][j]].data);
            }
        }
    }
    cout<<-1<<endl;
    return 0;
}