2019冬季PAT甲级第二题

#define HAVE_STRUCT_TIMESPEC
#include<bits/stdc++.h>
using namespace std;
typedef struct{
    int add,data,nex;
}Node;
Node node[100007],ans[100007],ans2[100007];
map<int,int>mp;
int main(){
    //ios::sync_with_stdio(false);
    //cin.tie(NULL);
    //cout.tie(NULL);
    int s,n,k;
    scanf("%d%d%d",&s,&n,&k);
    for(int i=1;i<=n;++i){
        scanf("%d%d%d",&node[i].add,&node[i].data,&node[i].nex);
        mp[node[i].add]=i;
    }
    int cnt=0;
    while(s!=-1){
        int num=mp[s];
        ans[++cnt].add=node[num].add;
        ans[cnt].data=node[num].data;
        ans[cnt].nex=node[num].nex;
        s=ans[cnt].nex;
    }
    int cnt2=0;
    int x=cnt/k;
    if(cnt%k)
        ++x;
    for(int i=x;i;--i){
        for(int j=(i-1)*k+1;j<=min(cnt,i*k);++j){
            ans2[++cnt2].add=ans[j].add;
            ans2[cnt2].data=ans[j].data;
            ans2[cnt2].nex=ans[j].nex;
        }
        if(i==1)
            ans2[cnt2].nex=-1;
        else
            ans2[cnt2].nex=ans[(i-2)*k+1].add;
    }
    for(int i=1;i<=cnt2;++i){
        if(i<cnt2)
            printf("%05d %d %05d
",ans2[i].add,ans2[i].data,ans2[i].nex);
        else
            printf("%05d %d -1",ans2[i].add,ans2[i].data);
    }
    return 0;
}