PAT1052

题目链接：http://pat.zju.edu.cn/contests/pat-a-practise/1052shui

水题一道，注意3点（题目没说清楚2,3）

（1）题目中可能存在多条链表。我们只要其中一条。

（2）如果给出的起始节点的地址为-1，则输出“0 -1”。

（3）如果根据给出的起始节点的地址无法找到一条链表，则输出“0 起始节点地址”。

 

#include<cstdio>
#include<vector>
#include<iomanip>
#include<algorithm>
using namespace std;

struct Node
{
    int address;
    int key;
    int next;
};

struct Record
{
    int key;
    int next;
}empty={0,-2};

bool comp(Node n1, Node n2)
{
    if(n1.key < n2.key)
        return true;
    else
        return false;
}

int main()
{
    int N,start;
    while(scanf("%d%d",&N, &start) != EOF)
    {
            vector<Record> v(100000, empty);
            for(int i=0; i<N; ++i)
            {
                int a,b,c;scanf("%d%d%d",&a, &b, &c);
                v[a].key=b; v[a].next=c;
            }
            if(start == -1)
                printf("0 -1
");
            else if(v[start].next == -2)
                printf("0 %05d
", start);
            else
            {
                vector<Node> inlist;
                while(start != -1)
                {
                    if(v[start].next == -2)
                        break;
                    Node n={start, v[start].key, v[start].next};
                    inlist.push_back(n);
                    start = v[start].next;
                }
                sort(inlist.begin(), inlist.end(), comp);
                int length = inlist.size();
                for(int i=0; i<length-1; ++i)
                    inlist[i].next = inlist[i+1].address;
                printf("%d %05d
", length, inlist[0].address);
                for(int i=0; i<length-1; ++i)
                    printf("%05d %d %05d
", inlist[i].address, inlist[i].key, inlist[i].next);
                printf("%05d %d -1
", inlist[length-1].address, inlist[length-1].key);
            }
    }
    return 0;
}