当前位置: 代码迷 >> 综合 >> PAT 乙级 1075 链表元素分类
  详细解决方案

PAT 乙级 1075 链表元素分类

热度:85   发布时间:2023-12-28 03:00:54.0

给定一个单链表,请编写程序将链表元素进行分类排列,使得所有负值元素都排在非负值元素的前面,而 [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 (≤10?5??);以及正整数K (≤10?3??)。结点的地址是 5 位非负整数,NULL 地址用 ?1 表示。

接下来有 N 行,每行格式为:

Address Data Next

其中 Address 是结点地址;Data 是该结点保存的数据,为 [?10?5??,10?5??] 区间内的整数;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;
}