图的存储
图一般有两种存储方式:
邻接矩阵。开个二维数组,g[i][j] 表示点 ii 和点 jj 之间的边权。
邻接表。邻接表有两种常用写法,y总推荐第二种,代码更简洁,效率也更高,后面有代码模板:
(1) 二维vector:vector<vector<int>> edge,edge[i][j] 表示第 ii 个点的第 jj 条邻边。
(2) 数组模拟邻接表:为每个点开个单链表,分别存储该点的所有邻边。
树的直径相关
1.大臣的旅费
很久以前,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
vector代码实现
#include<iostream>
#include<vector>
using namespace std;
const int N = 1e6 + 10;
int n;
struct Edge{
int id,w;//id点的编号,w是边的长度
};
vector<Edge> h[N];
int dist[N];
void dfs(int u,int father,int distance){
dist[u] = distance;//当前节点的距离
//遍历当前节点的邻点
for(auto node : h[u])
if(node.id != father)
{
dfs(node.id,u,distance + node.w);
}
}
int main(){
int n;
scanf("%d",&n);
for(int i = 0;i < n - 1;i++)
{
int a,b,c;
scanf("%d%d%d",&a,&b,&c);//城市的点和距离
h[a].push_back({b,c});
h[b].push_back({a,c});
}
//从任意一点到某一点的距离,注意往回遍历的可能性
dfs(1,-1,0);//1为任意一点,-1为不存在的结点(从哪个节点下去),0是距离
int u = 1;
for(int i = 1;i <= n;i++)
if(dist[i] > dist[u])
u = i;
dfs(u,-1,0);
for(int i = 1;i <= n;i++)
if(dist[i] > dist[u])
u = i;
int w = dist[u];
cout << w*10 + w*(w+1ll)/2 << endl;
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;
int dist[N];
void add(int a, int b, int c)
{
e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx ++ ;
}
void dfs(int u, int father, int distance)
{
dist[u] = distance;
for (int i = h[u]; ~i; 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);
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;
}
真强