按顺序读进来然后reverse,输出的时候输出下一个条目的地址。
PAT链表题特别爱搞的一个case是可能会有那么几项是多的,不在待排序的链里。所以要重新计算链表长度,不能直接用给的项数。
#include <bits/stdc++.h>
using namespace std;
struct Node{
int add,val,next;
}node[100010];
vector<Node> vec;
int main(){
int root,num,k;
cin>>root>>num>>k;
for(int i=0;i<num;i++){
int add,val,nxt;
cin>>add>>val>>nxt;
node[add].add=add;
node[add].val=val;
node[add].next=nxt;
}
int temp=root,cnt=0;
while(temp!=-1){
vec.push_back(node[temp]);
temp=node[temp].next;
cnt++;
}
for(int i=0;i+k<=cnt;i+=k){
reverse(vec.begin()+i,vec.begin()+i+k);
}
for(int i=0;i<cnt;i++){
if(i!=cnt-1)printf("%05d %d %05d\n",vec[i].add,vec[i].val,vec[i+1].add);
else printf("%05d %d -1\n",vec[i].add,vec[i].val);
}
}