CF533A Berland Miners
生活随笔
收集整理的這篇文章主要介紹了
CF533A Berland Miners
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
線段樹維護貪心
/**/
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#include<queue>
#include<cmath>
#include<set>
#define ll long long
#define M 500010
#define mmp make_pair
#define INF 1000000000
using namespace std;
int read() {int nm = 0, f = 1;char c = getchar();for(; !isdigit(c); c = getchar()) if(c == '-') f = -1;for(; isdigit(c); c = getchar()) nm = nm * 10 + c - '0';return nm * f;
}
vector<int> v[M] ;
int h[M], hm[M], st_tmp[M], scnt ;
struct P {int id, hm ;bool operator < (const P &rhs) const {return hm<rhs.hm ;}
} cave[M];void dfs0(int x, int f, int nowh) {hm[x]=nowh ;for(auto i : v[x]) if(i != f)dfs0(i, x, min(nowh, h[i])) ;
}
int tmp[M], tcnt ;
void dfs2(int x, int f, int nowhm) {tmp[tcnt++]=nowhm ;for(auto i : v[x]) if(i != f) {int newhm=min(nowhm, h[i]) ;if(newhm > hm[i]) dfs2(i, x, newhm) ;}
}multiset<int> st ;
bool check() {if(tcnt < st.size()) return 0 ;scnt=0 ;for(int i=0; !st.empty() && i<tcnt; i++) {auto it=st.upper_bound(tmp[i]) ;if(it != st.begin())it-- , st_tmp[scnt++]=*it , st.erase(it) ;}int sz=st.size() ;for(int i=0; i<scnt; i++) st.insert(st_tmp[i]) ;return sz==0 ;
}int LIM ;
bool dfs(int x, int f, int add) {if(h[x]<=LIM) {int newhm= (f==-1 ? h[x]+add : min(hm[f], h[x]+add)) ;if(newhm>hm[x]) {tcnt=0 ;dfs2(x, f, newhm) ;if(check()) return 1 ;}}for(auto i : v[x]) if(i != f && dfs(i, x, add))return 1 ;return 0 ;
}main() {int n = read();for(int i=1; i<=n; i++) h[i] = read();for(int i=1; i<n; i++) {int x = read(), y = read();v[x].push_back(y) ;v[y].push_back(x) ;}dfs0(1, -1, h[1]) ;int k = read();for(int i=1, x; i<=k; i++) x = read(), st.insert(x) ;for(int i=k+1; i<=n; i++) st.insert(0) ;for(int i=1; i<=n; i++) cave[i]=(P) {i, hm[i]} ;sort(cave+1, cave+n+1) ;LIM=-1 ;for(int i=1; i<=n; i++) {auto it=st.upper_bound(cave[i].hm) ;if(it != st.begin()) st.erase(--it) ;else if(LIM==-1) LIM=cave[i].hm ;}if(st.empty()) {printf("0\n") ;return 0;}if(!dfs(1, -1, INF)) {printf("-1\n") ;return 0 ;}int l=0 , r=INF ;while(r-l>1) {int mid=(r+l)/2 ;if(dfs(1, -1, mid)) r=mid ;else l=mid ;}printf("%d\n", r) ;
}
轉載于:https://www.cnblogs.com/luoyibujue/p/10527039.html
總結
以上是生活随笔為你收集整理的CF533A Berland Miners的全部內容,希望文章能夠幫你解決所遇到的問題。