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 (≤)、以及正整数 K (≤)，即要求反转的子链结点的个数。结点的地址是 5 位非负整数，NULL 地址用 − 表示。

接下来有 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<cstring>
#include<string>
#include<cmath>
#include<cstdlib>
#include<algorithm>
#include<iostream>

using namespace std;

struct node{
    int address;
    int data;
    int next;
}a[100010],b[100010];

int main(){
    int i,j=0,head,n,k,address,data,next;  //head,n,k是第一行输入的n是总的录入个数，k是每k个结点翻转一下
    scanf("%d %d %d",&head,&n,&k); //输入第一行
    for(i=0;i<n;i++){    //录入数据
        scanf("%d %d %d",&address,&data,&next);
        a[address].address=address;
        a[address].data=data;
        a[address].next=next;
    }
    for(i=head;i!=-1;i=a[i].next){  //把录入的数据按地址的顺序放在数组b中
        b[j++]=a[i];
    }
    for(i=0;i<=j-k;i+=k){  //设置j-k是因为在最后一段，如果所剩的结点数小于k则不翻转
        reverse(b+i,b+i+k);  //设置b+i，b+i+k；是因为每间隔k个结点，这些k个结点内部翻转
    }                        //所以要引入<algorithm>提供的reverse函数
    for(i=0;i<j;i++){    //按照题目要求输出结果
        if(i!=j-1){
            printf("%05d %d %05d
",b[i].address,b[i].data,b[i+1].address);
        }else{          //这么分开的原因是除了最后一个结点，其他结点的地址和next地址都需要保持5位
            printf("%05d %d -1
",b[i].address,b[i].data);  //最后一个结点下一个地址是NULL，所以要输出-1，所以单独写出来
        }
    }
    return 0;
}