【JZOJ5433】图

description

有一个n个点A+B条边的无向连通图，有一变量x，每条边的权值都是一个关于x的简单多项式，其中有A条边的权值是k+x，另外B条边的权值是k-x，如果只保留权值形如k+x的边，那么这个图仍是一个连通图，如果只保留权值形如k-x的边，这个图也依然是一个连通图。
给出q组询问，每组询问给出x的值，问此时这个无向连通图的最小生成树权值是多少。

---

analysis

• 首先容易知道图任意时候的$MST$只由$A$集合或$B$集合内的边组成

• 当$x$从$-∞$逐渐变为$+∞$，$MST$会从只由$A$的边构成逐渐变成只由$B$的边构成

• 这里$MST$只会变化$n-1$次，那么用$LCT$维护$MST$的添边删边

• 当然先$kruskal$把$A$和$B$各有用的$n-1$条边搞出来，把$A$的边塞到$LCT$上面

• 升序排序$B$的边一条条加进$LCT$里面，然后代替掉新环里$k$最大的$A$边

• 用当前$B$边的$k-x$减去删去该$A$边的$k+x$，会得到$k'-2x$一类式子

• 注意如果查询的环边上有$B$边，因为只换掉$A$边无视$B$边，所以连$B$边时不用连边节点

• 如果有多条$B$边，由于升序加入$LCT$，所以$k$更小的$B$边会贡献得早

• 对于最后的求答案，把询问排序，且已经得到$n-1$条$k-2x$这样的式子

• 把式子再按$k$排序，看每个询问会用到前替换的多少条边（也就是前多少条式子）

• 扫一遍这堆式子，只要$k-2*$当前询问的数$<=0$，说明替换这条边是有贡献的，继续扫下去

• 注意答案还要算上每个$k$以及剩下还没有被替换的$A$边里的$+x$

---

code

#pragma GCC optimize("O3")
#pragma G++ optimize("O3")
#include<stdio.h>
#include<string.h>
#include<algorithm>
#define MAXN 500005
#define ha 1926081719491001
#define ll long long
#define reg register ll
#define fo(i,a,b) for (reg i=a;i<=b;++i)
#define fd(i,a,b) for (reg i=a;i>=b;--i)

using namespace std;

ll tr[MAXN][2],fa[MAXN],pf[MAXN],st[MAXN],fat[MAXN],val[MAXN],answer[MAXN];
ll n,A,B,q,tot,cnt,ans;
struct quiry{ll x,y;}inquiry[MAXN];
struct edge{ll x,y,z;}f[2][MAXN],g[MAXN];
struct node{ll mx,val,size;bool rev;}a[MAXN];
inline ll read(){ll x=0,f=1;char ch=getchar();while (ch<'0' || '9'<ch){if (ch=='-')f=-1;ch=getchar();}while ('0'<=ch && ch<='9')x=x*10+ch-'0',ch=getchar();return x*f;}
inline void swap(ll &x,ll &y){ll z=x;x=y,y=z;}
inline bool cmp(edge a,edge b){return a.z<b.z;}
inline bool cmpp(quiry a,quiry b){return a.x<b.x;}
inline ll getfa(ll x){return !fat[x]?x:fat[x]=getfa(fat[x]);}
inline void update(ll x){if (!x)return;a[x].mx=x;if (a[a[tr[x][0]].mx].val>a[a[x].mx].val)a[x].mx=a[tr[x][0]].mx;if (a[a[tr[x][1]].mx].val>a[a[x].mx].val)a[x].mx=a[tr[x][1]].mx;a[x].size=a[tr[x][0]].size+a[tr[x][1]].size+1;}
inline void reverse(ll x){if (x)swap(tr[x][0],tr[x][1]),a[x].rev^=1;}
inline void down(ll x){if (a[x].rev)reverse(tr[x][0]),reverse(tr[x][1]),a[x].rev=0;}
inline void downdata(ll x){while (x)st[++st[0]]=x,x=fa[x];while (st[0])down(st[st[0]--]);}
inline ll lr(ll x){return tr[fa[x]][1]==x;}
inline void rotate(ll x){ll y=fa[x],k=lr(x);tr[y][k]=tr[x][!k];if (tr[x][!k])fa[tr[x][!k]]=y;fa[x]=fa[y];if (fa[y])tr[fa[y]][lr(y)]=x;tr[x][!k]=y,fa[y]=x,pf[x]=pf[y],update(y),update(x);}
inline void splay(ll x,ll y){downdata(x);while (fa[x]!=y){if (fa[fa[x]]!=y)rotate(lr(fa[x])==lr(x)?fa[x]:x);rotate(x);}}
inline void access(ll x){for (ll y=0;x;update(x),y=x,x=pf[x])splay(x,0),fa[tr[x][1]]=0,pf[tr[x][1]]=x,tr[x][1]=y,fa[y]=x,pf[y]=0;}
inline void makeroot(ll x){access(x),splay(x,0),reverse(x);}
inline void link(ll x,ll y){makeroot(x),pf[x]=y;}
inline void cut(ll x,ll y){makeroot(x),access(y),splay(x,0),tr[x][1]=fa[y]=pf[y]=0,update(x);}
inline ll query(ll x,ll y){makeroot(x),access(y),splay(y,0);return a[y].mx;}
int main()
{
	//freopen("T3.in","r",stdin);
	freopen("graph.in","r",stdin);
	freopen("graph.out","w",stdout);
	n=read(),A=read(),B=read(),q=read();
	fo(i,1,A)f[0][i].x=read(),f[0][i].y=read(),f[0][i].z=read();
	fo(i,1,B)f[1][i].x=read(),f[1][i].y=read(),f[1][i].z=read();
	sort(f[0]+1,f[0]+A+1,cmp),sort(f[1]+1,f[1]+B+1,cmp);
	fo(i,0,n)a[i].val=-ha;
	fo(i,1,A)
	{
		ll x=f[0][i].x,y=f[0][i].y,z=f[0][i].z;
		if (getfa(x)!=getfa(y))fat[getfa(x)]=getfa(y),ans+=z,
		a[n+i].val=z,a[n+i].mx=n+i,link(x,n+i),link(n+i,y);
	}
	memset(fat,0,sizeof(fat));
	fo(i,1,B)
	{
		ll x=f[1][i].x,y=f[1][i].y,z=f[1][i].z;
		if (getfa(x)!=getfa(y))fat[getfa(x)]=getfa(y),g[++tot]=f[1][i];
	}
	fo(i,1,tot)
	{
		ll x=g[i].x,y=g[i].y,z=g[i].z,tmp=query(x,y);
		if (tmp<=n)continue;
		cut(f[0][tmp-n].x,tmp),cut(tmp,f[0][tmp-n].y);
		link(x,y),val[++cnt]=z-a[tmp].val;
	}
	fo(i,1,q)inquiry[i].x=read(),inquiry[i].y=i;
	sort(inquiry+1,inquiry+q+1,cmpp),sort(val+1,val+cnt+1);
	ll m=1;
	fo(i,1,q)
	{
		while (m<=tot && val[m]<=inquiry[i].x*2)ans=ans+val[m++];
		answer[inquiry[i].y]=ans+inquiry[i].x*(n-(m-1)*2-1);
	}
	fo(i,1,q)printf("%lld\n",answer[i]);
	return 0;
}