题目链接:https://www.patest.cn/contests/pat-b-practise/1025
1025. 反转链表 (25)
时间限制
300 ms
内存限制
65536 kB
代码长度限制
8000 B
判题程序
Standard
作者
CHEN, Yue
给定一个常数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 <stdio.h>
#include<vector>
#include<algorithm>
using namespace std;
struct Node{
int Data;
int Next;
int ad;
};
int main(){
int address,N,K;
scanf("%d%d%d",&address,&N,&K);
Node a[100005];
int Address;
for(int i=0;i<N;i++){
scanf("%d",&Address);
scanf("%d%d",&a[Address].Data,&a[Address].Next);
a[Address].ad=Address;
}
vector<Node> node;
Node p;
int cnt=0;
for(p=a[address];;p=a[p.Next])
{
node.push_back(p);
cnt++;
if(p.Next==-1) break;
}
int m=cnt/K;
vector<Node> ::iterator it=node.begin();
for(int i=0;i<m;i++){
reverse(it,it+K);
it=it+K;
}
for(int i=0;i<node.size()-1;i++){
printf("%05d %d %05d\n",node[i].ad,node[i].Data,node[i+1].ad);
}
printf("%05d %d %d\n",node[(node.size())-1].ad,node[(node.size())-1].Data,-1);
return 0;
}
附ac图: