大臣的旅费

最新推荐文章于 2024-06-25 21:10:30 发布

赚钱给孩子买茅台喝

最新推荐文章于 2024-06-25 21:10:30 发布

阅读量94

点赞数 3

分类专栏：蓝桥杯文章标签：深度优先算法

本文链接：https://blog.csdn.net/2302_76649176/article/details/136822997

版权

蓝桥杯专栏收录该内容

16 篇文章 0 订阅

订阅专栏

该文章介绍了使用Java编程解决图论问题，包括添加边、深度优先搜索(DFS)寻找最远点、广度优先搜索(BFS)确定直径，并利用这些信息计算等差数列的前n项和。

摘要由CSDN通过智能技术生成

import java.util.Arrays;
import java.util.Scanner;

public class Main{
    static int n,idx;
    static int N = 100010;
    static int M = 200010;
    static int[] h = new int[N];
    static int[] ne = new int[M];
    static int[] e = new int[M];
    static int[] w = new int[M];
    static int[] dist = new int[M];

    public static void add(int a,int b,int c){
        e[idx] = b;
        w[idx] = c;
        ne[idx] = h[a];
        h[a] = idx++;
    }

    public static 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]);
            }
        }
    }

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        n = sc.nextInt();

        Arrays.fill(h,-1);

        for(int i = 1;i <= n - 1;i++){
            int a = sc.nextInt();
            int b = sc.nextInt();
            int c = sc.nextInt();
            add(a,b,c);
            add(b,a,c);
        }
        //随便找个数搜到最远
        dfs(1,-1,0);

        //找到最远那个点
        int max_idx = 1;
        for(int i = 2;i <= n;i++){
            if (dist[max_idx] < dist[i])
                max_idx = i;
        }
        
        //从最远那个点搜回去,就是直径
        dfs(max_idx,-1,0);
        int max = dist[1];
        for(int i = 2;i <= n;i++){
            if (max < dist[i])
                max = dist[i];
        }

        //等差数列前n项和
        System.out.println(10 * max + (long) max * (1 + max) / 2);
    }
}