1025. 反转链表 (25)

题目

PS：测试数据包含不再链表上的节点，需要手动删除。

给定一个常数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

代码

#include<stdio.h>
#include<stdlib.h>

struct node{
    int address;
    int data;
    int next;
};
void Reverse(node a[],int l,int r){
    node t;
    for(int i=l;i<(l+r+1)/2;i++){
        t=a[i];
        a[i]=a[l+r-i];
        a[l+r-i]=t;
    }
    if(l!=0)
        a[l-1].next=a[l].address;
    for(int i=l;i<=r;i++){
            a[i].next=a[i+1].address;
        }
}
int main(){
    int add,n,k;
    node *a,temp;
    scanf("%d %d %d",&add,&n,&k);

    a=(node *)malloc((n+1)*sizeof(node));
    for(int i=0;i<n;i++)
        scanf("%d %d %d",&a[i].address,&a[i].data,&a[i].next);
    a[n].address=-1;  //用来标志表尾

    int num=0;
    int kk=0;  //表示要反转的真实下标
    while(add!=-1){  //将链表按顺序存储在数组中
        int j=num;
        while(a[j].address!=add)
            j++;
        temp=a[num];
        a[num]=a[j];
        a[j]=temp;
        add=a[num].next;
        num++;
    }
    n=num;  //这步决定你的最后一组数据是否能通过（去掉不再链表上的节点）
    for(int i=0;i<n/k;i++){
        Reverse(a,k*i,k*(i+1)-1);
    }
    //Reverse(a,0,kk);
    //Reverse(a,kk+1,n-1);
    int i=0;
    while(a[i].next!=-1){
        printf("%05d %d %05d\n",a[i].address,a[i].data,a[i].next);
        i++;
    }
    printf("%05d %d %d",a[i].address,a[i].data,a[i].next);
    return 0;
}