PTA 1075 链表元素分类 (25 分) C++实现

1075 链表元素分类 (25 分)

给定一个单链表，请编写程序将链表元素进行分类排列，使得所有负值元素都排在非负值元素的前面，而 [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
结尾无空行

解题思路：

先利用结构体数组建立起节点之间的联系，再建立一个一对多的关系来分别存储值小于 0 的地址，值在 0 到 k 之间的地址和值大于 k 的地址

未AC的代码：

#define _CRT_SECURE_NO_DEPRECATE
#include <iostream>
#include <algorithm>
#include <iomanip>
#include <vector>
#include <string>
#include <numeric>
#include <map>
#include <cmath>

using namespace std;

struct st
{
	string address;
	int data;
	string next;
};

int main()
{
	string first_ip;
	int n, k;
	cin >> first_ip >> n >> k;

	vector<st> res;

	while (n--)
	{
		st temp;
		cin >> temp.address >> temp.data >> temp.next;
		res.push_back(temp);
	}

	vector<st> real_res;
	for (int i = 0; i < res.size(); i++)
	{
		if (res[i].address == first_ip)
		{
			st temp;
			temp.address = res[i].address;
			temp.data = res[i].data;
			temp.next = res[i].next;
			real_res.push_back(temp);

			first_ip = res[i].next;
			i = -1;
		}
	}

	//输出负数
	int flag1 = 1;
	for (int i = 0; i < real_res.size(); i++)
	{
		if (flag1 > 1)
		{
			if (real_res[i].data < 0)
			{
				cout << real_res[i].address << endl;
				cout << real_res[i].address << " " << real_res[i].data << " ";
			}
		}
		else if (real_res[i].data < 0)
		{
			cout << real_res[i].address << " " << real_res[i].data
				<< " ";
			flag1++;
		}
	}

	//输出0—k的值
	int flag2 = 1;
	for (int i = 0; i < real_res.size(); i++)
	{
		if (flag2 > 1)
		{
			if (real_res[i].data >= 0 && real_res[i].data <= k)
			{
				cout << real_res[i].address << endl;
				cout << real_res[i].address << " " << real_res[i].data << " ";
			}
		}
		else if (real_res[i].data >= 0 && real_res[i].data <= k)
		{
			cout << real_res[i].address << "\n" << real_res[i].address
				<< " " << real_res[i].data << " ";
			flag2++;
		}
	}

	//输出大于k的值
	int flag3 = 1;
	for (int i = 0; i < real_res.size(); i++)
	{
		if (flag3 > 1)
		{
			if (real_res[i].data > k)
			{
				cout << real_res[i].address << endl;
				cout << real_res[i].address << " " << real_res[i].data << " ";
			}
		}
		else if (real_res[i].data > k)
		{
			cout << real_res[i].address << "\n" << real_res[i].address
				<< " " << real_res[i].data << " ";
			flag3++;
		}
	}
	cout << "-1" << endl;
}

运行结果：

调整思路后的代码：

#include <iostream>
#include <vector>

using namespace std;

struct st
{
	int data;
	int next;
}s[100005];

vector<int> ans[3]; //一对多关系，即一个下标对应一个数组

int main()
{
	int first_ip, n, k;
	cin >> first_ip >> n >> k;

	while (n--)
	{
		int temp_ip;
		cin >> temp_ip; 
		cin >> s[temp_ip].data >> s[temp_ip].next; //利用数组下标建立起地址与其值和其下一个地址的联系
	}

	int p = first_ip;
	while (p != -1)
	{
		int data = s[p].data;
		if (data < 0)
		{
			ans[0].push_back(p); //依次存负数的地址
		}
		else if (data >= 0 && data <= k)
		{
			ans[1].push_back(p); //依次存0~k的值的地址
		}
		else
		{
			ans[2].push_back(p); //依次存大于k的值的地址
		}
		p = s[p].next;
	}
	
	//依次输出即可
	int flag = 1; //输出控制
	for (int i = 0; i < 3; i++)
	{
		for (int j = 0; j < ans[i].size(); j++)
		{
			if (flag > 1)
			{
				printf("%05d\n%05d %d ", ans[i][j], ans[i][j], s[ans[i][j]].data);
			}
			else
			{
				printf("%05d %d ", ans[i][j], s[ans[i][j]].data);
				flag++;
			}
		}
	}
	cout << "-1" << endl;
}

运行结果：

思路借鉴：

https://blog.youkuaiyun.com/shiliang97/article/details/99777610