社区讨论

10pts 求条

P11855[CSP-J 2022 山东] 部署参与者 1已保存回复 1

讨论操作

快速查看讨论及其快照的属性,并进行相关操作。

当前回复
1 条
当前快照
1 份
快照标识符
@mhj0eh7j
此快照首次捕获于
2025/11/03 18:40
4 个月前
此快照最后确认于
2025/11/03 18:40
4 个月前
查看原帖
CPP
#include <bits/stdc++.h>
using namespace std;

const int N = 1e6 + 5;
int n, m, q, a[N];
vector<vector<int>> tree_f;
vector<vector<int>> tree_s;

void dfs(int x, int y) {
	if (tree_f[x].size() == 0) {
		return;
	}
	for (int i = 0; i < tree_f[x].size(); i++) {
		a[tree_f[x][i]] += y;
		dfs(tree_f[x][i], y);
	}
}

int main() {
	cin >> n;
	tree_f.resize(n + 1);
	tree_s.resize(n + 1);
	for (int i = 1; i <= n; i++) cin >> a[i];
	for (int i = 1, x, y; i < n; i++) {
		cin >> x >> y;
		tree_f[x].push_back(y);
		tree_s[x].push_back(y);
		tree_s[y].push_back(x);
	}
	cin >> m;
	for (int i = 1, p, x, y; i <= m; i++) {
		cin >> p >> x >> y;
		if (p == 1) {
			dfs(x, y);
			a[x] += y;
		} else {
			for (int i = 0; i < tree_s[x].size(); i++) {
				a[tree_s[x][i]] += y;
			}
			a[x] += y;
		}
	}
	cin >> q;
	for (int i = 1, x; i <= q; i++) {
		cin >> x;
		cout << a[x] << '\n';
	}
	
	return 0;
}

回复

1 条回复,欢迎继续交流。

正在加载回复...