题目地址:1075.链表元素分类
易错分析:
1. 可能存在多余结点,即部分结点即使输入了,也不在链表中
2. 可能存在大于K的数组不存在元素
解题过程:
创建一个结构体数组,并用空间换时间的方法,把地址作为下标,值作为结构体对象,其中存储Data和Next。通过一个三维向量存储属于不同类的元素,最后输出即可。
程序:
#include <vector>
#include <iostream>
#include <cstdio>
using namespace std;
struct Node
{
int Data;
int Next;
}LinkList[100001];
int main(int argc, char const *argv[])
{
int start, N, K, Add, Data, Next;
cin >> start >> N >> K;
vector <int> v[3];
for (int i = 0; i < N; i++) // 保存结点信息
{
cin >> Add >> Data >> Next;
LinkList[Add].Data = Data;
LinkList[Add].Next = Next;
}
while (start != -1) // 这里不要用for循环,因为会存在有多余结点的情况。
{
if (LinkList[start].Data < 0) // 如果小于0放入第一个向量
v[0].push_back(start);
else if (LinkList[start].Data > K) // 如果大于K放入第三个向量
v[2].push_back(start);
else // 如果在区间内放入第二个向量
v[1].push_back(start);
start = LinkList[start].Next; // start往后移
}
int flag = 0; // 方便输出
/* 这种输出方式可以避免没有大于K的情况而出现单独输出v[3]最后一个元素出现问题 */
for (int i = 0; i < 3; i++)
for (int j = 0; j < v[i].size(); j++)
{
if (flag == 0)
{
printf("%05d %d ", v[i][j], LinkList[v[i][j]].Data);
flag = 1;
}
else
printf("%05d\n%05d %d ", v[i][j], v[i][j], LinkList[v[i][j]].Data);
}
printf("-1");
return 0;
}