QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#662224 | #6438. Crystalfly | kukka | WA | 1ms | 3676kb | C++20 | 2.1kb | 2024-10-20 22:03:41 | 2024-10-20 22:03:44 |
Judging History
answer
#define GCC optimize(3)
#include <bits/stdc++.h>
using namespace std;
using i64 = int64_t;
using i128 = __int128;
#define IOS ios::sync_with_stdio(false), cin.tie(0), cout.tie(0)
#ifdef DEBUG
#define cout cout << "\033[1;92m"
#define endl "\033[0m" << '\n';
#else
#define endl '\n'
#endif
void solve() {
int n;
cin >> n;
vector<int> a(n+1,0), t(n+1,0);
for (int i = 1; i <= n; ++i) cin >> a[i];
for (int i = 1; i <= n; ++i) cin >> t[i];
vector<vector<int>> edges(n+1);
for (int i = 1; i <= n-1; ++i) {
int from, to;
cin >> from >> to;
edges[from].push_back(to);
edges[to].push_back(from);
}
vector<int> dis(n+1,0), mx(n+1,0);
function<void(int,int)> dfs2 = [&](int cur, int fa) -> void {
if (edges[cur].size() == 1 && edges[cur][0] == fa) {
dis[cur] = a[cur];
return;
}
int fi{}, se{}, temp1{};
for (auto&& to : edges[cur]) {
if (to == fa || t[to] != 3) continue;
if (a[to] >= temp1) {
temp1 = a[to];
se = fi;
fi = to;
}
}
int temp{};
for (auto&& to : edges[cur]) {
if (to == fa) continue;
if (a[to] > temp) {
temp = a[to];
mx[cur] = to;
}
dfs2(to, cur);
dis[cur] += dis[to] - a[to];
}
dis[cur] += a[mx[cur]] + a[cur];
for (auto&& to : edges[cur]) {
if (to == fa) continue;
if (to == fi) {
dis[cur] = max(dis[cur], dis[cur] - a[mx[cur]] + a[to] - a[mx[to]] + a[se]);
}
else {
dis[cur] = max(dis[cur], dis[cur] - a[mx[cur]] + a[to] - a[mx[to]] + a[fi]);
}
}
};
dfs2(1, 0);
cout << dis[1] << endl;
}
signed main() {
IOS;
int t{1};
cin >> t;
while (t--) {
solve();
}
#ifdef _WIN64
system("pause");
#endif
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 1ms
memory: 3676kb
input:
2 5 1 10 100 1000 10000 1 2 1 1 1 1 2 1 3 2 4 2 5 5 1 10 100 1000 10000 1 3 1 1 1 1 2 1 3 2 4 2 5
output:
10101 10111
result:
ok 2 number(s): "10101 10111"
Test #2:
score: -100
Wrong Answer
time: 0ms
memory: 3608kb
input:
10 6 8 1 1 5 8 9 2 1 2 2 2 2 1 2 2 3 2 4 1 5 2 6 6 6 4 4 1 3 6 2 1 3 3 3 3 1 2 1 3 3 4 4 5 5 6 6 10 5 1 8 5 1 1 3 1 2 2 2 1 2 2 3 2 4 2 5 3 6 10 6 8 8 9 6 9 5 6 6 4 2 1 3 3 2 2 2 2 3 1 1 2 1 3 3 4 4 5 5 6 4 7 2 8 7 9 9 10 7 10 9 1 5 7 5 4 1 1 1 2 1 3 2 1 2 1 3 3 4 3 5 5 6 1 7 5 7 1 1 4 2 3 1 3 2 2 1...
output:
25 24 24 56 31 14 16 34 19 19
result:
wrong answer 8th numbers differ - expected: '28', found: '34'