反转链表

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⁵)、以及正整数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 loc;
    int data;
};
struct temp{
    int data;
    int next;
};
int main()
{
    struct node array[100001];//按顺序存数据，用作栈
    struct temp book[100001];//读入数据,数组下标代表位置
    int n;
    int head;
    int k;
    scanf("%d",&head);
    scanf("%d",&n);
    scanf("%d",&k);
    for(int i=1;i<=n;i++){
        int loc;
        int data;
        int next;
        scanf("%d%d%d",&loc,&data,&next);
        book[loc].data=data;
        book[loc].next=next;
    }
    int top=0;//栈头
    int t=head;
    while(t!=-1){
       array[++top].loc=t;
       array[top].data=book[t].data;
       t=book[t].next;
    }
    int i=1;
    while(i*k<=top){
        int now=k*(i-1)+1;
        int end=k*i;
        for(int j=end;j>now;j--)
            printf("%05d %d %05d\n",array[j].loc,array[j].data,array[j-1].loc);
        int next;//当前段最后一个元素指向的位置
        if((i+1)*k<=top)
            next=array[(i+1)*k].loc;
        else{
            if(i*k==top)
                next=-1;
            else
                next=array[i*k+1].loc;

        }
        if(next!=-1)
            printf("%05d %d %05d\n",array[now].loc,array[now].data,next);
        else
            printf("%05d %d %d\n",array[now].loc,array[now].data,-1);

        i++;
    }
    int leftbegin=(top/k)*k+1;//剩下的不够k的元素们的开始位置
    if(leftbegin!=top+1){
        for(int i=leftbegin;i<top;i++)
            printf("%d %d %d\n",array[i].loc,array[i].data,array[i+1].loc);
        printf("%d %d %d\n",array[top].loc,array[top].data,-1);
    }
    return 0;
}