PAT 1025. 反转链表*

原题链接：http://pat.zju.edu.cn/contests/pat-b-practise/1025

给定一个常数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⁵)、以及正整数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 <stdio.h>
struct Node
{
  int data;
  int next;
};
const int MAX_NUM = 100000;
int main()
{
  Node *nodes = new Node[MAX_NUM];
  int headAddr, n, k;
  int addr, next;
  while(scanf("%d %d %d", &headAddr, &n, &k)!=EOF)
  {
    for(int i=0; i<n; ++i)
    {
      scanf("%d", &addr);
      scanf("%d %d", &nodes[addr].data, &nodes[addr].next);
    }
    int left_nodes = 0;
    next = headAddr;
    while(next != -1)//可能有节点不在链表上
    {
      ++left_nodes;
      next = nodes[next].next;
    }
    k = k>left_nodes ? left_nodes : k;
    int startAddr = headAddr;
    int last = -1;//上一轮反转k个节点的最后一个节点的地址
    int pre,cur;
    while(left_nodes >= k)
    {
      pre = startAddr;
      cur = nodes[pre].next;
      if(cur == -1)//只剩一个元素时
      {
        if(last != -1)//==1表示仅有一个元素
        {
          nodes[last].next = pre;
        }
        last = pre;
        left_nodes--;
        break;
      }
      next = nodes[cur].next;
      for(int i=1; i<k; ++i)
      {
        nodes[cur].next = pre;
        pre = cur;
        cur = next;
        next = nodes[cur].next;
      }
      if(last!=-1)
      {
        nodes[last].next = pre;
      }
      else {
        headAddr = pre;
      }
      left_nodes -= k;
      last = startAddr;
      startAddr = cur;
    }
    if(left_nodes > 0)
    {
      nodes[last].next = cur;
    }
    else {
      nodes[last].next = -1;
    }
    while(nodes[headAddr].next != -1)
    {
      printf("%05d %d %05d\n", headAddr, nodes[headAddr].data, nodes[headAddr].next);
      headAddr = nodes[headAddr].next;
    }
    printf("%05d %d %d\n", headAddr, nodes[headAddr].data, nodes[headAddr].next);
  }
  delete []nodes;
  nodes = NULL;
  return 0;
}