1025 反转链表

---

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 (≤105)、以及正整数 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

代码长度限制																	16 KB
时间限制																		400 ms
内存限制																		64 MB

/*
    1025 反转链表
    https://pintia.cn/problem-sets/994805260223102976/exam/problems/994805296180871168
*/

#include <stdio.h>
#include <string.h>

struct Node
{
    int address;
    int data;
    int nextAddress;
};

int main()
{   
    Node link[100010];//用数组下标代表元素的地址
    memset(link, 0, sizeof(link));
    int FirstAddress,num,K;
    scanf("%d %d %d",&FirstAddress,&num,&K);

    for(int i=0;i<num;i++)//录入数据
    {
        int address,data,nextAddress;
        scanf("%d %d %d",&address,&data,&nextAddress);
        link[address].address=address;
        link[address].data=data;
        link[address].nextAddress=nextAddress;
    }
    int link1[100010],link2[100101];
    memset(link1, 0, sizeof(link1));
    memset(link2, 0, sizeof(link2));

    int count=0;
    for(int i=FirstAddress;i!=-1;i=link[i].nextAddress)//将前K个节点地址存放到link1中，升序存放到link2
    {
        if(count<K)
            link1[count]=i;
        else
            link2[count-K]=i;
        count++;
    }

    for(int i=K-1;i>=0;i--)//反着输出前K个节点
    {
        if(i>0)
            printf("%05d %d %05d\n",link[link1[i]].address,link[link1[i]].data,link[link1[i-1]].address);
        else
            printf("%05d %d %05d\n",link[link1[i]].address,link[link1[i]].data,link[link2[0]].address);
    }

    int remain=num-K;//看看从第K个节点后还有多少个节点
    if(remain<K)//分开的链表前边多余后边
    {
         for(int i=0;i<remain;i++)//正着输出
         {
             if(i==remain-1)
                printf("%05d %d -1\n",link[link2[i]].address,link[link2[i]].data);
             else
             printf("%05d %d %05d\n",link[link2[i]].address,link[link2[i]].data,link[link2[i+1]].address);
         }
    }
    else //分开的链表前后相等
    {
         for(int i=remain-1;i>=0;i--)//反着输出
         {
             if(i==0)
                printf("%05d %d -1\n",link[link2[i]].address,link[link2[i]].data);
             else
             printf("%05d %d %05d\n",link[link2[i]].address,link[link2[i]].data,link[link2[i+1]].address);
         }
    }

    return 0;
}