1025 反转链表（PAT 乙级 中文 C/C++语言）

1025 反转链表 （25 point(s)）

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

分析：

1，定义静态单链表，数组大小大于 N。方便后面把链表链接起来。

2，因为地址为五位数，存放的时候可以是 addr[address]。

3，遍历链表，找出单链表所有有效结点。注意是有效结点，其它非法输入要排除掉。

4，用 sort 对链表进行排序。注意排序后 N 是有效结点，令它等于 count 。不然有一个 casse 无法通过。

5，按照要求输出。分两种情况：一种刚好 N 被 K 整除；另一种是最后一块剩下小于 K 个，需要做特殊处理一下。

 

详细代码：

#include <iostream>    
#include <vector>
#include <algorithm>
using namespace std;
     
const int size = 100005;
typedef struct{
    int num;
    int data;
    int address,next; 
}addr;

bool cmp(addr a,addr b){
    return (a.num<b.num);
}

// 1025 反转链表 （25 point(s)）
int main(void){       
    int n,k,addr0;
    cin>>addr0>>n>>k;

    addr ad[size];
    for(int i=0;i<size;++i){
        ad[i].num = size;
    }
    for(int i=0;i<n;++i){
        int address; cin>>address;
        cin>>ad[address].data>>ad[address].next;  
        ad[address].address = address;
    }

    int temp = addr0,count=0;
    while(temp != -1){
        ad[temp].num = (count++);
        temp = ad[temp].next;
    }

    sort(ad,ad+size,cmp); // 排序为单链表
    n = count;            // 不然最后一个 case 无法通过

    for(int i=0;i<n/k;++i){
        for(int j=(i+1)*k-1;j>i*k;--j){
            printf("%05d %d %05d\n",ad[j].address,ad[j].data,ad[j-1].address);
        }
        printf("%05d %d ",ad[i*k].address,ad[i*k].data);
        if(i<n/k-1){
            printf("%05d\n",ad[(i+2)*k-1].address);
        }else{  // 最后一块特殊处理
            if(n%k == 0){  // 最后一块刚好等于 k
                printf("-1\n");
            }else{
                printf("%05d\n",ad[(i+1)*k].address);
                for(int x=n/k*k;x<n;++x){  
                    if(x<n-1){
                        printf("%05d %d %05d\n",ad[x].address,ad[x].data,ad[x+1].address);  
                    }else{
                        printf("%05d %d -1\n",ad[x].address,ad[x].data); 
                    }
                }
            }
        }
    }
     
    return 0;
} // guangjinzheng 2019.1.19 13:23