PAT甲题题解-1052. Linked List Sorting (25)-排序

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <string.h>
using namespace std;
const int maxn=100000+5;
struct Node{
    int addr;
    int val;
    int to;
    bool operator<(const Node tmp)const{
        return val<tmp.val;
    }
}linked[maxn],node[maxn];
int main()
{
    int n,first;
    int a,b,c;
    scanf("%d %d",&n,&first);
    for(int i=0;i<n;i++){
        scanf("%d %d %d",&a,&b,&c);
        linked[a].addr=a;
        linked[a].val=b;
        linked[a].to=c;
    }
    int cnt=0;
    while(first!=-1){
        node[cnt].addr=linked[first].addr;
        node[cnt].val=linked[first].val;
        node[cnt].to=linked[first].to;
        cnt++;
        first=linked[first].to;
    }
    sort(node,node+cnt);
    if(cnt==0){
        printf("0 -1\n");  //最后一个样例有首地址为-1的情况。。。
        return 0;
    }
    printf("%d %05d\n",cnt,node[0].addr);
    for(int i=0;i<cnt;i++){
        printf("%05d %d ",node[i].addr,node[i].val);
        if(i==cnt-1)
            printf("-1\n");
        else
            printf("%05d\n",node[i+1].addr);
    }
    return 0;
}