C语言实现PAT 1025反转链表

1025 反转链表 （25 分)

给定一个常数 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 <iostream>
#include<cstdio>
using namespace std;
struct NODE{
  int data;
  int next;
  int pre;
  int address;

}node[100000];
int findhead(int address,int n)
{
    int i=1,p=address;
    for(i=1;i<n;i++)
    {
     p=node[p].next;
     if(p==-1)
        break;
    }
    return p;
}
int main()
{
    int s,n,k;
    scanf("%d%d%d",&s,&n,&k);
    int address,next,data;
    for(int i=0;i<n;i++)
    {
        scanf("%d%d%d",&address,&data,&next);
        node[address].data=data;
        node[address].next=next;
        node[next].pre=address;
        node[address].address=address;
    }

    int p,m=1;//m计数 到K就翻转
    int tem;
    int head=s;
    for(int i=1;i<k;i++)//找到头节点 以便最后输出
    {
        head=node[head].next;
    }

    for(p=s;p!=-1;p=node[p].next)
    {
        if(m!=k)
        {
            m++;
        }
        else
        {
           m=2; //m不能等于1，因为for循环 p下一次就要跳到下一个结点
           tem=node[p].next;

           while(p!=s)//进行翻转，s为终止翻转的节点
           {
               node[p].next=node[p].pre;
               p=node[p].pre;
           }
           if(tem!=-1)
           {
           int c=findhead(tem,k);//找到下一段翻转过后的头节点，和前一段连接起来
           if (c!=-1)
           node[p].next=c;
           else //最后一段节点不足，直接连接
           {
            node[p].next=tem;
            break;
           }
           s=tem;//s记录下一段的 翻转前的头节点，翻转时到这个节点结束
           p=tem;
           }
           else
           {
               node[p].next=-1;
           }

        }
    }
    for(int p=head;p!=-1;p=node[p].next)
        if(node[p].next!=-1)
        printf("%05d %d %05d\n",node[p].address,node[p].data,node[p].next);
        else
        printf("%05d %d -1",node[p].address,node[p].data);

    return 0;
}