1052. Linked List Sorting (25)再

题目连接：https://www.patest.cn/contests/pat-a-practise/1052

这道题也是改了好长时间。注意测试样例中会有节点不再该链表上！另外输出也要分下情况，特殊的是链表为空、root为-1等情况的输出。

其实我的代码中还有不足，就是当N不为空，但是没有在该链表上的节点的时候，没有考虑到这一点，程序会崩溃。虽然能AC……应该是考察到根据链表头找出整个链表即可吧。

#include<stdio.h>
#include<vector>
#include<algorithm>
#include<iostream>
#define MAXN 100005
using namespace std;

typedef struct Node
{
    int Address;
    int Key;
    int Next;
}memory;

memory node[MAXN];
int cmp(memory a,memory b)
{
    return (a.Key<b.Key);
}

int main()
{
    int N,root,tmp;
    scanf("%d %d",&N,&root);
    vector<memory>head;
    int i,a,k,n;
    for (i=0;i<N;i++)
    {
        scanf("%d %d %d",&a,&k,&n);
        node[a].Address=a;
        node[a].Key=k;
        node[a].Next=n;
    }
    tmp=root;
    while (tmp>=0)
    {
        head.push_back(node[tmp]);
        tmp=node[tmp].Next;
    }

    if (head.empty())
    {
        if (root==-1)printf("0 -1
");
        else printf("0 %05d",root);
    }

    else
    {
        sort(head.begin(),head.end(),cmp);
        printf("%d %05d
",head.size(),head[0].Address);
        int last=head.size()-1;

        for (i=0;i<=last;i++)
        {
            if (i==last)printf("%05d %d -1",head[last].Address,head[last].Key);
            else
            {
                head[i].Next=head[i+1].Address;
                printf("%05d %d %05d
",head[i].Address,head[i].Key,head[i].Next);
            }
    }
    }
    return 0;
}