给定一个常数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
注意,有压力测试,我用cout 或者 原来是字符串保存会超时
代码:
//
// main.cpp
//
#include <iostream>
#include <queue>
#include <iomanip>
#include <math.h>
#include <vector>
#include <string>
#include <algorithm>
#include <map>
using namespace std;
typedef struct Node{
int numb;
int addr;
int next;
int head;
}node;
int main(int argc, const char * argv[]) {
map<int, node> mymap;
int start;
cin>>start;
int sum = 0;
cin>>sum;
int count;
cin>>count;
for(int i = 0 ; i < sum;i++)
{
node b;
cin>>b.addr;
cin>>b.numb;
cin>>b.next;
mymap[b.addr] = b;
}
if(start == –1)
{
cout<<“-1”;
}else
{
vector<node> aftersort;
int next = start;
while (next!= –1) {
node temp = mymap[next];
aftersort.push_back(temp);
next = mymap[next].next;
// cout<<next;
}
vector<node> final;
int lastindex = count-1;
while(lastindex < aftersort.size()){
for(int i = lastindex; i > lastindex-count; –i){
final.push_back(aftersort[i]);
}
lastindex += count;
}
for(int i = lastindex-count+1; i < aftersort.size(); ++i){
final.push_back(aftersort[i]);
}
for(int i = 0; i < final.size()-1; ++i){
final[i].next = final[i+1].addr;
}
int i = 0;
for(i = 0;i < final.size() –1 ;i++)
{
printf(“%05d %d %05d\n”,final[i].addr,final[i].numb,final[i].next);
// cout<<final[i].addr<<” “<<final[i].numb<<” “<<final[i].next<<endl;
}
printf(“%05d %d -1\n”,final[i].addr,final[i].numb);
// cout<<final[i].addr<<” “<<final[i].numb<<” “<<-1<<endl;
}
}