每日小结2.1

最新推荐文章于 2024-07-23 22:06:37 发布

熵逝

最新推荐文章于 2024-07-23 22:06:37 发布

阅读量35

点赞数

文章标签：图论算法

本文链接：https://blog.csdn.net/weixin_74567525/article/details/128841809

版权

今天浅浅水一下

P3366 【模板】最小生成树

题目描述

如题，给出一个无向图，求出最小生成树，如果该图不连通，则输出 orz。

输入格式

第一行包含两个整数 N,MN,M，表示该图共有 NN 个结点和 MM 条无向边。

接下来 MM 行每行包含三个整数 X_i,Y_i,Z_iXi,Yi,Zi，表示有一条长度为 Z_iZi 的无向边连接结点 X_i,Y_iXi,Yi。

输出格式

如果该图连通，则输出一个整数表示最小生成树的各边的长度之和。如果该图不连通则输出 orz。

输入输出样例

输入 #1复制

输出 #1复制

说明/提示

数据规模：

对于 20\%20% 的数据，N\le 5N≤5，M\le 20M≤20。

对于 40\%40% 的数据，N\le 50N≤50，M\le 2500M≤2500。

对于 70\%70% 的数据，N\le 500N≤500，M\le 10^4M≤104。

对于 100\%100% 的数据：1\le N\le 50001≤N≤5000，1\le M\le 2\times 10^51≤M≤2×105，1\le Z_i \le 10^41≤Zi≤104。

样例解释：

所以最小生成树的总边权为 2+2+3=72+2+3=7。

思路：用Kruskal就能做出来，明天尝试Prim

#include<stdio.h>
struct edge
{
    int u;
    int v;
    int w;
} e[200005];
int fa[5005],sum=0,count1=0;
int find(int y)
{
    if(fa[y]==y)return y;
    else {
       fa[y]=find(fa[y]);
        return fa[y];   }
}
int merge(int x,int y)
{
    int t1,t2;
    t1=find(x);
    t2=find(y);
    if(t1!=t2){
            fa[t2]=t1;
            return 1;}
    else
    return 0;
}
void kuai(int left,int right)
{
    int i,j;
    struct edge t;
    if(left>=right)return ;
    i=left;
    j=right;
    while(i!=j){
        while(e[j].w>=e[left].w&&i<j)j--;
         while(e[i].w<=e[left].w&&i<j)i++;
         if(i<j){
            t=e[i];
            e[i]=e[j];
            e[j]=t;
         }}
         t=e[left];
         e[left]=e[i];
         e[i]=t;
         kuai(left,i-1);
         kuai(i+1,right);
    }
int main()
{
    int n,m;
      scanf("%d %d",&n,&m);
      for(int i=1;i<=m;i++){
        scanf("%d %d %d",&e[i].u,&e[i].v,&e[i].w);
      }
      kuai(1,m);
      for(int i=1;i<=n;i++)fa[i]=i;
      for(int i=1;i<=m;i++){
        if(merge(e[i].u,e[i].v)){sum+=e[i].w;count1++;
      }

        if(count1==n-1)break;
}
if(count1==n-1)
    printf("%d",sum);
    else printf("orz");
}