PAT乙 1075.链表元素分类

题目地址：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;    
}