PAT 乙级 1025 反转链表 (25分)

我的个人网站 Cheese的个人主页http://www.cheese.ren/

博客来源 PAT 乙级 1025 反转链表-Cheese的个人博客http://blog.cheese.ren/45

欢迎交换友链 :-)

---

#include <bits/stdc++.h>

using namespace std;

int main() {
    int head, n, k;
    scanf("%d %d %d", &head, &n, &k);
    int data[100005], next[100005];
    for (int i=0; i<n; i++) {
        int temp;
        scanf("%d", &temp);
        scanf("%d %d", &data[temp], &next[temp]);
    }
    int list[n], result[n], sum=0;
    while (head != -1) {
        list[sum++] = head;
        head = next[head];
    }
    for (int i=0; i<sum; i++)
        result[i] = list[i];
    for (int i=0; i<sum-sum%k; i++)
        result[i] = list[i/k*k+k-1-i%k];
    for (int i=0; i<sum-1; i++)
        printf("%05d %d %05d\n", result[i], data[result[i]], result[i+1]);
    printf("%05d %d -1\n", result[sum-1], data[result[sum-1]]);
    return 0;
}