蓝桥杯--户户通电

户户通电

题目
问题描述

2015年，全中国实现了户户通电。作为一名电力建设者，小明正在帮助一带一路上的国家通电。
　　这一次，小明要帮助 n 个村庄通电，其中 1 号村庄正好可以建立一个发电站，所发的电足够所有村庄使用。
　　现在，这 n 个村庄之间都没有电线相连，小明主要要做的是架设电线连接这些村庄，使得所有村庄都直接或间接的与发电站相通。
　　小明测量了所有村庄的位置（坐标）和高度，如果要连接两个村庄，小明需要花费两个村庄之间的坐标距离加上高度差的平方，形式化描述为坐标为 (x_1, y_1) 高度为 h_1 的村庄与坐标为 (x_2, y_2) 高度为 h_2 的村庄之间连接的费用为
　　sqrt((x_1-x_2)(x_1-x_2)+(y_1-y_2)(y_1-y_2))+(h_1-h_2)*(h_1-h_2)。
　　在上式中 sqrt 表示取括号内的平方根。请注意括号的位置，高度的计算方式与横纵坐标的计算方式不同。
　　由于经费有限，请帮助小明计算他至少要花费多少费用才能使这 n 个村庄都通电。

输入格式

输入的第一行包含一个整数 n ，表示村庄的数量。
　　接下来 n 行，每个三个整数 x, y, h，分别表示一个村庄的横、纵坐标和高度，其中第一个村庄可以建立发电站。

输出格式

输出一行，包含一个实数，四舍五入保留 2 位小数，表示答案。

样例输入

4
1 1 3
9 9 7
8 8 6
4 5 4

样例输出

17.41

评测用例规模与约定

对于 30% 的评测用例，1 <= n <= 10；
　　对于 60% 的评测用例，1 <= n <= 100；
　　对于所有评测用例，1 <= n <= 1000，0 <= x, y, h <= 10000。

import java.beans.Visibility;
import java.util.Scanner;


public class Main
{

    static int art[] = new int[1000];
    static int maze[][] = new int[1000][3];
    static int n ;
    static int brt[][] =new int[1005][2];
    static double min = 1000000;
    
    public static void print()
    {
        for(int i=0;i<n-1;i++)
        {
            System.out.println(brt[i][0]+"  "+brt[i][1]);
        }
        System.out.println();
    }
    
    public static double money()//计算金额
    {
        double sum = 0;
        for(int i=0;i<n-1;i++)
        {
            int a = brt[i][0];
            int b = brt[i][1];//两个村
            sum += Math.sqrt((maze[a][0]-maze[b][0])*(maze[a][0]-maze[b][0])+(maze[a][1]-maze[b][1])*(maze[a][1]-maze[b][1]))+(maze[a][2]-maze[b][2])*(maze[a][2]-maze[b][2]);
        }
        return sum;
    }
    
    public static void fun(int a,int max,boolean vis[])//当前村，当前总共多少边
    {
        //System.out.println(brt[max][0]+brt[max][1]);
        if(max == 3)
        {
            //print();
            if(money()<min)
            {
                min = money();
            }
            //System.out.println(money());
            return;
        }
        for(int i=0;i<n;i++)
        {
            if((a!=i) && !vis[i])//连接的这个村，和现在这个村不连同
            {
                brt[max][0] = a;
                brt[max][1] = i;
                vis[i] = true;//表示已经相连
                fun(a, max+1, vis);//继续以这个村子往下一个发展
                fun(i, max+1, vis);//以下一个作为发展
                vis[i] = false;//表示已经相连
            }
        }
        
    }
    
    public static void main(String[] args)
    {
        // TODO Auto-generated method stub
        boolean vis[] = new boolean[1005];
        Scanner scanner =new Scanner(System.in);
        n = scanner.nextInt();
        for(int i=0;i<n;i++)
        {
            art[i] = i;
            maze[i][0] = scanner.nextInt();
            maze[i][1] = scanner.nextInt();
            maze[i][2] = scanner.nextInt();
        }
        
        vis[0] = true;
        fun(0, 0,vis);
        System.out.println(String.format("%.2f", min));//输出小数点后两位
    }

}