#include<bits/stdc++.h>
using namespace std;
vector<int> cost, mic;int N;
struct node {
vector<long long> dp;
long long lock=-1,free = -1;
vector<node*> l;
int a;
};
// void dfs(node* n, node* p){
// for(node* c : n->l) if(c!=p) dfs(c,n);
// n->dp.resize(N+1);
// n->free = LONG_LONG_MAX;
// for(int k = n->a+1;k <= N;k++){
// n->dp[k] = cost[k];
// for(node* c : n->l){
// if(c==p) continue;
// if(!c->a){
// if(k == n->a+1) n->dp[k] += min(c->dp[k]-cost[k], c->free);
// else n->dp[k] += c->free;
// } else if(c->a == n->a+1){
// if(k>c->a) n->dp[k] += min(c->dp[k]-cost[k], c->free);
// else n->dp[k] += c->free;
// } else n->dp[k] += c->free;
// }
// n->free = min(n->free, n->dp[k]);
// }
// }
long long dfs(node* n, node* p, int l, int r){
if(l==r && r<N && n->lock != -1) return n->lock;
if(l==1&&r==N&&n->free != -1) return n->free;
long long ans = LONG_LONG_MAX;
for(node* c : n->l) if(c!=p) {
if(!c->a&&l<=n->a+1&&n->a+1<=r){
ans = dfs(c, n, n->a+1,n->a+1);
} else if(c->a == n->a+1&&r > c->a){
ans = dfs(c, n, max(l,c->a+1),r);
if(r == N) goto thi;
}
if(l==r) ans = min(ans,cost[l]+dfs(c,n, 1, N));
else ans = min(ans,mic[l]+dfs(c,n,1,N));
}
if(ans==LONG_LONG_MAX) {
if(l==r) ans = cost[l];
else ans = mic[l];
}
thi:
if(l==r&&r<N) n->lock=ans;
if(l==1&&r==N) n->free=ans;
return ans;
}
void solve(){
int n;cin >> n;
N=n;
node g[n+1];
cost.resize(n+1);
mic.resize(n+1);
for(int i = 1;i <= n;i++) cin >> g[i].a;
for(int i = 1;i <= n;i++) cin >> cost[i];
for(int i = n;i >= 1;i--){
mic[i] = cost[i];
if(i < n) mic[i] = min(mic[i],mic[i+1]);
}
for(int i = 0;i < n-1;i++){
int u,v;cin >> u >> v;
g[u].l.push_back(&g[v]);
g[v].l.push_back(&g[u]);
}
cout << dfs(&g[1], NULL, 1, N) << '\n';
}
int main(){
ios_base::sync_with_stdio(false);
cin.tie(0);
solve();
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
1372 KB |
Output is correct |
2 |
Correct |
2 ms |
1372 KB |
Output is correct |
3 |
Correct |
22 ms |
1468 KB |
Output is correct |
4 |
Correct |
3 ms |
1372 KB |
Output is correct |
5 |
Correct |
2 ms |
1372 KB |
Output is correct |
6 |
Correct |
2 ms |
1372 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
121 ms |
61268 KB |
Output is correct |
2 |
Correct |
154 ms |
61264 KB |
Output is correct |
3 |
Correct |
707 ms |
61456 KB |
Output is correct |
4 |
Execution timed out |
1537 ms |
61252 KB |
Time limit exceeded |
5 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
344 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
1372 KB |
Output is correct |
2 |
Correct |
2 ms |
1372 KB |
Output is correct |
3 |
Correct |
22 ms |
1468 KB |
Output is correct |
4 |
Correct |
3 ms |
1372 KB |
Output is correct |
5 |
Correct |
2 ms |
1372 KB |
Output is correct |
6 |
Correct |
2 ms |
1372 KB |
Output is correct |
7 |
Incorrect |
0 ms |
344 KB |
Output isn't correct |
8 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
1372 KB |
Output is correct |
2 |
Correct |
2 ms |
1372 KB |
Output is correct |
3 |
Correct |
22 ms |
1468 KB |
Output is correct |
4 |
Correct |
3 ms |
1372 KB |
Output is correct |
5 |
Correct |
2 ms |
1372 KB |
Output is correct |
6 |
Correct |
2 ms |
1372 KB |
Output is correct |
7 |
Correct |
121 ms |
61268 KB |
Output is correct |
8 |
Correct |
154 ms |
61264 KB |
Output is correct |
9 |
Correct |
707 ms |
61456 KB |
Output is correct |
10 |
Execution timed out |
1537 ms |
61252 KB |
Time limit exceeded |
11 |
Halted |
0 ms |
0 KB |
- |