PATA 1052 链表排序

题目大意
给定一个链表，请按照其关键字升序排序。

输入格式
第一行为一个小于$10^5$的正整数$N$以及头结点地址，其中$N$是 内存中结点的总个数，结点的地址是五位数字。用-1表示NULL。
接下来是$N$行，每行用以下形式表示一个结点：
地址 关键字($-10^5$ ~ $10^5$) 下一节点地址

输出格式
和输入格式相同，其中$N$表示链表中的结点总数，结点须按照关键字升序排序。

代码

#include <stdio.h>
#include <algorithm>

using namespace std;

const int max_n=100010;

struct node{
    int key,next,adr;
    bool flag=false;//是否为有效结点
}nodes[max_n];

//比较函数，将有效结点放在前面。若有效性相同，按照关键字从小到大排列
bool cmp(node a,node b){
    if(a.flag!=b.flag) return a.flag>b.flag;
    else return a.key<b.key;
}

int main(){
    int n, s_adr;
    scanf("%d %d",&n,&s_adr);
    int adr,key,next;
    //输入各个结点
    for(int i=0;i<n;i++){
        scanf("%d %d %d",&adr,&key,&next);
        nodes[adr].adr=adr;
        nodes[adr].key=key;
        nodes[adr].next=next;
    }
    int p=s_adr,cnt=0;
    //将有效结点进行标记并计算有效结点个数
    while(p!=-1){
        nodes[p].flag=true;
        p=nodes[p].next;
        cnt++;
    }
    //特判输出没有有效结点的情况
    if(cnt==0) printf("0 -1\n");
    else{
        //将结点排序，放置于静态链表起始段。地址保留为原有地址
        sort(nodes,nodes+max_n,cmp);
        //输出结果
        printf("%d %05d\n",cnt,nodes[0].adr);
        for(int i=0;i<cnt-1;i++){
            printf("%05d %d %05d\n",nodes[i].adr,nodes[i].key,nodes[i+1].adr);
        }
        printf("%05d %d -1\n",nodes[cnt-1].adr,nodes[cnt-1].key);
    }
    return 0;
}