因为这个图增广来增广去不会有负环 那么一条增广链必然是树上一条简单路径 那么考虑用树形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;
}