题目来源:http://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算法的思想很清晰:
1:第一次使用结构体数组存储信息,随后根据地址信息串成双向链表链表形式,最后根据翻转长度值输出对应信息。但是很明显超时了,在串成结构体链表时的时间复杂度O(n^2).
2:接下里修改了查找的方法,使用Map来存,随后按地址顺序存储在另一结构体数组中,再根据翻转长度输出信息。
结果还是超时。
3:借鉴了网上的思想,使用三组一维int数组来存,由于地址是5位,所以地址信息不用单独存,直接将改地址对应的信息存在下标为该地址的数组位置,随后根据首址head和next数组将地址信息按序保存在list数组中,随后根据翻转长度输出信息。这次没有超时,但是有两个测试点没通过,改了很久都没有通过,希望大神们帮我看看,哪里的问题?
#include<iostream>
#include<stdio.h>
using namespace std;
int main()
{
int head,N,len,n;
int i,j;
int val[100001]={0};
int next[100001]={0};
int list[100001]={0};
int final[100001]={0};
cin>>head>>N>>len;
for(i=1;i<=N;i++)
{
int link;
cin>>link;
cin>>val[link]>>next[link];
}
n=1;
while(head!=-1)
{
list[n]=head;
head=next[head];
n++;
}
int low=0;
n--;
while(low+len<=n)
{
for(j=low+len;j>low+1;j--)
{
printf("%05d %d %05d\n",list[j],val[list[j]],list[j-1]);
}
printf("%05d %d ",list[j],val[list[j]]);
low+=len;
if(low==n)
{
printf("%d\n",-1);
break;
}
else if(low+len>n)
{
printf("%05d\n",list[low+1]);
for(j=low+1;j<n;j++)
printf("%05d %d %05d\n",list[j],val[list[j]],next[list[j]]);
printf("%05d %d %d\n",list[j],val[list[j]],-1);
break;
}
else
{
printf("%05d\n",list[low+len]);
}
}
return 0;
}