题意:给一棵树,求任意{两点路径上的最大边权值-最小边权值}的总和。
解法:
sigma(maxVal[i]−minVal[i])=sigma(maxVal)−sigma(minVal)
;所以我们分别求所有两点路径上的最大值的和,还有最小值的和。再相减就可以了。求最大值的和的方法用带权并查集,把边按权值从小到大排序,一条边一条边的算,当我们算第
i
条边的时候权值为
wi
,两点是
ui,vi
,前面加入的边权值一定是小于等于当前
wi
的,假设与
ui
连通的点有
a
个,与
vi
连通的点有
b
个,那么在
a
个中选一个,在
b
个中选一个,这两个点的路径上最大值一定是
wi
,一共有
a∗b
个选法,爱情经验值为
a∗b∗wi
。
求最小值的和的方法类似。
槽点:
一:这题做数据的时候突然想到的把数据范围设在 unsigned long long 范围内,要爆 long long,这样选手在wa了之后可能心态不好找不到这个槽点,当是锻炼大家的心态和出现wa时的找错能力了,把这放在pretest..很良心的。
二,并查集的时候,用是递归的需要扩栈,一般上
10w
的递归都需要,所以看见有几个FST在栈溢出的,好桑心。
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
using namespace std;
typedef unsigned long long ll;
const int maxn = 150010;
ll maxv,minv;
int fa[maxn];
ll cnt[maxn];
struct node
{
int u,v;
ll w;
}e[maxn];
bool cmp1(node x,node y)
{
return x.w<y.w;
}
bool cmp2(node x,node y)
{
return x.w>y.w;
}
int find(int x)
{
return fa[x]==x?x:(fa[x]=find(fa[x]));
}
void merge(int x,int y,ll z,int flag)
{
int fx = find(x);
int fy = find(y);
if(fx!=fy)
{
if(flag) maxv += cnt[fx]*cnt[fy]*z;
else minv += cnt[fx]*cnt[fy]*z;
fa[fx] = fy;
cnt[fy] += cnt[fx];
//cnt[fx] = 0;
}
}
int main()
{
int n,x,y,cas=1;
ll z;
while(~scanf("%d",&n))
{
for(int i=1;i<=n;i++) fa[i]=i,cnt[i]=1;
for(int i=1;i<n;i++)
scanf("%d%d%I64ud",&e[i].u,&e[i].v,&e[i].w);
maxv = minv = 0;
sort(e+1,e+n,cmp1);
for(int i=1;i<n;i++)
{
x = e[i].u;
y = e[i].v;
z = e[i].w;
merge(x,y,z,1);
}
for(int i=1;i<=n;i++) fa[i]=i,cnt[i]=1;
sort(e+1,e+n,cmp2);
for(int i=1;i<n;i++)
{
x = e[i].u;
y = e[i].v;
z = e[i].w;
merge(x,y,z,0);
}
printf("Case #%d: %I64u\n",cas++,maxv-minv);
}
return 0;
}
对于ULL , 可以用 cout , %I64u , %llu 输出 。