[树形DP 费用流手动增广] Codeforces Gym 101190 NEERC 16 M. Mole Tunnels

因为这个图增广来增广去不会有负环 那么一条增广链必然是树上一条简单路径 那么考虑用树形DP找最短路增广

dls是这么说的

考虑费用流，每一次要找一条最短的增广路。

枚举LCA，然后维护一下每个子树往下走的最短路即可。

增广的时候加一下反向边。

可以直接树形dp，不需要数据结构维护。

#include<cstdio>
#include<cstdlib>
#include<algorithm>
using namespace std;
typedef long long ll;

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

inline void read(int &x){
  char c=nc(),b=1;
  for (;!(c>='0' && c<='9');c=nc()) if (c=='-') b=-1;
  for (x=0;c>='0' && c<='9';x=x*10+c-'0',c=nc()); x*=b;
}

const int N=100005;

int n,m; ll Ans;
int c[N],p[N];
int pc[N],lc[N],rc[N];
int f[N],pos[N];

inline void Pushup(int x){
  f[x]=1<<30,pos[x]=-1;
  if (c[x]) f[x]=0,pos[x]=x;
  if ((x<<1)<=n && f[x<<1]+(lc[x]?-1:1)<f[x])
    f[x]=f[x<<1]+(lc[x]?-1:1),pos[x]=pos[x<<1];
  if ((x<<1|1)<=n && f[x<<1|1]+(rc[x]?-1:1)<f[x])
    f[x]=f[x<<1|1]+(rc[x]?-1:1),pos[x]=pos[x<<1|1];
}

int main(){
  int u,v;
  freopen("t.in","r",stdin);
  freopen("t.out","w",stdout);
  read(n); read(m);
  for (int i=1;i<=n;i++) read(c[i]);
  for (int i=n;i;i--) Pushup(i);
  while (m--){
    read(v);
    int cur=0;
    int lca=-1,minv=1<<30;
    for (int i=v;i;i>>=1){
      if (f[i]+cur<minv)
	minv=f[i]+cur,lca=i;
      pc[i]?cur--:cur++;
    }
    printf("%I64d ",Ans+=minv);
    for (int i=v;i!=lca;i>>=1)
      if (pc[i])
	pc[i]--;
      else
	i&1?rc[i>>1]++:lc[i>>1]++;
    int u=pos[lca]; c[u]--;
    for (int i=u;i!=lca;i>>=1)
      if (i&1)
	rc[i>>1]?rc[i>>1]--:pc[i]++;
      else
	lc[i>>1]?lc[i>>1]--:pc[i]++;
    for (int i=u;i;i>>=1) Pushup(i);
    for (int i=v;i;i>>=1) Pushup(i);
  }
  return 0;
}