[NOIP2007 提高组] 树网的核
一、题目描述
设
T
=
(
V
,
E
,
W
)
T=(V,E,W)
T=(V,E,W) 是一个无圈且连通的无向图(也称为无根树),每条边都有正整数的权,我们称
T
T
T 为树网(treenetwork
),其中
V
V
V,
E
E
E 分别表示结点与边的集合,
W
W
W 表示各边长度的集合,并设
T
T
T 有
n
n
n 个结点。
路径:树网中任何两结点
a
a
a,
b
b
b 都存在唯一的一条简单路径,用
d
(
a
,
b
)
d(a, b)
d(a,b) 表示以
a
,
b
a, b
a,b 为端点的路径的长度,它是该路径上各边长度之和。我们称
d
(
a
,
b
)
d(a, b)
d(a,b) 为
a
,
b
a, b
a,b 两结点间的距离。
D ( v , P ) = min { d ( v , u ) } D(v, P)=\min\{d(v, u)\} D(v,P)=min{d(v,u)}, u u u 为路径 P P P 上的结点。
树网的直径:树网中最长的路径成为树网的直径。对于给定的树网 T T T,直径不一定是唯一的,但可以证明:各直径的中点(不一定恰好是某个结点,可能在某条边的内部)是唯一的,我们称该点为树网的中心。
偏心距 E C C ( F ) \mathrm{ECC}(F) ECC(F):树网 T T T 中距路径 F F F 最远的结点到路径 F F F 的距离,即
E C C ( F ) = max { D ( v , F ) , v ∈ V } \mathrm{ECC}(F)=\max\{D(v, F),v \in V\} ECC(F)=max{D(v,F),v∈V}
任务:对于给定的树网
T
=
(
V
,
E
,
W
)
T=(V, E, W)
T=(V,E,W) 和非负整数
s
s
s,求一个路径
F
F
F,他是某直径上的一段路径(该路径两端均为树网中的结点),其长度不超过
s
s
s(可以等于
s
s
s),使偏心距
E
C
C
(
F
)
\mathrm{ECC}(F)
ECC(F) 最小。我们称这个路径为树网
T
=
(
V
,
E
,
W
)
T=(V, E, W)
T=(V,E,W) 的核(Core
)。必要时,
F
F
F 可以退化为某个结点。一般来说,在上述定义下,核不一定只有一个,但最小偏心距是唯一的。
下面的图给出了树网的一个实例。图中,
A
−
B
A-B
A−B 与
A
−
C
A-C
A−C 是两条直径,长度均为
20
20
20。点
W
W
W 是树网的中心,
E
F
EF
EF 边的长度为
5
5
5。如果指定
s
=
11
s=11
s=11,则树网的核为路径DEFG
(也可以取为路径DEF
),偏心距为
8
8
8。如果指定
s
=
0
s=0
s=0(或
s
=
1
s=1
s=1、
s
=
2
s=2
s=2),则树网的核为结点
F
F
F,偏心距为
12
12
12。
输入格式
共 n n n 行。
第 1 1 1 行,两个正整数 n n n 和 s s s,中间用一个空格隔开。其中 n n n 为树网结点的个数, s s s 为树网的核的长度的上界。设结点编号以此为 1 , 2 … , n 1,2\dots,n 1,2…,n。
从第
2
2
2 行到第
n
n
n 行,每行给出
3
3
3 个用空格隔开的正整数
u
,
v
,
w
u, v, w
u,v,w,依次表示每一条边的两个端点编号和长度。例如,2 4 7
表示连接结点
2
2
2 与
4
4
4 的边的长度为
7
7
7。
输出格式
一个非负整数,为指定意义下的最小偏心距。
样例 #1
样例输入 #1
5 2
1 2 5
2 3 2
2 4 4
2 5 3
样例输出 #1
5
样例 #2
样例输入 #2
8 6
1 3 2
2 3 2
3 4 6
4 5 3
4 6 4
4 7 2
7 8 3
样例输出 #2
5
提示
- 对于 40 % 40\% 40% 的数据,保证 n ≤ 15 n \le 15 n≤15。
- 对于 70 % 70\% 70% 的数据,保证 n ≤ 80 n \le 80 n≤80。
- 对于 100 % 100\% 100% 的数据,保证 2 ≤ n ≤ 300 2\le n \le 300 2≤n≤300, 0 ≤ s ≤ 1 0 3 0\le s\le10^3 0≤s≤103, 1 ≤ u , v ≤ n 1 \leq u, v \leq n 1≤u,v≤n, 0 ≤ w ≤ 1 0 3 0 \leq w \leq 10^3 0≤w≤103。
二、解题思路
先找到直径,然后通过尺取算法寻找满足路径长度不超过 s s s 的路径,并不断更新最小偏心距。最后计算每个节点的贡献,得到最终结果
整体思路如下:
1.建图:根据输入构建树网的图结构,使用邻接表存储图。
2.求直径端点:从任意一个节点开始,通过深度优先搜索(DFS)找到离该节点最远的节点 u u u,然后从节点 u u u 出发,再次进行 DFS,找到离 u u u 最远的节点 v v v。这样, u u u 和 v v v 就是直径的两个端点。
3.尺取算法:从直径的其中一个端点开始,向另一个端点移动,并尽量保持路径的长度不超过 s s s。具体做法是使用尺取算法,不断缩短路径的长度,保证路径两端点到直径端点的最小贡献尽量小。
4.更新最小偏心距:在尺取算法的过程中,记录每次路径的两端点到直径端点的最小贡献,并不断更新最小偏心距。
5.标记直径:标记直径上的所有节点,并重新计算每个节点到直径的距离。
6.重新计算贡献:重新计算每个节点到直径的距离,以确保每个节点的贡献正确。
7.计算最终结果:遍历所有节点,找到每个节点的贡献,并取最大值作为最终结果,即最小偏心距。
8.输出结果:输出最小偏心距作为答案。
三、参考代码
M :为 500005,用于表示图的最大边数
n: 图的节点数
m: 核的长度上界
x: 边的起点
y: 边的终点
z: 边的权值
id: 用于遍历的临时变量
top: 最长路径的端点
ans: 最小偏心距的值(初始化为一个较大的数)
dis[M]: 存储节点距离
fa[M]: 存储父节点
head[M]: 存储链表头节点
bool mark[M]:声明了一个布尔数组,用于标记节点是否在直径上。
edge { int to, w, nxt; } E[M << 1]:定义了一个结构体 edge,表示图的边,包括边的终点、权值、以及下一条边的位置。然后定义了一个结构体数组 E,用于存储图的边。
#include <cstdio>
#include <algorithm>
#define M 500005
using namespace std;
int n, m, x, y, z, k, id, top, ans = 2e9;
int dis[M], fa[M], head[M];
bool mark[M];
struct edge {
int to, w, nxt;
} E[M << 1];
void add(int u, int v, int w):定义了一个函数 add,用于向图中添加边。
E[++id] = ((edge){v, w, head[u]}):将边的信息存入 E 数组中,id 自增表示存储边的位置,edge 是结构体,表示边的终点、权值、以及下一条边的位置。
head[u] = id:更新链表头节点,表示以节点 u 为起点的第一条边的位置。
void add(int u, int v, int w) {
E[++id] = ((edge){v, w, head[u]});
head[u] = id;
}
void dfs(int f, int x):定义了一个深度优先搜索函数 dfs,用于计算节点之间的距离,并找到直径的端点。
fa[x] = f;:更新节点 x 的父节点为 f。
if (dis[x] > dis[k]) k = x;:如果节点 x 到根节点的距离大于当前直径的端点 k 到根节点的距离,则更新直径的端点为 x。
for (int i = head[x]; i; i = E[i].nxt):遍历以节点 x 为起点的边。
int y = E[i].to;:获取边 i 的终点 y。
if (y == f || mark[y]) continue;:如果节点 y 是父节点或者已经在直径上,则跳过这条边。
dis[y] = dis[x] + E[i].w;:更新节点 y 到根节点的距离。
dfs(x, y);:递归调用 dfs 函数,继续搜索节点 y 的子节点。
void dfs(int f, int x) {
fa[x] = f;
if (dis[x] > dis[k]) k = x;
for (int i = head[x]; i; i = E[i].nxt) {
int y = E[i].to;
if (y == f || mark[y]) continue;
dis[y] = dis[x] + E[i].w;
dfs(x, y);
}
}
主函数部分:
scanf(“%d%d”, &n, &m);:读取节点数 n 和核的长度上界 m。
for (int i = 1; i < n; i++) { … }:循环读入边的信息,并添加到图中。
dis[1] = 1, dfs(0, 1);:初始化根节点到其他节点的距离,并进行深度优先搜索找到直径的一个端点。
dis[k] = 0, dfs(0, k);:将直径的一个端点到根节点的距离设为零,并再次进行深度优先搜索,找到直径的另一个端点。
top = k;:将直径的一个端点赋值给 top,作为直径的一个端点。
for(int i=top,j=top,l=1,r=0;i;i=fa[i]){:这是一个循环,从直径的端点 top 开始,向直径的另一个端点移动,即从一个端点沿着父节点不断向根节点移动。循环条件是 i 不为零,表示还未遍历完整条直径。
while(dis[j]-dis[i]>m) j=fa[j];:这是一个尺取的过程,j 向直径的另一个端点移动,同时保证路径长度不超过 s s s。即当直径上的节点到当前节点 i 的距离大于 s s s 时,不断向直径的另一个端点移动。
x=max(dis[top]-dis[j],dis[i]);:计算当前路径的两端点到直径端点的最小贡献 x。其中 dis[top] - dis[j] 表示直径的两个端点到当前节点 j 的距离之差,即路径的长度;dis[i] 表示当前节点 i 到直径端点的距离。因为我们要找的是路径长度不超过 s s s,所以取这两个值的最大值作为当前路径的贡献。
ans=min(ans,x);:将当前路径的贡献 x 与当前最小偏心距 ans 比较,取较小值作为新的最小偏心距。这样做的目的是确保找到的路径具有最小的偏心距。
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<n;i++){
scanf("%d%d%d",&x,&y,&z);
add(x,y,z),add(y,x,z);
}
dis[1]=1,dfs(0,1);
dis[k]=0,dfs(0,k);
top=k;
for(int i=top,j=top,l=1,r=0;i;i=fa[i]){
while(dis[j]-dis[i]>m)j=fa[j];
x=max(dis[top]-dis[j],dis[i]);
ans=min(ans,x);
}
// 标记直径上的所有节点,并重新计算每个节点到直径的距离。
for(int i=top;i;i=fa[i])mark[i]=1;
// 重新计算每个节点到直径的距离。首先,将直径上的一个端点赋值给 k,然后从该点开始进行深度优先搜索,计算每个节点到直径的距离。
for(int i=top;i;i=fa[i]){
k=i,dis[k]=0;
dfs(fa[i],i);
}
// 计算每个节点的贡献,即每个节点到直径的距离的最大值,并更新 ans。
for(int i=1;i<=n;i++)
ans=max(ans,dis[i]);
// 输出ans.
printf("%d\n",ans);
return 0;
}
四、完整代码
#include<cstdio>
#include<algorithm>
#define M 500005
using namespace std;
int n,m,x,y,z,k,id,top,ans=2e9;
int dis[M],fa[M],head[M];
bool mark[M];
struct edge{
int to,w,nxt;
}E[M<<1];
void add(int u,int v,int w){
E[++id]=((edge){v,w,head[u]});
head[u]=id;
}
void dfs(int f,int x){
fa[x]=f;
if(dis[x]>dis[k])k=x;
for(int i=head[x];i;i=E[i].nxt){
int y=E[i].to;
if(y==f||mark[y])continue;
dis[y]=dis[x]+E[i].w;
dfs(x,y);
}
}
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<n;i++){
scanf("%d%d%d",&x,&y,&z);
add(x,y,z),add(y,x,z);
}
dis[1]=1,dfs(0,1);
dis[k]=0,dfs(0,k);
top=k;
for(int i=top,j=top,l=1,r=0;i;i=fa[i]){
while(dis[j]-dis[i]>m)j=fa[j];
x=max(dis[top]-dis[j],dis[i]);
ans=min(ans,x);
}
for(int i=top;i;i=fa[i])mark[i]=1;
for(int i=top;i;i=fa[i]){
k=i,dis[k]=0;
dfs(fa[i],i);
}
for(int i=1;i<=n;i++)
ans=max(ans,dis[i]);
printf("%d\n",ans);
return 0;
}