题目链接:https://www.nowcoder.com/pat/6/problem/4051
题目描述
给定一个常数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(<= 105)、以及正整数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<iostream>
#include<vector>
using namespace std;
struct Lian{
int address;
int data;
int next_address;
};
vector<Lian> list(100000);
vector<Lian> new_list;
vector<Lian> ans;
int main(){
Lian temp;
int start,n,k;
scanf("%d%d%d",&start,&n,&k);
for(int i=0;i<n;i++){
scanf("%d%d%d",&temp.address,&temp.data,&temp.next_address);
//联系向量的下标和地址
list[temp.address]=temp;
}
int t=start;
//首尾衔接
while(t!=-1){
new_list.push_back(list[t]);
t=list[t].next_address;
}
//题目中给出的节点可能是连接不上没用的节点故结点数不一定为n
//有效结点个数
int num=new_list.size();
int right=k-1;
while(right<num){
for(int i=right;i>right-k;i--){
ans.push_back(new_list[i]);
}
right+=k;
}
for(int i=right-k+1;i<num;i++){
ans.push_back(new_list[i]);
}
//修改除最后一个节点之外的所有节点的next指针,并输出每个节点信息;
for(int i=0;i<num-1;i++){
ans[i].next_address=ans[i+1].address;
printf("%05d %d %05d\n",ans[i].address,ans[i].data,ans[i].next_address);
}
//输出最后一个节点信息,注意next_address为-1
printf("%05d %d -1\n",ans[num-1].address,ans[num-1].data);
return 0;
}