如题:http://www.patest.cn/contests/pat-b-practise/1025
给定一个常数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<cstdio>
#include<cstring>
#include<map>
using namespace std;
struct node
{
int now,next;
int data;
node(){}
node(int s1,int dat,int s2):now(s1),next(s2),data(dat){}
}list[100005];
map<int,node>MAP;
int main()
{
// freopen("C:\\1.txt","r",stdin);
int start;
int n,k;
cin>>start>>n>>k;
int i;
for(i=0;i<n;i++)
{
int s1,s2;
int data;
cin>>s1>>data>>s2;
MAP[s1]=node(s1,data,s2);
}
int spos=start;
int m=1;
while(spos!=-1)
{
list[m++]=MAP[spos];
spos=MAP[spos].next;
}
m--;
int first=m-m%k-k+1;
int cnt=0;
while(first>0)
{
cnt++;
int end=first+k-1;
for(i=end;i>first;i--)
MAP[list[i].now].next=MAP[list[i-1].now].now;
int pre;
if(end==m)
pre=-1;
else
{
if(cnt==1)
pre=list[end+1].now;
else
pre=start;
}
MAP[list[first].now].next=pre;
first-=k;
start=list[end].now;
}
while(MAP[start].next!=-1)
{
//cout<<MAP[start].now<<" "<<MAP[start].data<<" "<<MAP[start].next<<endl
printf("%05d %d %05d\n",MAP[start].now,MAP[start].data,MAP[start].next);
start=MAP[start].next;
}
printf("%05d %d -1\n",MAP[start].now,MAP[start].data);
return 0;
}