思路:
1.依照题目要求,创建两个数组,一个用来存放原始的结点,另一个用来存放有效的结点
2.从头结点开始直到遇见尾结点遍历链表,并将遍历到的链表结点依照次序存放在第二个数组中
3.在第二个数组中每K个结点进行逆转,若不满K个则跳出循环
需要注意的是,PAT甲级的与链表相关的题目,输入中都可能包含无效的结点,即从题目给出的头结点出发无法遍历到的结点,这些结点是不予输出的。
#include <cstdio>
#include <algorithm>
const int maxn = 1000010;
struct node{
int addr, data, next;
}temp[maxn], list[maxn];
int main()
{
int head, n, k;
scanf("%d %d %d", &head, &n, &k);
int addr, data, next; //for temporary store
for (int i = 0; i < n; ++i) {
scanf("%d %d %d", &addr, &data, &next);
temp[addr].addr = addr;
temp[addr].data = data;
temp[addr].next = next;
}
int i = 0, p = head;
while (p != -1) {
list[i++] = temp[p];
p = temp[p].next;
} //i equals to the number of valid nodes in the list when the loop ends
n = i;
for (int j = 0; j < n; j += k) {
if (j + k <= n) std::reverse(list + j, list + j + k);
}
for (int j = 0; j < n; ++j) {
if (j < n - 1) printf("%05d %d %05d\n", list[j].addr, list[j].data, list[j + 1].addr);
else printf("%05d %d -1\n", list[j].addr, list[j].data);
}
return 0;
}
完~