给定一个常数 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
分析
单链表的部分转置问题。由于结点的地址是 5 位非负整数,我们可以直接利用vector将数据起始地点直接储存到相应下标上去,然后再输入完后将其排序到sort中,然后再根据K进行转置(就是根据K确定遍历顺序再push进一个新的链表中),最后输出,注意输出格式及第一个地址就为-1的情况判断。
#include<iostream>
#include<vector>
#include<cstdio>
using namespace std;
struct node{
int pos; //结点地址
int data; //数据
int next; //下一节点地址
};
int main(){
vector<node> vin(100000); //输入数据,数据起始地点直接储存到相应下标上去;
vector<node> sort; //保存排序后的结果
vector<node> vout; //保存输出的结果
node temp;
int first,N,K;
cin>>first>>N>>K;
for(int i=0;i<N;i++){
cin>>temp.pos>>temp.data>>temp.next;
vin[temp.pos]=temp;
}
if(first==-1){ //起始地址是-1直接输出
cout<<"-1";
}
else{
int now=first;
while(now!=-1){ //循环将链表串起来
sort.push_back(vin[now]);
now=vin[now].next;
}
int lensort=sort.size(); //获取排序后的链表长度
int right=K-1; //依次从(下标)0~k-1,k~2*k-1.....转置直到right>(lensort-1)
while(right<lensort){
for(int i=right;i>right-K;i--){ //转置
vout.push_back(sort[i]);
}
right+=K;
}
for(int i=right-K+1;i<lensort;i++){ //把剩下不需要转置的入
vout.push_back(sort[i]);
}
for(int i=0;i<lensort;i++){
if(i!=lensort-1){
vout[i].next=vout[i+1].pos;
printf("%05d %d %05d\n",vout[i].pos,vout[i].data,vout[i].next);
}
else{
vout[i].next=-1;
printf("%05d %d %d\n",vout[i].pos,vout[i].data,vout[i].next);
}
}
}
return 0;
}