标签：Kattis int Freckles ++ static double new Problem dis

Kattis Problem - Freckles

原题地址

题目类型：最小生成树

题意：

在纸上有 n 个点（坐标形式给出），可以用钢笔划线将点连接起来。现在需要用钢笔连线使得所有的点都连通，并使用墨水最少，画线使用墨水的值等于线的长度。

分析：

很明显是要求其最小生成树。在每两个点之间建边，长度为其之间的距离，然后使用 Prim 算法求其最小生成树。使用 Kruskal 算法会超时。

代码

static double[] x;
static double[] y;
static double[][] G;

public static void solve() throws IOException {
    int m = nextInt();
    
    init(m);
    
    for (int i = 0; i < m; i++) {
        x[i] = nextDouble();
        y[i] = nextDouble();
    }
    
    for (int i = 0; i < m; i++) {
        for (int j = 0; j < m; j++) {
            G[i][j] = getDis(x[i], y[i], x[j], y[j]);
        }
    }
    
    double ans = prim(m);
    
    pw.println(String.format("%.2f", ans));
}   

public static double prim(int m) {
    double[] dis = new double[m];
    boolean[] vis = new boolean[m];
    Arrays.fill(dis, INF);
    
    double re = 0;
    for (int i = 0; i < m; i++) {
        int u = -1;
        for (int j = 0; j < m; j++) 
            if (!vis[j] && (u == -1 || dis[u] > dis[j])) u = j;
        
        vis[u] = true;
        if (i != 0) re += dis[u];
        
        for (int j = 0; j < m; j++) dis[j] = Math.min(dis[j], G[u][j]);
    }
    
    return re;
}

public static void init(int m) {
    x = new double[m];
    y = new double[m];
    G = new double[m][m];
    for (int i = 0; i < m; i++)
        for (int j = 0; j < m; j++)
            G[i][j] = (i == j ? 0 : INF);
}

public static double getDis(double x1, double y1, double x2, double y2) {
    return Math.sqrt((x1 - x2) * (x1 - x2) + (y1 - y2) * (y1 - y2));
}

标签：Kattis,int,Freckles,++,static,double,new,Problem,dis
来源： https://blog.csdn.net/w_weirdo/article/details/119191930

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。