DFS(树的直径)-第四届蓝桥杯省赛C++A组-大臣的旅费
题目:
很久以前,T王国空前繁荣。
为了更好地管理国家,王国修建了大量的快速路,用于连接首都和王国内的各大城市。
为节省经费,T国的大臣们经过思考,制定了一套优秀的修建方案,使得任何一个大城市都能从首都直接或者通过其他大城市间接到达。
同时,如果不重复经过大城市,从首都到达每个大城市的方案都是唯一的。
J是T国重要大臣,他巡查于各大城市之间,体察民情。
所以,从一个城市马不停蹄地到另一个城市成了J最常做的事情。
他有一个钱袋,用于存放往来城市间的路费。
聪明的J发现,如果不在某个城市停下来修整,在连续行进过程中,他所花的路费与他已走过的距离有关,在走第x千米到第x+1千米这一千米中(x是整数),他花费的路费是x+10这么多。也就是说走1千米花费11,走2千米要花费23。
J大臣想知道:他从某一个城市出发,中间不休息,到达另一个城市,所有可能花费的路费中最多是多少呢?
输入格式
输入的第一行包含一个整数 n,表示包括首都在内的T王国的城市数。
城市从 1 开始依次编号,1 号城市为首都。
接下来 n−1 行,描述T国的高速路(T国的高速路一定是 n−1 条)。
每行三个整数 Pi,Qi,Di,表示城市 Pi 和城市 Qi 之间有一条双向高速路,长度为 Di 千米。
输出格式
输出一个整数,表示大臣J最多花费的路费是多少。
数据范围
1≤n≤105,
1≤Pi,Qi≤n,
1≤Di≤1000
输入样例:
5
1 2 2
1 3 1
2 4 5
2 5 4
输出样例:
135
题意:
n 个 节 点 , 任 何 一 个 节 点 P i 都 可 以 通 过 1 号 节 点 到 达 节 点 Q i , 长 度 为 D i , 花 费 为 ∑ j = 1 D i ( 10 + j ) = 10 D i + ( D i + 1 ) D i 2 。 n个节点,任何一个节点P_i都可以通过1号节点到达节点Q_i,长度为D_i,花费为\sum_{j=1}^{Di}(10+j)=10D_i+\frac{(D_i+1)D_i}{2}。 n个节点,任何一个节点Pi都可以通过1号节点到达节点Qi,长度为Di,花费为∑j=1Di(10+j)=10Di+2(Di+1)Di。
问 所 有 的 通 路 当 中 , 花 费 最 大 的 时 多 少 , 也 就 是 最 长 的 一 条 路 是 多 长 。 问所有的通路当中,花费最大的时多少,也就是最长的一条路是多长。 问所有的通路当中,花费最大的时多少,也就是最长的一条路是多长。
整 个 图 可 以 看 作 一 个 以 1 号 节 点 为 根 的 树 , 要 求 树 的 直 径 。 整个图可以看作一个以1号节点为根的树,要求树的直径。 整个图可以看作一个以1号节点为根的树,要求树的直径。
任 选 一 个 点 u , d f s 到 u 的 最 远 的 点 v , 再 从 点 v 开 始 d f s 到 v 的 最 远 距 离 即 直 径 。 就 是 找 树 上 距 离 最 远 的 两 个 边 界 点 。 任选一个点u,dfs到u的最远的点v,再从点v开始dfs到v的最远距离即直径。就是找树上距离最远的两个边界点。 任选一个点u,dfs到u的最远的点v,再从点v开始dfs到v的最远距离即直径。就是找树上距离最远的两个边界点。
建 图 : 建图: 建图:
① 、 v e c t o r 。 ①、vector。 ①、vector。
② 、 数 组 模 拟 邻 接 表 , 即 n 个 单 链 表 ②、数组模拟邻接表,即n个单链表 ②、数组模拟邻接表,即n个单链表——《单链表的数组实现》。
代码:
①、vector:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#include<vector>
#define P pair<int,int>
#define x first
#define y second
#define ll long long
#define inf 0x7fffffff
using namespace std;
const int N=1e5+10;
int n;
struct Edge
{
int id,w;
};
vector<Edge>V[N];
int dis[N];
void dfs(int u,int father,int distance)
{
dis[u]=distance;
for(auto e :V[u])
if(e.id!=father)
dfs(e.id,u,distance+e.w);
}
int main()
{
int n;
cin>>n;
for(int i=0;i<n-1;i++)
{
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
V[a].push_back({b,c});
V[b].push_back({a,c});
}
dfs(1,-1,0);
///找到到点1距离最大的点
int max_pos=1;
for(int i=1;i<=n;i++)
if(dis[i]>dis[max_pos])
max_pos=i;
///找到到点max_pos距离最大的点
dfs(max_pos,-1,0);
for(int i=1;i<=n;i++)
if(dis[i]>dis[max_pos])
max_pos=i;
int s=dis[max_pos];
printf("%lld\n",s*10+s*(s+1ll)/2);
return 0;
}
②、数组模拟:
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 100010, M = 200010;
int n;
int h[N], e[M], w[M], ne[M], idx; ///h邻接表头节点,e终点,w权值,ne下一个相连终点,idx当前插入节点的编号
int dist[N];
void add(int a, int b, int c)
{
e[idx] = b;
w[idx] = c;
ne[idx] = h[a];
h[a] = idx;
idx ++;
}
void dfs(int u, int father, int distance)
{
dist[u] = distance;
for (int i = h[u]; i!=-1 ; i = ne[i])
{
int j = e[i];
if (j != father)
dfs(j, u, distance + w[i]);
}
}
int main()
{
scanf("%d", &n);
memset(h, -1, sizeof h); ///所有头节点置-1
for (int i = 0; i < n - 1; i ++ )
{
int a, b, c;
scanf("%d%d%d", &a, &b, &c);
add(a, b, c), add(b, a, c);
}
dfs(1, -1, 0);
int u = 1;
for (int i = 2; i <= n; i ++ )
if (dist[u] < dist[i])
u = i;
dfs(u, -1, 0);
for (int i = 1; i <= n; i ++ )
if (dist[u] < dist[i])
u = i;
printf("%lld\n", dist[u] * 10 + (dist[u] + 1ll) * dist[u] / 2);
return 0;
}