东方博宜1501 - 数的操作

题目描述

给定一个 N 个数的数组， M 次操作，每次操作为下列操作之一。求最后的数组。
操作 1 ：在第 X 个数之后插入一个数 Y 。
操作 2 ：删除第 X 个数。

输入

第一行两个整数 N,M（N,M≤100000），N 表示数组中一开始有 N 个数，M 表示 M 次操作。
第二行 N 个整数，表示原来的数组。
接下来 M 行，每行第一个数 OPT，表示操作类型。
对于操作 1，接下来两个数 X ，Y ，表示在第 X 数之后插入一个数 Y，保证0≤X≤当前数的个数，若 X=0，表示在数组开头插入。
对于操作 2 ，接下来一个数 X ，表示要删除第 X 个数，保证 1≤X≤当前数的个数。

输出

输出若干个数，表示最后的数组。

样例

输入

5 3
1 2 3 4 5
1 1 6
2 1
2 2

输出

6 3 4 5

代码1

​
#include <iostream>
#include <vector>
using namespace std;

int main() {
    int N, M;
    cin >> N >> M;

    vector<int> nums(N);
    for (int i = 0; i < N; i++) {
        cin >> nums[i];
    }

    for (int i = 0; i < M; i++) {
        int opt;
        cin >> opt;

        if (opt == 1) {
            int X, Y;
            cin >> X >> Y;

            // 在第 X 个数之后插入一个数 Y
            nums.insert(nums.begin() + X, Y);
        } else if (opt == 2) {
            int X;
            cin >> X;

            // 删除第 X 个数
            nums.erase(nums.begin() + X - 1);
        }
    }

    // 输出最后的数组
    for (int i = 0; i < nums.size(); i++) {
        cout << nums[i] << " ";
    }

    return 0;
}

​

在这段代码中，我们使用一个向量 nums 来存储原始的数组。首先，我们按照输入的要求读取原始数组。然后，我们根据操作的类型，分别进行插入和删除操作。对于插入操作，我们使用 insert 函数在第 X 个数之后插入一个数 Y。对于删除操作，我们使用 erase 函数删除第 X 个数。最后，我们按照题目要求输出最后的数组。

代码2

#include <iostream>
#include <list>
using namespace std;

int main() {
    int N, M;
    cin >> N >> M;

    list<int> nums;
    for (int i = 0; i < N; i++) {
        int num;
        cin >> num;
        nums.push_back(num);
    }

    for (int i = 0; i < M; i++) {
        int opt;
        cin >> opt;

        if (opt == 1) {
            int X, Y;
            cin >> X >> Y;

            // 在第 X 个数之后插入一个数 Y
            auto it = nums.begin();
            advance(it, X);
            nums.insert(it, Y);
        } else if (opt == 2) {
            int X;
            cin >> X;

            // 删除第 X 个数
            auto it = nums.begin();
            advance(it, X - 1);
            nums.erase(it);
        }
    }

    // 输出最后的数组
    for (auto it = nums.begin(); it != nums.end(); it++) {
        cout << *it << " ";
    }

    return 0;
}

在这段代码中，我们使用了一个双向链表 nums 来存储原始的数组。我们将数字依次添加到链表的尾部。对于插入操作，我们使用 advance 函数移动到第 X 个数的位置，并在其后插入一个数 Y。对于删除操作，我们同样使用 advance 函数移动到第 X 个数的位置，并删除该节点。最后，我们使用迭代器遍历链表，并输出最后的数组。使用链表可以更加高效地进行插入和删除操作。