#12169. 珅泽教育CSP-J第一轮模拟考第二十二套 第 28 题
珅泽教育CSP-J第一轮模拟考第二十二套 第 28 题
二、阅读程序(程序输入不超过数组或字符串定义的范围;除特殊说明外,判断题 2 分,选择题 3 分,共计 40 分)
程序(3)
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
const int mod = 1000000000 + 7;
int w0[100005];
int w1[100005];
int w2[100005];
int n, m, k, f[100005], d[100005], id[100005];
vector<int> e[100005];
void dfs(int u, int fa) {
d[u] = d[fa] + 1;
f[u] = fa;
for (auto &v : e[u]) if (v != fa) {
dfs(v, u);
}
}
bool cmp(int a, int b) {
return d[a] < d[b];
}
int main() {
cin >> n >> m >> k;
for (int i = 2; i <= n; i++) {
int u, v;
cin >> u >> v;
e[u].push_back(v);
e[v].push_back(u);
}
dfs(1, 0); // ①
for (int i = 1; i <= m; i++) {
int x, w;
cin >> x >> w;
w1[x] = (w1[x] + w) % mod;
w2[x] = (w2[x] + w) % mod;
}
for (int i = 1; i <= n; i++)
id[i] = i;
sort(id + 1, id + 1 + n, cmp);
for (int i = 1; i <= k; i++) {
for (int j = n; j >= 1; j--) {
int x = id[j];
for (auto &y : e[x]) if (y != f[x]) {
w1[y] = (w1[y] + w1[x]) % mod;
}
w1[x] = 0;
}
for (int x = 1; x <= n; x++)
w1[x] = (w1[x] - w0[x] + mod) % mod,
w0[x] = 0;
for (int j = 1; j <= n; j++) { // ②
int x = id[j];
if (f[x]) {
w1[f[x]] = (w1[f[x]] + w2[x]) % mod;
w2[f[x]] = (w2[f[x]] + w2[x]) % mod;
w0[x] = (w0[x] + w2[x]) % mod;
w2[x] = 0;
}
}
}
for (int i = 1; i <= n; i++)
cout << w1[i] << " ";
return 0;
}
保证输入的 不超过 , 不超过 20,且 ,。
- 如果更改 ① 处
dfs(1, 0)为dfs(n, 0),则输出结果可能有变化。( )
{{ select(1) }}
- 正确
- 错误