CSP-S 2025 第一轮 第30题：程序阅读第 3 题 · 第 3 小题

题目

#include <algorithm>
#include <cstdio>
#include <cstring>
#include <vector>
#define ll long long
int n, m;
std::vector<int> k, p;
inline int mpow(int x, int k) {
    int ans = 1;
    for (; k; k >>= 1, x = x * x) {
        if (k & 1)
            ans = ans * x;
    }
    return ans;
}
std::vector<int> ans1, ans2;
int cnt1, cnt2;
inline void dfs(std::vector<int>& ans, int& cnt, int l, int r, int v) {
    if (l > r) {
        ++cnt;
        ans.push_back(v);
        return;
    }
    for (int i = 1; i <= m; ++i) {
        dfs(ans, cnt, l + 1, r, v + k[l] * mpow(i, p[l]));
    }
    return;
}
std::vector<int> cntans1;
int main() {
    scanf("%d%d", &n, &m);
    k.resize(n + 1);
    p.resize(n + 1);
    for (int i = 1; i <= n; ++i) {
        scanf("%d%d", &k[i], &p[i]);
    }
    dfs(ans1, cnt1, 1, n >> 1, 0);
    dfs(ans2, cnt2, (n >> 1) + 1, n, 0);
    std::sort(ans1.begin(), ans1.end());
    int newcnt1 = 1;
    cntans1.push_back(1);
    for (int i = 1; i < cnt1; ++i) {
        if (ans1[i] == ans1[newcnt1 - 1]) {
            ++cntans1[newcnt1 - 1];
        } else {
            ans1[newcnt1++] = ans1[i];
            cntans1.push_back(1);
        }
    }
    cnt1 = newcnt1;
    std::sort(ans2.begin(), ans2.end());
    int las = 0;
    ll ans = 0;
    for (int i = cnt2 - 1; i >= 0; --i) {
        for (; las < cnt1 && ans1[las] + ans2[i] < 0; ++las)
            ;
        if (las < cnt1 && ans1[las] + ans2[i] == 0)
            ans += cntans1[las];
    }
    printf("%lld\n", ans);
    return 0;
}

代码中第 39 行到第 50 行的目的是为了将 $ans1$ 数组进行“去重”操作。

选项

• T. 正确
• F. 错误

答案

T

题解

选 T（正确）。

这段代码先对 ans1 排序，让相同的数相邻，再进行去重：

• 如果当前元素与上一个保留的元素相同，只增加它的出现次数 cntans1[newcnt1 - 1]。
• 如果不同，就把它放到 ans1[newcnt1]，并让 newcnt1 加一。
• 最后用 cnt1 = newcnt1 更新有效元素个数。

例如，原来是 [1, 1, 2, 2, 2, 3]，处理后有效部分为 [1, 2, 3]，对应次数为 [2, 3, 1]。

注意：虽然没有缩小 vector 的实际长度，但后续只使用前 cnt1 个元素，所以已经实现了逻辑上的去重。