JZOJ 5926. 【NOIP2018模拟10.25】naive 的图

Description

众所周知，小 naive 有一张 n 个点，m 条边的带权无向图。第 i 个点的颜色为 ci。d(s, t)表示从点 s 到点 t 的权值最小的路径的权值，一条路径的权值定义为路径上权值最大的边的权值。
求所有满足 u < v, |cu − cv| ≥ L 的点对 (u, v) 的 d(u, v) 之和。

Input

输入文件为 graph.in。
第一行，三个整数 n, m, L，表示点数，边数和参数 L。
第二行，n 个整数，第 i 个数为第 i 个点的颜色 ci。接下来 m 行，每行三个整数 ui, vi, wi，描述了一条边。

Output

输出文件为 graph.out。
共一行，一个整数，表示答案。

Sample Input

4 5 2
6 4 5 2
1 2 8
2 3 7
2 4 8
1 3 2
1 4 1

Sample Output

17

样例解释

满足条件的点对：(1, 2),(1, 4),(2, 4),(3, 4)，答案为 7 + 1 + 7 + 2 = 17。

Data Constraint

对于每个测试点内的数据：

Solution

• 首先我们发现两点间的最短距离所经过的边都在最小生成树上。

• 于是我们建出Kruskal重构树！

• 什么叫Kruskal重构树呢？

• 我们在做Kruskal算法时不是每次选择了一条边作为最小生成树上的一条边吗？

• 这是我们新建一个点分别连向那条边的两个端点，并将点权设为那条边的边权。

• 那么这个重构树有什么性质呢？

1. 这是一个有 $2n-1$ 个点的二叉树，每个叶子结点都是原图中的点；
2. 原树两点之间的边权最大值是重构树上两点Lca的权值；

• 于是当 $L=0$ 时，一条边（最小生成树上）的贡献次数就是其两个子树的 Size 取 min 。

• 那么我们遍历这个重构树，对于一个非叶子结点，枚举其较小的子树，

• 我们需要计算出每个点与其较大子树中的点匹配所产生的答案。

• 容易想到离线后做二维数点，一维是dfs序（连续一段），

• 另一维是颜色c（两段，$c_0\le c[x]-L或c_0\ge c[x]+L$）。

• 于是这样排序后用树状数组计算即可。

• 时间复杂度 $O(nlo{g}^{2}n)$ 。

• 这题还能用点分治做，我们考虑在最小生成树上做点分治。

• 对于一个点为根的子树，我们将其子树内的点按边权mx从小到大排序。

• 之后枚举最大值是多少，那么前面的符合条件点就能用树状数组维护了。

• 时间复杂度同样是 $O(nlo{g}^{2}n)$ 。

Code

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cctype>
using namespace std;
typedef long long LL;
const int N=2e5+5;
struct data
{
	int x,y,z;
}a[N*3],b[N*18],d[N*18];
int n,m,L,tot,cnt,num1,num2,pd;
LL ans;
int l[N<<1],r[N<<1],size[N<<1],dfn[N<<1];
int c[N],v[N<<1],f[N<<1];
inline int read()
{
    int X=0,w=0; char ch=0;
    while(!isdigit(ch)) w|=ch=='-',ch=getchar();
    while(isdigit(ch)) X=(X<<3)+(X<<1)+(ch^48),ch=getchar();
    return w?-X:X;
}
inline bool cmp(data x,data y)
{
	return x.z<y.z;
}
inline bool cmpx(data x,data y)
{
	return x.x<y.x || x.x==y.x && x.z<0 && y.z>=0;
}
inline bool cmpy(data x,data y)
{
	return x.x>y.x || x.x==y.x && x.z<0 && y.z>=0;
}
int get(int x)
{
	return f[x]==x?x:f[x]=get(f[x]);
}
void dg(int x,int y)
{
	if(size[x]==1)
	{
		if(c[x]-L>=0) b[++num1]=(data){c[x]-L,r[y],v[y]};
		d[++num2]=(data){c[x]+L+pd,r[y],v[y]};
		return;
	}
	dg(l[x],y);
	dg(r[x],y);
}
void dfs(int x)
{
	size[x]=1;
	dfn[x]=++tot;
	if(!l[x])
	{
		b[++num1]=(data){c[x],dfn[x],-1};
		d[++num2]=(data){c[x],dfn[x],-1};
		return;
	}
	dfs(l[x]);
	size[x]+=size[l[x]];
	dfs(r[x]);
	size[x]+=size[r[x]];
	if(size[l[x]]>size[r[x]]) swap(l[x],r[x]);
	dg(l[x],x);
}
inline void change(int x)
{
	while(x<=cnt) f[x]++,x+=x&-x;
}
inline int find(int x)
{
	int sum=0;
	while(x) sum+=f[x],x-=x&-x;
	return sum;
}
int main()
{
	freopen("graph.in","r",stdin);
	freopen("graph.out","w",stdout);
	n=cnt=read(),m=read(),L=read(),pd=!L?1:0;
	for(int i=1;i<=n;i++) c[f[i]=i]=read();
	for(int i=1;i<=m;i++) a[i].x=read(),a[i].y=read(),a[i].z=read();
	sort(a+1,a+1+m,cmp);
	for(int i=1,k=1;i<=m;i++)
	{
		int x=get(a[i].x),y=get(a[i].y);
		if(x^y)
		{
			f[++cnt]=cnt;
			f[x]=f[y]=cnt;
			l[cnt]=x;
			r[cnt]=y;
			v[cnt]=a[i].z;
			if(++k==n) break;
		}
	}
	tot=0;
	dfs(cnt);
	sort(b+1,b+1+num1,cmpx);
	memset(f,0,sizeof(f));
	for(int i=1;i<=num1;i++)
		if(b[i].z<0)
		{
			change(b[i].y);
		}else
		{
			int num=find(dfn[b[i].y]+size[b[i].y]-1)-find(dfn[b[i].y]-1);
			ans+=(LL)num*b[i].z;
		}
	sort(d+1,d+1+num2,cmpy);
	memset(f,0,sizeof(f));
	for(int i=1;i<=num2;i++)
		if(d[i].z<0)
		{
			change(d[i].y);
		}else
		{
			int num=find(dfn[d[i].y]+size[d[i].y]-1)-find(dfn[d[i].y]-1);
			ans+=(LL)num*d[i].z;
		}
	printf("%lld",ans);
	return 0;
}