hihocoder #1112 树上的好路径
时间限制:1000ms
单点时限:1000ms
内存限制:256MB
描述
现在有一棵有N个带权顶点的树,顶点编号为1,2,...,N。我们定义一条路径的次小(最小)权为它经过的所有顶点(包括起点和终点)中权值次小(最小)顶点的权值。现在给定常数c,你需要求出:存在多少个使得u<v的顶点组(u,v),满足从u到v的最短路的次小权恰为c但最小权不为c。
输入
第一行有两个数N和c。(1<=n<=100000)
第二行N个数,依次表示每个顶点的权值。
接下来N-1行,每行两个数,代表这棵树的一条边所连接的两个顶点的编号。
我们保证输入中的数都在int以内。
输出
一个数,为答案。
样例输入
8 2
2 2 3 3 1 2 3 2
1 2
3 2
3 8
4 2
5 2
5 6
6 7
样例输出
17
Solution
为了方便, 把我们要考虑的树记作$T=(V, E)$, 用$w[u]$表示节点$u$ ($u\in V$) 的权值.
先考虑一个简化的问题:
求最小权小于$c$且次小权不小于$c$的路径$(u, v)$的数目.
为了解决这个问题, 我们考虑如下的添边过程:
我们考虑一个动态的图$S(V, E'), E'\subseteq E$.
从$S=(V, \emptyset)$开始, 先把所有满足$w[u]\ge c \land w[v] \ge c$的边$(u, v)$加到$S$中,
然后考虑满足
\[w[u]<c \land w[v]\ge c \lor w[u]\ge c \land w[v] <c\]
的边$(u, v)$, 不失一般性, 不妨设 $w[u]<c, w[v]\ge c$.
我们先把$u$固定为$u_0$, 考虑若将所有符合上述条件的边$\{(u_0, v)\}$加到$s$中将能获得多少满足条件的路径.
显然这些满足条件的路径上的最小权就是$w[u_0]$.
(未完待续...)
(无力写了, 先把代码贴上)
UPD
前面写得太罗嗦了, 结果现在自己都看不大懂了. 其实做法一句话就能说清楚:
最小权小于$c$, 次小权不小于$c$的路径数 $-$ 最小权小于$c$, 次小权大于$c$的路径数
Implementation
#include <bits/stdc++.h>
using namespace std;
using LL=long long;
const int N{<<}; int a[N]; struct edge{
int u, v;
void read(){
cin>>u>>v;
}
}e[N]; struct DSU{
int par[N], size[N];
int n;
DSU(int n):n(n){}
void init(){
for(int i=; i<=n; i++){
par[i]=i;
size[i]=;
}
}
int find(int x){
return x==par[x]?x:par[x]=find(par[x]);
}
void unite(int x, int y){
x=find(x), y=find(y);
if(x!=y) par[x]=y, size[y]+=size[x];
}
}; vector<int> f[N]; void prep(DSU &b, int n, int c){
b.init();
for(int i=; i<=n; i++) f[i].clear();
for(int i=; i<n; i++){
int u=e[i].u, v=e[i].v;
if(a[u]>=c && a[v]>=c){
b.unite(u, v);
}
}
} int main(){
int n, c;
cin>>n>>c;
DSU b(n); for(int i=; i<=n; i++)
cin>>a[i];
for(int i=; i<n; i++)
e[i].read(); LL res=; prep(b, n, c); for(int i=; i<n; i++){
int u=e[i].u, v=e[i].v;
if(a[u]<c ^ a[v]<c){ //tricky
// cout<<u<<' '<<v<<endl;
if(a[v]<c) swap(u, v);
int rv=b.find(v);
// res+=LL(b.size[u])*LL(b.size[v]);
// if(ru!=rv)
f[u].push_back(b.size[rv]);
}
} for(int i=; i<=n; i++){
// if(f[i].size()) cout<<"#"<<i<<endl;
LL sum=, t=;
for(auto &x: f[i])
sum+=x;
for(auto &x: f[i]) t+=LL(x)*(sum-x);
res+=t>>;
res+=sum;
} prep(b, n, c+); for(int i=; i<n; i++){
int u=e[i].u, v=e[i].v;
if(a[u]<c && a[v]>c || a[u]>c && a[v]<c){ //tricky
if(a[v]<c) swap(u, v);
int rv=b.find(v);
// res+=LL(b.size[u])*LL(b.size[v]);
f[u].push_back(b.size[rv]);
}
} for(int i=; i<=n; i++){
LL sum=, t=;
for(auto &x: f[i])
sum+=x;
for(auto &x: f[i]) t+=LL(x)*(sum-x);
res-=t>>, res-=sum;
} cout<<res<<endl;
}
hihocoder #1112 树上的好路径的更多相关文章
- Codefroces Gym 100781A(树上最长路径)
http://codeforces.com/gym/100781/attachments 题意:有N个点,M条边,问对两两之间的树添加一条边之后,让整棵大树最远的点对之间的距离最近,问这个最近距离是多 ...
- Codeforces Round #526 D - The Fair Nut and the Best Path /// 树上两点间路径花费
题目大意: 给定一棵树 树上每个点有对应的点权 树上每条边有对应的边权 经过一个点可得到点权 经过一条边必须花费边权 即从u到v 最终得分=u的点权-u到v的边权+v的点权 求树上一条路径使得得分最大 ...
- 【POJ 3162】 Walking Race (树形DP-求树上最长路径问题,+单调队列)
Walking Race Description flymouse's sister wc is very capable at sports and her favorite event is ...
- ●hihocoder #1394 网络流四·最小路径覆盖
题链: http://hihocoder.com/problemset/problem/1394 题解: 有向图最小路径覆盖:最少的路径条数不重不漏的覆盖所有点. 注意到在任意一个最小路径覆盖的方案下 ...
- hihoCoder 1394 : 网络流四·最小路径覆盖
题目链接:https://hihocoder.com/problemset/problem/1394 题目说是网络流,但是其实就是求有向无环图的最小路径覆盖. 不会网络流,只好用二分匹配了. 把每个点 ...
- hihocoder 1676 树上等差数列 黑科技树形dp
#1676 : 树上的等差数列 时间限制:10000ms 单点时限:1000ms 内存限制:256MB 描述 给定一棵包含N个节点的无根树,节点编号1~N.其中每个节点都具有一个权值,第i个节点的权值 ...
- hihocoder #1394 : 网络流四·最小路径覆盖(最小路径覆盖)
#1394 : 网络流四·最小路径覆盖 时间限制:10000ms 单点时限:1000ms 内存限制:256MB 描述 国庆期间正是旅游和游玩的高峰期. 小Hi和小Ho的学习小组为了研究课题,决定趁此机 ...
- HDU 6043:Colorful Tree(树上统计所有路径总颜色数)***
题目链接 题意 给出一棵有n个结点的树,每个结点有一个颜色,问在这棵树的所有路径中,每条路径的颜色数求和是多少. 思路 求每种颜色的贡献可以转化为总的和减去每种颜色在哪些路径上没有出现的贡献,一个颜色 ...
- SPOJ 1825 经过不超过K个黑点的树上最长路径 点分治
每一次枚举到重心 按子树中的黑点数SORT一下 启发式合并 #include<cstdio> #include<cstring> #include<algorithm&g ...
随机推荐
- 如何重现难以重现的bug
生活中有这么一种现象:如果你关注某些东西,它就会经常出现在你眼前,例如一个不出名的歌手的名字,一种动物的卡通形象,某个非常专业的术语,等等等等.这种现象也叫做“孕妇效应”.还有类似的一种效应叫做“视网 ...
- 利用Canvas进行绘制XY坐标系
首先来一发图 绘制XY的坐标主要是利用Canvas setLeft和setBottom功能(Canvas内置坐标的功能) 1.首先WPF中的坐标系都是从左到右,从上到下的 即左上角位置(0,0)点,所 ...
- Java:泛型
一.序言 变化一: 在引入范型之前,Java中的类型分为原始类型.复杂类型,其中复杂类型分为数组和类:引入范型后,一个复杂类型可以细分成更多的类型. 例如,原先的List类型,现在细分成List< ...
- C语言字符串与数字相互转换
在C/C++语言中没有专门的字符串变量,通常用字符数组来存放字符串.字符串是以“\0”作为结束符.C/C++提供了丰富的字符串处理函数,下面列出了几个最常用的函数. ● 字符串输出函数puts. ● ...
- XSS attack
<html> <form action="" method="post"> <input type="text" ...
- Linux chkconfig命令
chkconfig命令主要用来更新(启动或停止)和查询系统服务的运行级信息.谨记chkconfig不是立即自动禁止或激活一个服务,它只是简单的改变了符号连接. 使用语法:chkconfig [--ad ...
- 利用ajaxfileupload.js异步上传文件
1.引入ajaxfileupload.js 2.html代码 <input type="file" id="enclosure" name="e ...
- git 创建别名
git config --global alias.shortname command 例子如下 git config --global alias.psm 'push origin master' ...
- 【POJ 2318】TOYS 叉积
用叉积判断左右 快速读入写错了卡了3小时hhh #include<cmath> #include<cstdio> #include<cstring> #includ ...
- python 中的sort 和java中的Collections.sort()函数的使用
x=[1,2,3] x.sort()对的,x这个都变了 y=x.sort()错误 y=sorted(x)对的,x拍好序的一个副本 python中用匿名函数和自定义函数排序:(很奇怪的是比较函数返回的是 ...