PAT (Basic Level) Practice 1025 反转链表

题目：编写程序将链表L中每K个结点反转。

示例：

链表L：1→2→3→4→5→6

K = 4

反转后：4→3→2→1→5→6

注意：最后不到 K 个元素不反转

输入格式：

 第1行给出：第1个结点的地址beginAddress、结点总个数N、正整数K

随后N行：每行按照Address Data Next的格式，给出各结点信息。

输出格式：

顺序输出反转后的链表，其上每个结点占一行，格式与输入相同 。

 思路：首先定义结构体保存链表结点信息，再读入输入的n个结点的信息，然后通过第1个结点的地址，将链表链接起来（输入的结点的信息是无序的且有些结点不在beginAddress引导的链表上），最后，从后向前将链表按每K个元素反转。

代码：

#include<iostream>
#include<string>
#include<vector>
#include<map>
using namespace std;
typedef struct Node{
    int address;
    int data;
    int next;
}node; // 链表结点
int main(){
    int begAdd,n,k;cin >> begAdd >> n >> k;
    map<int,node> mp;node a;// 地址:结点，暂存输入的结点
    while(cin >> a.address >> a.data >> a.next){
        mp.insert(make_pair(a.address,a));
    }
    n = 0; // 给出的结点，不一定都在一条链上
    /*
    测试点6中：太坑了这一点
    给定的n个节点，只有不到n个节点在起始位置为begAdd的链表上
    */
    vector<node> v; // 保存链接起来的链表
    while(begAdd != -1){
        node temp = mp[begAdd];
        v.push_back(temp);
        begAdd = temp.next;
        n ++;
    }
    int count = n / k;
    for(int i = (count - 1) * k ; i >= 0 ; i -= k){ // 反转链表
        // i , i + 1 , ... , i + k-1
        int left = i , right = i + k - 1;
        while(left < right){
            node temp = v[left];
            v[left] = v[right];
            v[right] = temp;
            left ++; right --;
        }
        for(int j = i ; j < i + k - 1 ; j ++){
            v[j].next = v[j+1].address;
        }
        v[i + k - 1].next = ((i + k < n) ? v[i + k].address : -1);
    }
    // cout << "反转后的链表" << endl;
    for(int i = 0 ; i < n ; i ++){
        if(v[i].next != -1)
            printf("%05d %d %05d\n",v[i].address,v[i].data,v[i].next);
        else
            printf("%05d %d %d\n",v[i].address,v[i].data,v[i].next);
    }
    return 0;
}