PAT A1074 Reversing Linked List (25 分) 链表

最新推荐文章于 2021-12-04 20:19:09 发布

原创最新推荐文章于 2021-12-04 20:19:09 发布 · 266 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

PAT 专栏收录该内容

165 篇文章

订阅专栏

本文介绍了一种链表操作算法，将链表按每K个节点分组并反转，最后一组不足K个节点时不反转。通过存储节点到vector中，使用reverse函数实现分组反转，最后输出反转后的链表。

题目大意：给出N个节点，以及头节点的地址。要求将构成的链表按照每K个节点分组，每组内的节点顺序反转。如果最后一组不足K个节点，那么不反转。

常规的静态链表问题，首先存储所有节点，然后遍历一遍，将构成的链表存储在 vector<Node> list 中，之后对list进行分组反转，直接用reverse函数就好，reverse的范围是左闭右开。

要注意的是题目要求最后一组不足K个不反转，所以要反转的组数只有list.size() / K。

AC代码：

#include <vector>
#include <algorithm>
#include <cstdio>

using namespace std;

struct Node
{
    int addr;
    int data;
    int next;
    Node(){};
    Node(int addr, int data, int next):addr(addr), data(data), next(next){};
}allNodes[100010];

int main()
{
    int head, N, K;
    scanf("%d%d%d", &head, &N, &K);
    for (int i = 0; i < N; ++i)
    {
        int addr, data, next;
        scanf("%d%d%d", &addr, &data, &next);
        allNodes[addr] = Node(addr, data, next);
    }
    vector<Node> list;
    for (int p = head; p != -1; p = allNodes[p].next)
        list.push_back(allNodes[p]);
    for (int i = 0; i < list.size() / K; ++i)
        reverse(&list[i * K], &list[i * K + K]);
    for (int i = 0; i < list.size(); ++i)
    {
        if(i < list.size() - 1) printf("%05d %d %05d\n", list[i].addr, list[i].data, list[i+1].addr);
        else printf("%05d %d -1\n", list[i].addr, list[i].data);
    }
    return 0;
}