@description@

给定 N 个点,第 i 点有一个点权 Xi,再给定一棵边带权的树,第 i 条 (Ai, Bi) 边权为 Ci。

构建一个完全图,完全图中边 (i, j) 的边权为 dist(i, j) + Xi + Xj,其中 dist(i, j) 是点 i 与点 j 在树上的距离。

求该完全图的最小生成树。

Constraints

2≤N≤200000; 1≤Xi≤10^9; 1≤Ai,Bi≤N; 1≤Ci≤10^9

Input

输入形式如下:

N

X1 X2 … XN

A1 B1 C1

A2 B2 C2

:

AN−1 BN−1 CN−1

Output

输出最小生成树的边权总和。

Sample Input 1

4

1 3 5 1

1 2 1

2 3 2

3 4 3

Sample Output 1

22

选择边 (1, 2), (1, 4), (3, 4),边权总和为 5 + 8 + 9 = 22

点此跳转到题目

@solution@

完全图并不能直接套 prim 或者 kruskal。

并且也不像最小曼哈顿生成树一样,有比较好用的性质可以去除大量无用边。

考虑这类题的一个通用解法:boruvka(不会念)。

其实是基本无人问津的最小生成树算法(我也不知道为什么),但是可以将其算法思想拓展,解决一些完全图的最小生成树问题。

说是完全图,边权也有性质,而且往往会和点权挂钩(比如这道题)。

扯了这么多,所谓的 boruvka 算法是什么呢?

我们对于每一个连通块去找与它相邻的最小边。可以证明这样的边一定是存在于最小生成树之中(破圈法之类的都能证)。

于是我们把这些边加入最小生成树,并将连通块合并。

因为每次选最小边的时候,一条边最多会被选两次,所以最多执行 log 次寻找最小边的操作。

而这类题的特点是:往往寻找最小边的过程可以优化。

什么?你说这道题需要点分治来寻找路径最小值?然后复杂度就炸成 O(nlog^3n) 了?

NONONO。我们其实可以 O(n) 一次性给所有点找到它对应的最小值。

考虑所有连通块都是一个点的时候,我们可以做一个换根 dp 求出所有点的最小边(求距离一个点最近的点)。

由于是找最小值,重复经过一条边肯定不优,我们甚至不用去存 dp 的次小值,来避免从上往下传递 dp 值的时候走入同一棵子树。

这个过程是 O(n) 的。

那么连通块是一堆点的时候,我们怎么去排除与它同一连通块的点呢?

其实。。。很简单嘛。

假如最小边是同一连通块的,我们就去找与最小边不在同一连通块的次小边,不就解决了嘛。

dp 的时候存两维:最小边,与不和最小边在同一连通块的次小边。转移时讨论一下即可。

一次求最小边的复杂度为 O(n),所以总复杂度为 O(nlogn)。

@accepted code@

#include<cstdio>
#include<iostream>
using namespace std;
typedef long long ll;
#define mp make_pair
#define fi first
#define se second
typedef pair<ll, int> pli;
typedef pair<pli, pli> st;
const int MAXN = 200000;
const ll INF = (1LL<<60);
struct edge{
int to; ll dis;
edge *nxt;
}edges[2*MAXN + 5], *adj[MAXN + 5], *ecnt = edges;
void addedge(int u, int v, int w) {
edge *p = (++ecnt);
p->to = v, p->dis = w, p->nxt = adj[u], adj[u] = p;
p = (++ecnt);
p->to = u, p->dis = w, p->nxt = adj[v], adj[v] = p;
}
pli lnk[MAXN + 5];
int fa[MAXN + 5], clr[MAXN + 5];
ll X[MAXN + 5];
int find(int x) {
return fa[x] = (fa[x] == x ? x : find(fa[x]));
}
bool unite(int x, int y) {
int fx = find(x), fy = find(y);
if( fx != fy ) {
fa[fx] = fy;
return true;
}
else return false;
}
st dp[MAXN + 5];
void update(st &a, st b) {
if( b.fi.fi < a.fi.fi ) {
if( b.fi.se != a.fi.se )
a.se = a.fi;
a.fi = b.fi;
if( b.se.fi < a.se.fi )
a.se = b.se;
}
else {
if( b.fi.se != a.fi.se ) {
if( b.fi.fi < a.se.fi )
a.se = b.fi;
}
else if( b.se.fi < a.se.fi )
a.se = b.se;
}
}
void dfs1(int x, int f) {
dp[x] = mp(mp(X[x], clr[x]), mp(INF, -1));
for(edge *p=adj[x];p;p=p->nxt) {
if( p->to == f ) continue;
dfs1(p->to, x); st t = dp[p->to];
t.fi.fi += p->dis, t.se.fi += p->dis;
update(dp[x], t);
}
}
void dfs2(int x, int f, st k) {
update(dp[x], k);
for(edge *p=adj[x];p;p=p->nxt) {
if( p->to == f ) continue;
st t = dp[x];
t.fi.fi += p->dis, t.se.fi += p->dis;
dfs2(p->to, x, t);
}
}
int num[MAXN + 5];
int main() {
int N; scanf("%d", &N);
for(int i=1;i<=N;i++)
scanf("%lld", &X[i]), fa[i] = i;
for(int i=1;i<N;i++) {
int u, v, w; scanf("%d%d%d", &u, &v, &w);
addedge(u, v, w);
}
ll ans = 0;
while( true ) {
int cnt = 0;
for(int i=1;i<=N;i++)
if( fa[i] == i ) num[clr[i] = (++cnt)] = i;
if( cnt == 1 ) break;
for(int i=1;i<=N;i++)
clr[i] = clr[find(i)];
dfs1(1, 0), dfs2(1, 0, mp(mp(INF, -1), mp(INF, -1)));
for(int i=1;i<=cnt;i++)
lnk[i] = mp(INF, -1);
for(int i=1;i<=N;i++) {
if( dp[i].fi.se == clr[i] )
lnk[clr[i]] = min(lnk[clr[i]], mp(dp[i].se.fi + X[i], dp[i].se.se));
else lnk[clr[i]] = min(lnk[clr[i]], mp(dp[i].fi.fi + X[i], dp[i].fi.se));
}
for(int i=1;i<=cnt;i++)
if( unite(num[i], num[lnk[i].se]) )
ans += lnk[i].fi;
}
printf("%lld\n", ans);
}

@details@

被数据结构困住的我,用点分治 + 优先队列写了一个 prim。

然后它 MLE 了。

当时我就哭了。

咳咳。不过也算是增长了一点见识,同时还告诉我一个深刻的道理:不要被套路所困。不一定非得要数据结构才能维护的啊。

@atcoder - CODE FESTIVAL 2017 Final - J@ Tree MST的更多相关文章

  1. Atcoder CODE FESTIVAL 2016 Final G - Zigzag MST[最小生成树]

    题意:$n$个点,$q$次建边,每次建边选定$x,y$,权值$c$,然后接着$(y,x+1,c+1),(x+1,y+1,c+2),(y+1,x+2,c+3),(x+2,y+2,c+4)\dots$(画 ...

  2. 【赛时总结】 ◇赛时·IV◇ CODE FESTIVAL 2017 Final

    ◇赛时-IV◇ CODE FESTIVAL 2017 Final □唠叨□ ①--浓浓的 Festival 气氛 ②看到这个比赛比较特别,我就看了一看--看到粉粉的界面突然开心,所以就做了一下 `(* ...

  3. [AtCoder Code Festival 2017 QualB D/At3575] 101 to 010 - dp

    [Atcoder Code Festival 2017 QualB/At3575] 101 to 010 有一个01序列,每次可以选出一个101,使其变成010,问最优策略下能操作几次? 考虑像 11 ...

  4. 【AtCoder】CODE FESTIVAL 2017 Final

    A - AKIBA 模拟即可 代码 #include <bits/stdc++.h> #define fi first #define se second #define pii pair ...

  5. AtCoder Code Festival 2017 Team Relay J - Indifferent

    题目大意:共$2n$个价格$p_i$.两人轮流取.你每次取最大的,对方每次随机取.问你取的期望和是多少. 题解:从小到大排序,$\sum\limits_{i=0}^{2n-1} \frac{i*p_i ...

  6. 题解【AtCoder - CODE FESTIVAL 2017 qual B - D - 101 to 010】

    题目:https://atcoder.jp/contests/code-festival-2017-qualb/tasks/code_festival_2017_qualb_d 题意:给一个 01 串 ...

  7. 【题解】Popping Balls AtCoder Code Festival 2017 qual B E 组合计数

    蒟蒻__stdcall终于更新博客辣~ 一下午+一晚上=一道计数题QAQ 为什么计数题都这么玄学啊QAQ Prelude 题目链接:这里是传送门= ̄ω ̄= 下面我将分几个步骤讲一下这个题的做法,大家不 ...

  8. Atcoder CODE FESTIVAL 2017 qual C D - Yet Another Palindrome Partitioning 回文串划分

    题目链接 题意 给定一个字符串(长度\(\leq 2e5\)),将其划分成尽量少的段,使得每段内重新排列后可以成为一个回文串. 题解 分析 每段内重新排列后是一个回文串\(\rightarrow\)该 ...

  9. Atcoder CODE FESTIVAL 2017 qual C C - Inserting 'x' 回文串

    题目链接 题意 给定字符串\(s\),可以在其中任意位置插入字符\(x\). 问能否得到一个回文串,若能,需插入多少个\(x\). 思路 首先统计出现次数为奇数的字符\(cnt\). \(cnt\ge ...

随机推荐

  1. beego 批量删除问题

    o := orm.NewOrm()    qs := o.QueryTable(new(ExecutionJobs))    javaTimestamp = 1557738394000    qs = ...

  2. java知识点---文件分隔符

    本篇讲述java编程中,怎样解决跨平台时,因不同系统中分隔符不同导致的文件或路径找不到的问题 首先来看两个例子: 一.linux系统和windows系统中的文件路径: Linux系统: Windows ...

  3. Jboss配置HTTPS

    配置jboss的HTTP请求走SSL(HTTPS协议) l         生成keystore 文件 用keytool生成server.keystore文件: 进入命令行 C:\Documents ...

  4. 阿里云maven中央仓库

    1.配置文件路径:config/settings.xml 2.<mirrors>镜像集合中添加镜像 <mirror> <id>alimaven</id> ...

  5. day36 03-Hibernate检索方式:排序、参数绑定、投影查询

    排序之后是分页查询. 检索单个对象 还可以进行参数的绑定. HQL的参数绑定,按参数名称绑定或者是按参数位置绑定. 还可以用投影的操作,投影的操作是只查询这里面的某几个属性.只查询某一个属性,查询多个 ...

  6. vue 使用 element ui动态添加表单

    html部分 <div class="hello"> <el-form :model="dynamicValidateForm" ref=&q ...

  7. 【洛谷】P1876 开灯

    P1876 开灯 题目背景 该题的题目是不是感到很眼熟呢? 事实上,如果你懂的方法,该题的代码简直不能再短. 但是如果你不懂得呢?那...(自己去想) 题目描述 首先所有的灯都是关的(注意是关!),编 ...

  8. 【CodeVS】1978 Fibonacci数列3

    1978 Fibonacci数列 3 时间限制: 1 s 空间限制: 64000 KB 题目等级 : 青铜 Bronze 题目描述 Description 斐波纳契数列是这样的数列: f1 = 1 f ...

  9. Android的ADB学习笔记

    1.ADB的常用命令   Pull命令:adb -e|-d pull {文件的路径} {获取文件路径} 2. 文件操作的基本命令 ls -al:显示当下目录下用户对文件的操作权限.  = la -al ...

  10. django中静态资源

    创建静态资源存放路径,为了设置静态媒体,你需要设立存储它们的目录.在你的项目目录(例如/myproject/),创建叫做static的目录.在static里再创建一个images目录和js目录 设置项 ...