PAT 乙级 1025 反转链表 (25 分) and 甲级 1074 Reversing Linked List (25 分)

#include <iostream>
#include <algorithm>
using namespace std;
int main() {
    int first,n,k,temp;               
    cin>>first>>n>>k;
    int data[100005],next[100005],list[100005];
    for(int i=0;i<n;i++){
        cin>>temp;
        cin>>data[temp]>>next[temp];
    }
    int sum=0;						//不⼀定所有的输⼊的结点都是有⽤的，加个计数器
    while(first!=-1){        		 //链表按顺序排好
        list[sum++]=first;
        first=next[first];
    }
    for(int i=0;i<sum-sum%k;i+=k){     //隔k个数字逆转链表
        reverse(list+i,list+i+k);
    }
    for (int i = 0; i < sum - 1; i++)
        printf("%05d %d %05d\n", list[i], data[list[i]], list[i + 1]);
    printf("%05d %d -1", list[sum - 1], data[list[sum - 1]]);
    system("pause");
    return 0;
}