PAT甲1133 Splitting A Linked List（25 分）

#include <stdio.h>
#include <string.h>
#include <map>
#include <set>
#include <queue>
#include <string>
#include <algorithm>
#include <iostream>
using namespace std;

int st,high,N;
struct node
{
    int data;
    int now;
    int next;
}A[100010];

int main()
{
    scanf("%d%d%d",&st,&N,&high);
    int now,data,next;
    for(int i=0;i<N;i++)
    {
        scanf("%d%d%d",&now,&data,&next);
        A[now].now=now;
        A[now].data=data;
        A[now].next=next;
    }
    int head=100005,result=100006;
    A[head].next=st;
    A[result].next=-1;
    int pre=head,p=st,r;
    int rp=result;
    while(p!=-1)
    {
        if(A[p].data<0)
        {
            r=A[p].next;
            A[pre].next=r;
            A[p].next=-1;
            A[rp].next=p;
            rp=p;
            p=r;
        }
        else
        {
            pre=p;
            p=A[p].next;
        }
    }
    pre=head;
    p=A[head].next;
    while(p!=-1)
    {
        if(A[p].data>=0&&A[p].data<=high)
        {
            r=A[p].next;
            A[pre].next=r;
            A[p].next=-1;
            A[rp].next=p;
            rp=p;
            p=r;
        }
        else
        {
            pre=p;
            p=A[p].next;
        }
    }
    pre=head;
    p=A[head].next;
    while(p!=-1)
    {
        if(A[p].data>high)
        {
            r=A[p].next;
            A[pre].next=r;
            A[p].next=-1;
            A[rp].next=p;
            rp=p;
            p=r;
        }
        else
        {
            pre=p;
            p=A[p].next;
        }
    }
    p=A[result].next;
    while(p!=-1)
    {
        if(A[p].next==-1)
        {
            printf("%05d %d -1\n",A[p].now,A[p].data);
        }
        else
        {
            printf("%05d %d %05d\n",A[p].now,A[p].data,A[p].next);
        }
        p=A[p].next;
    }
    return 0;
}