Code:
// luogu-judger-enable-o2 #include<bits/stdc++.h> #define ll long long #define lson (now<<1) #define rson ((now<<1)|1) #define setIO(s) freopen(s".in","r",stdin) #define maxn 300000 const long long inf = 100000000000000; using namespace std; char uuu[9]; ll V[maxn]; int fa[maxn],hd[maxn],to[maxn],nex[maxn],siz[maxn],hson[maxn]; int F[maxn][2]; int top[maxn],bot[maxn],dfn[maxn],ln[maxn]; int edges,tim,n,Q; void add(int u,int v) { nex[++edges]=hd[u],hd[u]=edges,to[edges]=v; } struct Matrix { ll a[2][2]; ll*operator[](int x){return a[x]; } }t[maxn<<1],tmp[maxn]; Matrix operator*(Matrix a,Matrix b) { Matrix c; c[0][0]=min(a[0][0]+b[0][0],a[0][1]+b[1][0]); c[0][1]=min(a[0][0]+b[0][1],a[0][1]+b[1][1]); c[1][0]=min(a[1][0]+b[0][0],a[1][1]+b[1][0]); c[1][1]=min(a[1][0]+b[0][1],a[1][1]+b[1][1]); return c; } void dfs1(int u,int ff) { fa[u]=ff,siz[u]=1; for(int i=hd[u];i;i=nex[i]) { int v=to[i]; if(v==ff) continue; dfs1(v,u); siz[u]+=siz[v]; if(siz[v]>siz[hson[u]]) hson[u]=v; } } void dfs2(int u,int tp) { top[u]=tp,ln[++tim]=u,dfn[u]=tim; if(hson[u]) dfs2(hson[u],tp),bot[u]=bot[hson[u]]; else bot[u]=u; for(int i=hd[u];i;i=nex[i]) { int v=to[i]; if(v==fa[u]||v==hson[u]) continue; dfs2(v,v); } } void dfs(int u) { F[u][0]=0,F[u][1]=V[u]; for(int i=hd[u];i;i=nex[i]) { int v=to[i]; if(v==fa[u]) continue; dfs(v); F[u][0]+=F[v][1]; F[u][1]+=min(F[v][0],F[v][1]); } } void build(int l,int r,int now) { if(l>r) return; if(l==r) { int u=ln[l]; ll f0=0,f1=V[u]; for(int i=hd[u];i;i=nex[i]) { int v=to[i]; if(v==fa[u]||v==hson[u]) continue; f0+=F[v][1]; f1+=min(F[v][0],F[v][1]); } t[now]=tmp[l]=(Matrix){inf, f0,f1,f1}; return; } int mid=(l+r)>>1; build(l,mid,lson),build(mid+1,r,rson); t[now]=t[lson]*t[rson]; } void modify(int l,int r,int now,int p) { if(l==r) { t[now]=tmp[l]; return; } int mid=(l+r)>>1; if(p<=mid) modify(l,mid,lson,p); else modify(mid+1,r,rson,p); t[now]=t[lson]*t[rson]; } Matrix query(int l,int r,int now,int L,int R) { if(l==L&&r==R) return t[now]; int mid=(l+r)>>1; if(R<=mid) return query(l,mid,lson,L,R); if(L>mid) return query(mid+1,r,rson,L,R); return query(l,mid,lson,L,mid)*query(mid+1,r,rson,mid+1,R); } void Update(int u,ll w) { tmp[dfn[u]][1][0]=tmp[dfn[u]][1][1]+=w-V[u], V[u]=w; while(u) { Matrix a=query(1,n,1,dfn[top[u]],dfn[bot[u]]); modify(1,n,1,dfn[u]); Matrix b=query(1,n,1,dfn[top[u]],dfn[bot[u]]); u=fa[top[u]]; if(!u) return; int x=dfn[u]; ll f0=a[0][1],f1=a[1][1],g0=b[0][1],g1=b[1][1]; tmp[x][1][0]=tmp[x][1][1]=tmp[x][1][0]+min(g1,g0)-min(f1,f0); tmp[x][0][1]=tmp[x][0][1]+g1-f1; } } void solve(int a,int x,int b,int y) { if(!x&&!y&&(fa[a]==b||fa[b]==a)) { printf("-1\n"); return ;} ll o=V[a],oo=V[b]; Update(a,x?o-inf:o+inf), Update(b,y?oo-inf:oo+inf); Matrix ans=query(1,n,1,dfn[top[1]],dfn[bot[1]]); ll h=min(ans[0][1],ans[1][1]); if(x) h+=inf; if(y) h+=inf; printf("%lld\n",h); Update(a,o), Update(b,oo); } int main() { // setIO("input"); scanf("%d%d%s",&n,&Q,uuu); for(int i=1;i<=n;++i) scanf("%lld",&V[i]); for(int i=1,u,v;i<n;++i) scanf("%d%d",&u,&v),add(u,v),add(v,u); dfs1(1,0),dfs2(1,1),dfs(1),build(1,n,1); while(Q--) { int a,x,b,y; scanf("%d%d%d%d",&a,&x,&b,&y); solve(a,x,b,y); } return 0; }