uoj#397. 【NOI2018】情报中心

uoj#397. 【NOI2018】情报中心

---

题意

给出一棵$N \left(\le5\times10^4 \right)$个点的树，每条边有一个非负边权。再给出树上的$M \left(\le10^5 \right)$条链，链有费用。求选出两条边相交的链，它们的并的边权和减去费用之和的最大值。
多组数据，$\Sigma N \le10^6+233 ，\Sigma M \le2\times 10^6+233$。

---

背景

我好菜啊 我好菜啊 我套路都不会 我暴力都写挂
在赛场上获得了和自己水平很相配的5分。

---

题解

开始照着题解念
将一条链的边权和记为 $len$，费用记为 $cst$，点的带权深度记为 $dep$。
我们分类讨论一波。

当两条链的 $lca$ 不同（记为 $b$ 和 $c$）时，答案是两条链的价值去掉中间的交，也即 $len_1+len_2-cst_1-cst_2-dep_a+max\left(dep_b,dep_c\right)$。
我们把每条链拆成两条由 $lca$ 和一个端点组成的链。对于每个点 $i$，维护 $f_0\left(i,j\right),f_1\left(i,j\right)$ 表示下端点在 $i$ 的子树里，上端点不带权深度为 $j$ 的链的 $len-cst \left(+dep_{lca}\right)$ 的最大值。这个东西线段树合并上去就可以。
对于线段树上的一个节点，用其左子节点的 $f_0$ +右子节点的 $f_1$（再减去当前枚举的点的 $dep$ ）更新答案。在合并时，因为要求两条链分属当前点不同子节点的子树，需要用被合并的两个节点的左/右子节点交叉更新答案。//说不明白 代码里会涉及到

当两条链的 $lca$ 相同时，答案可以记为 $\frac1 2\left(len_1+len_2-2cst_1-2cst_2+dis(d,e)+dep_b+dep_c-dep_a\right)$。
我们把一条链的总价值即 $len-2cst+一个端点的dep$ 挂到另一个端点上，枚举分叉的位置 $a$，对于一个端点在它不同子节点的子树中的链的另一个端点求最远点对即为答案。因为边权非负/*之前问jcy 链的总价值不是非负的 会不会出问题 得到解答后我认为自己非常sb 然后jcy对我的观点表示赞同*/，两个集合的最远点对信息可合并。所以我们实现的时候先枚端点到 $lca$ 的倒数第二个节点 $f$ ，对于所有链的端点建虚树，然后自下至上合并。
总复杂度$O\left(n\log n\right)$。

---

代码

#include<bits/stdc++.h>
#define N 50004
#define V 1800000
#define mid (l+r>>1)
#define il inline
#define re return
#define mx(xx,yy) xx<yy?xx=yy:0
using namespace std;
typedef long long ll;
const ll inf=-1e17;
int T,n,m,D,f[N],dep[N],lg[N*2],to[N],hd[N],lk[N],
fa[18][N*2],siz[N],son[N],top[N],dfn[N],cnt,st[N],tot,
rt[N],c[V][2];
ll Dep[N],ans,tmx,tnow,add,mx0[V],mx1[V],tmp,ww;
struct sig{int u;ll w;}t1,t2;
struct dat{int u;sig v;};
vector<dat>s2[N];
il int lca(int u,int v){
    if(u==v)re u;
    if(dfn[u]>dfn[v])swap(u,v);
    register int d=lg[dfn[v]-dfn[u]];
    return min(fa[d][dfn[u]],fa[d][dfn[v]-(1<<d)+1]);
}
il ll dis(int u,int v){re u&&v?Dep[u]+Dep[v]-2*Dep[lca(u,v)]:inf;}
il ll dis(sig u,sig v){re dis(u.u,v.u)+u.w+v.w;}
il void upd(sig u,sig v){(tnow=dis(u,v))>tmx?t1=u,t2=v,tmx=tnow:0;}
struct mxd{
    sig a,b;int no;
    il void ini(){a.u=b.u=0;}
    il void uni(mxd &c){
        if(!a.u)re a=c.a,b=c.b,c.ini();
        t1=a,t2=c.a,tmx=dis(a,t2);
        upd(a,c.b),upd(b,c.a),upd(b,c.b);
        mx(ans,(tmx>>1)-Dep[no]);
        upd(a,b),upd(c.a,c.b);
        a=t1,b=t2,c.ini();
    }
}d[N],ad;
il int nxt(int u,int v){
    for(;top[u]^top[v];u=f[top[u]])
    if(f[top[u]]==v)re top[u];
    re son[v];
}
void dfs(int x){
    dep[x]=dep[f[x]]+1,Dep[x]+=Dep[f[x]];
    mx(D,dep[x]);
    fa[0][++cnt]=x,dfn[x]=cnt;
    for(register int s,i=lk[x];i;i=hd[i]){
        dfs(s=to[i]);
        fa[0][++cnt]=x,siz[x]+=siz[s];
        siz[s]>siz[son[x]]?son[x]=s:0;
    }
}
bool cmp(dat a,dat b){re dfn[a.u]<dfn[b.u];}
int u,v,w,i;
il void upd(int x,int y){mx(ans,mx0[c[x][0]]+mx1[c[y][1]]-add);}
il void pu(int k){
    mx0[k]=mx0[c[k][mx0[c[k][0]]<mx0[c[k][1]]]];
    mx1[k]=mx1[c[k][mx1[c[k][0]]<mx1[c[k][1]]]];
}
void ins(int &k,int l,int r,int d){
    if(!k)k=++cnt,c[k][0]=c[k][1]=0,mx0[k]=mx1[k]=inf;
    l<r?d>mid?ins(c[k][1],mid+1,r,d):ins(c[k][0],l,mid,d),pu(k),upd(k,k),0:
    (mx(mx0[k],tmp),mx(mx1[k],tmx));
}
void del(int &k,int l,int r,int lim){
    l>lim?k=0:0;
    if(l<=lim&&r>lim)del(c[k][0],l,mid,lim),del(c[k][1],mid+1,r,lim),pu(k);
}
void uni(int &x,int y,int l,int r){
    x&&y?l<r?upd(x,y),upd(y,x),
    uni(c[x][0],c[y][0],l,mid),uni(c[x][1],c[y][1],mid+1,r),pu(x),0:
    (mx(mx0[x],mx0[y]),mx(mx1[x],mx1[y])):(x|=y);
}
il void ins(int x){tmx=tmp+Dep[w],add=Dep[x],ins(rt[x],1,D,dep[w]);}
int main(){
    mx0[0]=mx1[0]=inf;
    for(scanf("%d",&T);T--;){
        scanf("%d",&n),ans=inf,D=cnt=0;
        for(i=1;i<=n;i++)
        lk[i]=son[i]=top[i]=rt[i]=0,siz[i]=1,d[i].no=i,s2[i].clear();
        for(i=1;i<n;i++)scanf("%d%d%d",&u,&v,&w),
        f[v]=u,Dep[v]=w,to[i]=v,hd[i]=lk[u],lk[u]=i;
        lg[0]=-1,dfs(1);
        for(i=1;i<=n;i++)!top[i]?top[i]=i:0,top[son[i]]=top[i];
        for(i=1;i<n<<1;i++)lg[i]=lg[i-1]+!(i&i-1);
        for(i=0;1<<i+1<cnt;i++)
        for(int j=1;j<=cnt-(1<<i+1)+1;j++)
        fa[i+1][j]=min(fa[i][j],fa[i][j+(1<<i)]);
        for(scanf("%d",&m),cnt=0;m--;){
            scanf("%d%d%lld",&u,&v,&ww),w=lca(u,v),tmp=dis(u,v)-ww;
            if(u^w)s2[nxt(u,w)].push_back((dat){u,(sig){v,tmp+Dep[u]-ww}}),ins(u);
            if(v^w)s2[nxt(v,w)].push_back((dat){v,(sig){u,tmp+Dep[v]-ww}}),ins(v);
        }
        for(i=n;i>2;i--)
        add=Dep[f[i]],del(rt[i],1,D,dep[i]-2),uni(rt[f[i]],rt[i],1,D);
        for(i=2;i<=n;i++)if(s2[i].size()>1){
            sort(s2[i].begin(),s2[i].end(),cmp);
            for(auto j:s2[i]){
                u=j.u,ad.ini(),ad.a=j.v,d[u].uni(ad);
                if(tot){
                    if(st[tot]==u)continue;
                    for(v=0,w=lca(u,st[tot]);dep[w]<dep[st[tot]];)
                    v?d[st[tot]].uni(d[v]),0:0,v=st[tot--];
                    if(v)d[w].uni(d[v]);
                    w^st[tot]?st[++tot]=w:0;
                }
                st[++tot]=u;
            }
            for(;tot>1;tot--)d[st[tot-1]].uni(d[st[tot]]);
            if(tot)d[st[1]].ini(),tot=0;
        }
        ans<-1e15?puts("F"):printf("%lld\n",ans);
    }
}