1025 反转链表 (25 分)

 1025 反转链表 (25 分)

 

输入样例：

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

 

分析：输入样例正确连接顺序应该是：
/*
00100 1 12309
12309 2 33218
33218 3 00000
00000 4 99999
99999 5 68237
68237 6 -1
*/
还应该考虑输入样例中有不在链表中的结点的情况。所以用个sum计数

而且，algorithm头文件里面有reverse函数可以直接调用

 

#include <iostream>
#include <algorithm>
using namespace std;
int main(){
    int first,n,k,temp;
    cin>> first >> n >> k;
    int data[100005], next[100005], list[100005];
    for(int i=0;i<n;i++){
        cin>>temp;// 输入样例：第一行的 00000
        cin>>data[temp] >> next[temp];// 第一行的4存入data[00000],  99999存入next[00000] 
    }
    int sum=0;//不一定所有的输入的结点都是有用的，加个计数器
    while(first!= -1){
        list[sum++]=first;//00100
        first=next[first];//例如 12309=next[00100]
    }
    
    for(int i=0;i<(sum-sum % k); i+=k){//每k行
        reverse(begin(list)+i, begin(list)+i+k);
    }
    for(int i=0;i<sum-1;i++)
        printf("%05d %d %05d\n", list[i], data[list[i]],list[i+1]);
    printf("%05d %d -1",list[sum-1], data[list[sum-1]]);
    return 0;
}