PTA 1075 链表元素分类

题目如下

给定一个单链表，请编写程序将链表元素进行分类排列，使得所有负值元素都排在非负值元素的前面，而 [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⁵)；以及正整数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

解题思路

这题没有技术含量，也没有难度。唯一难受的就是最后一个测试点通过不了（其实在自己写代码的时候就已经感觉会运行超时过不了了）。最终果然，没有办法只能换一种方法了。直接上数组吧，不用过多解释，直接利用数组然后遍历将地址当作数组的索引存放在对应的位置，搜索的时候直接搜索next为该数组中对应的index即可。

下面是第一次超时的代码

源代码

#include<iostream>
#include<vector>
#include<string>
#include<algorithm>
using namespace std;

struct node{
	string address;
	int num;
	string next;
};

void fun(node n)
{
	cout<<n.address<<" "<<n.num<<" "<<n.next<<endl;
}

int main()
{
	string begin;
	int N,K;
	vector<node> vec;
	vector<node> order;
	vector<node> underzero;
	vector<node> underK;
	vector<node> upperK;
	cin>>begin>>N>>K;
	for(int i=0;i<N;i++)
	{
		node n;
		cin>>n.address>>n.num>>n.next;
		if(n.address==begin)
		{
			order.push_back(n);
			continue;
		}
		vec.push_back(n);
	}
	int i=0;
	while(order[i].next!="-1")
	{
		for(int j=0;j<vec.size();j++)
		{
			if(order[i].next==vec[j].address)
			{
				order.push_back(vec[j]);
				vector<node>::iterator iter=vec.begin();
				iter+=j;
				vec.erase(iter);
				break;
			}
		}
		i++;
	}
	
	for(int i=0;i<order.size();i++)
	{
		if(order[i].num<0)
			underzero.push_back(order[i]);
		else if(order[i].num>=0&&order[i].num<=K)
			underK.push_back(order[i]);
		else
			upperK.push_back(order[i]);
	}
	underzero.insert(underzero.end(),underK.begin(),underK.end());
	underzero.insert(underzero.end(),upperK.begin(),upperK.end());
	for(int i=0;i<underzero.size()-1;i++)
	{
		cout<<underzero[i].address<<" "<<underzero[i].num<<" "<<underzero[i+1].address<<endl;
	}
	cout<<underzero[underzero.size()-1].address<<" "<<underzero[underzero.size()-1].num<<" -1";
}

果真是最后一个点过不了。。。没办法，改代码咯

改进代码

#include<iostream>
using namespace std;
int datas[100005],nexts[100005],list[100005],order[100005];
int main()
{
	int firstAddress,N,K,Address;
	
	cin>>firstAddress>>N>>K;
	for(int i=0;i<N;i++)
	{
		cin>>Address;
		cin>>datas[Address]>>nexts[Address];
	}
	int sum=0,x=0;
	 while (firstAddress != -1)
	 {
        list[sum++] = firstAddress;	
        firstAddress = nexts[firstAddress];
    }
    //标记 
	for(int i=0;i<sum;i++)
	{
		if(datas[list[i]]<0){
			order[x++]=list[i];
		}
	}
	for(int i=0;i<sum;i++)
	{
		if(datas[list[i]]>=0&&datas[list[i]]<=K){
			order[x++]=list[i];
		}
	}
	for(int i=0;i<sum;i++){
		if(datas[list[i]]>K){
			order[x++]=list[i];
		}
	}
	//输出 
	for(int i=0;i<sum-1;i++)
	{
		 printf("%05d %d %05d\n", order[i], datas[order[i]], order[i + 1]);
	}
	printf("%05d %d -1", order[sum - 1], datas[order[sum - 1]]);
	
	return 0;
}

运行结果