#include <bits/stdc++.h>
#define int long long
#define mi(x, y) (x = min(x, y))
#define ma(x, y) (x = max(x, y))
using namespace std;
const int NS = (int)3e5 + 4;
int n;
int a[NS], f[NS], mn[NS];
int dp[NS];
vector<int> len[NS];
vector<int> way[NS];
int md;
void sol3(int rt, int x, int pr){
for(auto&nxt:way[x]){
if(nxt == pr){
continue;
}
if(a[x] + 1 == a[nxt]){
sol3(rt, nxt, x);
}
else{
len[rt].push_back(a[x]);
}
}
}
int sol2(int x, int pr){
int rv = 0;
for(auto&nxt:way[x]){
if(nxt == pr){
continue;
}
if((a[x] + 1) % md == a[nxt]){
rv += sol2(nxt, x);
}
else{
rv += dp[nxt];
}
}
if(dp[x]){
rv = min(rv, dp[x]);
}
return rv;
}
void sol(int x, int pr){
for(auto&nxt:way[x]){
if(nxt == pr){
continue;
}
sol(nxt, x);
}
md = (int)1e18;
dp[x] = sol2(x, pr) + mn[a[x]];
sol3(x, x, pr);
for(auto&i:len[x]){
md = i + 1;
dp[x] = min(dp[x], sol2(x, pr) + f[i]);
}
}
signed main(){
ios_base::sync_with_stdio(false);
cin.tie(0);
cin >> n;
for(int i = 0; i < n; ++i){
cin >> a[i];
}
for(int i = 0; i < n; ++i){
cin >> f[i];
}
for(int i = n - 1; i >= 0; --i){
mn[i] = f[i];
if(i + 1 < n){
mn[i] = min(mn[i], mn[i + 1]);
}
}
for(int i = 1; i < n; ++i){
int x, y;
cin >> x >> y;
--x, --y;
way[x].push_back(y);
way[y].push_back(x);
}
sol(0, -1);
cout << dp[0] << '\n';
return 0;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
99 ms |
15320 KB |
Output is correct |
2 |
Correct |
94 ms |
15248 KB |
Output is correct |
3 |
Correct |
131 ms |
15220 KB |
Output is correct |
4 |
Correct |
38 ms |
15104 KB |
Output is correct |
5 |
Correct |
11 ms |
15120 KB |
Output is correct |
6 |
Correct |
9 ms |
15060 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Execution timed out |
1578 ms |
46268 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
7 ms |
14420 KB |
Output is correct |
2 |
Correct |
8 ms |
14368 KB |
Output is correct |
3 |
Correct |
7 ms |
14420 KB |
Output is correct |
4 |
Incorrect |
7 ms |
14376 KB |
Output isn't correct |
5 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
99 ms |
15320 KB |
Output is correct |
2 |
Correct |
94 ms |
15248 KB |
Output is correct |
3 |
Correct |
131 ms |
15220 KB |
Output is correct |
4 |
Correct |
38 ms |
15104 KB |
Output is correct |
5 |
Correct |
11 ms |
15120 KB |
Output is correct |
6 |
Correct |
9 ms |
15060 KB |
Output is correct |
7 |
Correct |
7 ms |
14420 KB |
Output is correct |
8 |
Correct |
8 ms |
14368 KB |
Output is correct |
9 |
Correct |
7 ms |
14420 KB |
Output is correct |
10 |
Incorrect |
7 ms |
14376 KB |
Output isn't correct |
11 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
99 ms |
15320 KB |
Output is correct |
2 |
Correct |
94 ms |
15248 KB |
Output is correct |
3 |
Correct |
131 ms |
15220 KB |
Output is correct |
4 |
Correct |
38 ms |
15104 KB |
Output is correct |
5 |
Correct |
11 ms |
15120 KB |
Output is correct |
6 |
Correct |
9 ms |
15060 KB |
Output is correct |
7 |
Execution timed out |
1578 ms |
46268 KB |
Time limit exceeded |
8 |
Halted |
0 ms |
0 KB |
- |