[线段树] [二分图] Codeforces VK Cup 2015 - Round 2 A. Berland Miners

首先把木棒按长度排序。
找找规律可以发现每个节点都会覆盖一个区间，而这个区间是一个前缀，
再找找规律发现，序列中第i个木棒被覆盖至少i次就会存在解（可以用霍尔定理证明），那么就可以线段树乱搞了。

找出最右边的没有满足条件的木棒，增加的长度就至少是这个木棒的长度减去这个节点到根路径的最小值。用线段树乱搞一下就好了

考场上多写了个二分，多了个log……就被卡成40分

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <set>
#include <vector>
#include <ctime>

using namespace std;

const int N=500010;

typedef pair<int,int> par;
typedef set<par>::iterator IT;

int n,m,cnt,u,v,cur,lst;
int G[N],h[N],b[N],mn[N],smn[N],c[N];
struct edge{
  int t,nx;
}E[N<<1];
struct seg{
  int mn,add;
}T[N<<2];

set<par> S;
vector<int> V[N];

inline char nc(){
  static char buf[100000],*p1=buf,*p2=buf;
  return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}

inline void rea(int &x){
  char c=nc(); x=0;
  for(;c>'9'||c<'0';c=nc());for(;c>='0'&&c<='9';x=x*10+c-'0',c=nc());
}

inline void addedge(int u,int v){
  E[++cnt].t=v; E[cnt].nx=G[u]; G[u]=cnt;
  E[++cnt].t=u; E[cnt].nx=G[v]; G[v]=cnt;
}

void dfs(int x,int p){
  S.insert(par(h[x],x));
  IT mins=S.begin();
  mn[x]=mins->first;
  V[mins->second].push_back(x);
  mins++;
  if(mins!=S.end()) smn[x]=mins->first;
  else smn[x]=1<<30;
  for(int i=G[x];i;i=E[i].nx)
    if(E[i].t!=p) dfs(E[i].t,x);
  S.erase(S.find(par(h[x],x)));
}

inline void Push(int g){
  if(T[g].add==0) return ;
  T[g<<1].add+=T[g].add;
  T[g<<1].mn+=T[g].add;
  T[g<<1|1].add+=T[g].add;
  T[g<<1|1].mn+=T[g].add;
  T[g].add=0;
}

inline void Up(int g){
  T[g].mn=min(T[g<<1].mn,T[g<<1|1].mn);
}

void Modify(int g,int l,int r,int x,int L,int R){
  if(l==L&&r==R){
    T[g].mn+=x; T[g].add+=x;
    return ;
  }
  int mid=L+R>>1; Push(g);
  if(r<=mid) Modify(g<<1,l,r,x,L,mid);
  else if(l>mid) Modify(g<<1|1,l,r,x,mid+1,R);
  else Modify(g<<1,l,mid,x,L,mid),Modify(g<<1|1,mid+1,r,x,mid+1,R);
  Up(g);
}

int Query(int g,int x,int L,int R){
  if(L==R) return T[g].mn;
  int mid=L+R>>1; Push(g);
  if(x<=mid) return Query(g<<1,x,L,mid);
  else return Query(g<<1|1,x,mid+1,R);
}

void Build(int g,int l,int r){
  if(l==r){
    T[g].mn=-(m-l+1);
    return ;
  }
  int mid=l+r>>1;
  Build(g<<1,l,mid); Build(g<<1|1,mid+1,r);
  T[g].mn=min(T[g<<1].mn,T[g<<1|1].mn);
}

int main(){
  rea(n);
  for(int i=1;i<=n;i++) rea(h[i]);
  for(int i=1;i<n;i++)
    rea(u),rea(v),addedge(u,v);
  rea(m);
  for(int i=1;i<=m;i++) rea(b[i]);
  sort(b+1,b+1+m); Build(1,1,m);
  dfs(1,0);
  for(int i=1;i<=n;i++){
    c[i]=upper_bound(b+1,b+1+m,mn[i])-b-1;
    if(c[i]) Modify(1,1,c[i],1,1,m);
  }
  if(T[1].mn>=0) return puts("0"),0;
  for(int i=m;i;i--)
    if(Query(1,i,1,m)<0){
      lst=i; break;
    }
  int ans=1<<30;
  for(int i=1;i<=n;i++){
    if(V[i].size()<-T[1].mn) continue;
    if(h[i]>=b[lst]) continue;
    for(int j=0;j<V[i].size();j++){
      int u=V[i][j],nw=min(b[lst],smn[u]);
      int coverd=upper_bound(b+1,b+1+m,nw)-b-1;
      if(c[u]<coverd) Modify(1,c[u]+1,coverd,1,1,m);
    }
    if(T[1].mn>=0) ans=min(ans,b[lst]-h[i]);
    for(int j=0;j<V[i].size();j++){
      int u=V[i][j],nw=min(b[lst],smn[u]);
      int coverd=upper_bound(b+1,b+1+m,nw)-b-1;
      if(c[u]<coverd) Modify(1,c[u]+1,coverd,-1,1,m);
    }
  }
  if(ans==1<<30) return puts("-1"),0;
  printf("%d\n",ans);
  return 0;
}