给定一个单链表,请编写程序将链表元素进行分类排列,使得所有负值元素都排在非负值元素的前面,而 [0, K] 区间内的元素都排在大于 K 的元素前面。但每一类内部元素的顺序是不能改变的。例如:给定链表为 18→7→-4→0→5→-6→10→11→-2,K 为 10,则输出应该为 -4→-6→-2→7→0→5→10→18→11。
输入格式:
每个输入包含一个测试用例。每个测试用例第 1 行给出:第 1 个结点的地址;结点总个数,即正整数N (≤105);以及正整数K (≤103)。结点的地址是 5 位非负整数,NULL 地址用 −1 表示。
接下来有 N 行,每行格式为:
Address Data Next
其中 Address
是结点地址;Data
是该结点保存的数据,为 [−105,105] 区间内的整数;Next
是下一结点的地址。题目保证给出的链表不为空。
输出格式:
对每个测试用例,按链表从头到尾的顺序输出重排后的结果链表,其上每个结点占一行,格式与输入相同。
输入样例:
00100 9 10
23333 10 27777
00000 0 99999
00100 18 12309
68237 -6 23333
33218 -4 00000
48652 -2 -1
99999 5 68237
27777 11 48652
12309 7 33218
输出样例:
33218 -4 68237
68237 -6 48652
48652 -2 12309
12309 7 00000
00000 0 99999
99999 5 23333
23333 10 00100
00100 18 27777
27777 11 -1
思路参考了https://blog.csdn.net/liyuanyue2017/article/details/80156778,感谢!
链表类型的题可以考虑用数组来进行模拟,开始是这个想法,用一个数组n[100010]来存储某个地址的后一个地址,用数组d[100010]来存储这个地址的对应数值,用一个数组flag[100010]来判断这些数字的性质,最后通过三次for循环将这些地址放入数组m[100010],显而易见,经过四次for循环,已经严重超时了。如何优化,首先第一步,利用if条件判断先消除flag的影响,三次for循环仍是不可避免的。第二步,既然题目要求的是输出,那么便将结点加入数组 m[100010]这一步给省去,转而控制条件直接输出,而不是添加入数组再输出。第三步,注意到scanf与cin的区别,添加语句,提高输入输出的速度。
代码如下:
#include <bits/stdc++.h>
using namespace std;
int d[100010];
int n[100010];
int index1[100010];
int main()
{
std::ios_base::sync_with_stdio(false);
std::cin.tie(0);
int address0,N,K;
cin>>address0>>N>>K;
int i;
int address;
for(i=0;i<N;i++)
{
cin>>address;
cin>>d[address];
cin>>n[address];
}
int h=0;
index1[h]=address0;
h++;
int start=address0;
while(n[start]!=-1)
{
index1[h++]=n[start];
start=n[start];
}
//此处将for循环的规模缩小
//其二,本来需要将待添加的放入一个数组中进行输出,这里就直接控制格式输出,节省了时间
int flag=0;
for(i=0;i<h;i++)
{
if(d[index1[i]]<0)
{
if(flag==1)
{
cout<<setw(5)<<setfill('0')<<index1[i]<<endl;
}
cout<<setw(5)<<setfill('0')<<index1[i]<<" "<<d[index1[i]]<<" ";
flag=1;
}
}
for(i=0;i<h;i++)
{
if(d[index1[i]]>=0&&d[index1[i]]<=K)
{
if(flag==1)
{
cout<<setw(5)<<setfill('0')<<index1[i]<<endl;
}
cout<<setw(5)<<setfill('0')<<index1[i]<<" "<<d[index1[i]]<<" ";
flag=1;
}
}
for(i=0;i<h;i++)
{
if(d[index1[i]]>K)
{
if(flag==1)
{
cout<<setw(5)<<setfill('0')<<index1[i]<<endl;
}
cout<<setw(5)<<setfill('0')<<index1[i]<<" "<<d[index1[i]]<<" ";
flag=1;
}
}
cout<<-1<<endl;
return 0;
}