一个关于C++的问题

Question

有一个长度为 N 的整数序列 A=(A[1],A[2],…,A[N])，其中所有元素的初始值都设为 0。 此外，还有一个初始值为空的集合 S。
依次执行以下 Q 查询。处理完所有 Q 查询后，找出序列 A 中每个元素的值。第 i 个查询的格式如下：
给出一个整数 x[i]。然后，对于每个 j=1,2,…,N，如果 j∈S，将 |S| 添加到 A[j] 中。
这里，|S| 表示集合 S 中元素的个数。例如，如果 S={3,4,7}，那么 |S|=3。

输入格式
N Q
x[1] x[2] …x[Q]

输入样例1
3 4
1 3 3 2
输出样例1
6 2 2

输入样例2
4 6
1 2 3 2 4 2
输出样例2
15 9 12 7

用C++做一下，谢谢！

Answer 1

晚上好🌙🌙🌙
本答案参考ChatGPT-3.5

解决这个问题，可以按以下步骤进行：

1. 首先，根据输入的 N 创建一个大小为 N 的整数数组 A，并初始化所有元素为 0。
2. 创建一个空集合 S 用来保存查询中的整数。
3. 遍历输入的查询 x[i] 数组，依次执行以下操作：
   • 将 x[i] 添加到集合 S 中。
   • 对于数组 A 的每个元素 A[j]，如果 j 在集合 S 中，则将集合 S 的大小（即 |S|）添加到 A[j] 中。
4. 遍历数组 A，输出每个元素的值。

下面是相应的 C++ 代码：

#include <iostream>
#include <set>
#include <vector>

using namespace std;

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

    vector<int> A(N, 0);
    set<int> S;

    for (int i = 0; i < Q; i++) {
        int x;
        cin >> x;

        S.insert(x);

        for (int j = 0; j < N; j++) {
            if (S.count(j + 1) > 0) {
                A[j] += S.size();
            }
        }
    }

    for (int i = 0; i < N; i++) {
        cout << A[i] << " ";
    }
    cout << endl;

    return 0;
}

以上代码根据输入的查询逐步执行，计算每个元素的值，并输出结果。

希望对你有帮助！