PAT B1025 反转链表

给定一个常数 K 以及一个单链表 L，请编写程序将 L 中每 K 个结点反转。例如：给定 L 为 1→2→3→4→5→6，K 为 3，则输出应该为 3→2→1→6→5→4；如果 K 为 4，则输出应该为 4→3→2→1→5→6，即最后不到 K 个元素不反转。

输入格式：
每个输入包含 1 个测试用例。每个测试用例第 1 行给出第 1 个结点的地址、结点总个数正整数 N (≤$10^5$)、以及正整数 K (≤N)，即要求反转的子链结点的个数。结点的地址是 5 位非负整数，NULL 地址用 −1 表示。

接下来有 N 行，每行格式为：

Address Data Next
其中 Address 是结点地址，Data 是该结点保存的整数数据，Next 是下一结点的地址。

输出格式：
对每个测试用例，顺序输出反转后的链表，其上每个结点占一行，格式与输入相同。

输入样例：

00100 6 4
00000 4 99999
00100 1 12309
68237 6 -1
33218 3 00000
99999 5 68237
12309 2 33218

输出样例：

00000 4 33218
33218 3 12309
12309 2 00100
00100 1 99999
99999 5 68237
68237 6 -1

代码：

#include <cstdio>
#include<algorithm>
#define maxn 100010
using namespace std;


struct Node
{
    int addr;
    int data;
    int next;
    int order=100011;
}node[maxn];
bool cmp(Node a, Node b)
{
    return a.order < b.order;
}
int main()
{
    int L, N, K,p,i,j;
    scanf("%d %d %d", &L, &N, &K);
    
    int addr, data, next;
    int n = N;
    while (N--)
    {
        scanf("%d %d %d", &addr, &data, &next);
        node[addr].addr = addr;
        node[addr].data = data;
        node[addr].next = next;
    }
    p = L;
    int cnt = 0;
    while (p != -1)
    {
        cnt++;
        node[p].order = cnt;
        p = node[p].next;

    }
    sort(node, node + 100010, cmp);
    int last = node[K - 1].next;

    for (i = 0; i < cnt-(cnt%K); i+=K)
    {
        reverse(node + i, node + i + K);
    }
    for (i = 0; i < cnt; i++)
    {
        printf("%05d %d", node[i].addr, node[i].data);
        if (cnt - 1 != i)
        {
            printf(" %05d\n", node[i + 1]);
        }
        else
        {
            printf(" -1");
        }
    }
    return 0;
}

先说一下这个题目的坑吧。最后两个测试点的链表上有无效的结点，这些结点是无效的，不能输出，所以我们需要用cnt标记处一个有效且有序的链表。在审题的时候，我忽略了“请编写程序将 L 中每 K 个结点反转”，错误理解为将前K个节点反转，导致了测试点1、5 、6都没有通过。

思路：用数组存储节点，另外给节点按照顺序编号order。排好序后，每K个节点反转（reverse）一次。最后处理好的数组按序输出，注意，此时每个节点的.next为下一个节点的.addr。

代码中用大小为100010的结构体数组存储节点，空间开销大。其他人用map容器写的代码相对更加高效。如下：

引用链接 Map容器存储节点

#include<bits/stdc++.h>
using namespace std;
struct node
{
	int data;
	int next;
};
int main()
{
	vector<int> l;	//地址序列 
	map<int,node> s; 
	int first,n,k,add;
	scanf("%d %d %d",&first,&n,&k);
	while(n--)
	{
		node t;
		cin>>add>>t.data>>t.next;
		s[add]=t;
	}
	while(first!=-1)	//遍历链表，将所有链上地址存到地址序列中 
	{
		l.push_back(first);
		first=s[first].next;
	}
	for(int i=0;i<(l.size()-l.size()%k);i+=k)	//将序列中的地址按k个一组进行倒置，末尾不足一组的不处理 
		reverse(l.begin()+i,l.begin()+i+k);
	for(int i=0;i<l.size();i++)
	{
		printf("%05d %d ",l[i],s[l[i]].data);
		if(i==l.size()-1)	//最后一个结点的next应为-1 
			printf("-1");
		else
			printf("%05d\n",l[i+1]);
	}
	return 0;
}