QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#662144 | #6438. Crystalfly | kukka | WA | 0ms | 3812kb | C++20 | 2.5kb | 2024-10-20 21:25:05 | 2024-10-20 21:25:06 |
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> d1(n+1,0), d2(n+1,0), mx(n+1,0);
function<void(int,int)> dfs1 = [&](int cur, int fa) -> void {
if (edges[cur].size() == 1 && edges[cur][0] == fa) {
d1[cur] = a[cur];
// mx[cur] = cur;
return;
}
int temp{};
for (auto&& to : edges[cur]) {
if (to == fa) continue;
if (a[to] > temp) {
temp = a[to];
mx[cur] = to;
}
dfs1(to, cur);
d1[cur] += d1[to] - a[to];
}
d1[cur] += a[mx[cur]] + a[cur];
};
dfs1(1, 0);
function<void(int,int)> dfs2 = [&](int cur, int fa) -> void {
d2[cur] = d1[cur];
int fi{}, se{}, temp1{}, temp2{};
for (auto&& to : edges[cur]) {
if (to == fa || t[to] != 3) continue;
if (a[to] > temp1) {
temp1 = a[to];
fi = to;
}
else if (a[to] > temp2) {
temp2 = a[to];
se = to;
}
}
int temp{};
for (auto&& to : edges[cur]) {
if (to == fa) continue;
dfs2(to, cur);
temp += d2[to] - a[to];
if (to == fi) {
d2[cur] = max(d2[cur], d2[cur] - a[mx[cur]] + a[to] - a[mx[to]] + a[se]);
}
else {
d2[cur] = max(d2[cur], d2[cur] - a[mx[cur]] + a[to] - a[mx[to]] + a[fi]);
}
}
for (auto&& to : edges[cur]) {
if (to == fa) continue;
d2[cur] = max(d2[cur], temp + a[to] + a[cur]);
}
};
dfs2(1, 0);
cout << d2[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: 0ms
memory: 3812kb
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: 3568kb
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 32 19 19
result:
wrong answer 8th numbers differ - expected: '28', found: '32'