最小生成树(prim算法)

1.朴素prim算法(重在理解prim算法思想)

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const int N = 300 + 9;
const ll inf = 4e18;

ll a[N][N], d[N], n, m;

bitset<N> intree;

void solve()
{
   cin >> n >> m ;

   memset(a,0x3f,sizeof a);
   memset(d,0x3f,sizeof d);

   for(int i = 1;i <= n;i++)a[i][i] = 0;

   for(int i = 1;i <= m;i++)
   {
       ll u , v , w;cin >> u >> v >> w;
       // 这个是无向图啊
       // 所以这里两边都要更新一下子
       a[u][v] = min(a[u][v] , w);
       a[v][u] = min(a[v][u] , w);
   }

   /*
   d[1] = 0; // 1 放入了intree集合(intree 和 outtree 都不是实际创建的容器集合)
   intree[1] = true;
   */

   ll ans = 0;

   for(int i = 1;i <= n;i++)
   {
       // 为了找出与 intree最新点 距离最近的下一点
       int u = 1;
       for(int j = 1;j <= n;j++)
           if(intree[u] || (!intree[j] && d[j] < d[u]))u = j; 

       if(u ^ 1)ans += d[u];
       intree[u] = true;
       d[u] = 0;

       for(int j = 1;j <= n;j++)
       {
           if(intree[j])continue;

           d[j] = min(a[u][j],d[j]);
       }
   }

   cout << ans << '\n';

}

int main()
{
   ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);

   int _ = 1;
   while(_--)
       solve();

   return 0;
}

2.用优先队列优化prim算法

#include<bits/stdc++.h>
using namespace std;
using ll=long long;
const int N=1e5+9;
ll d[N];
bitset<N>intree;
struct Edge
{
  ll x,w;
  bool operator <(const Edge &e)const
  {
    return w==e.w?x<e.x:w>e.w;
  }
};
vector<Edge>g[N];
int main()
{
    memset(d,0x3f,sizeof d);
    ll n,m;cin>>n>>m;
    for(int i=1;i<=m;i++)
    {
        ll u,v,w;cin>>u>>v>>w;
        g[u].push_back({v,w});
        g[v].push_back({u,w});
    }
    priority_queue<Edge>pq;
    ll ans=0;
    pq.push({1,0});
    d[1]=0;
    while(pq.size())
    {
        auto[x,w]=pq.top();pq.pop();
        if(intree[x])continue;
        intree[x]=true;
        ans+=w;
        d[x]=0;
        for(auto &[y,w]:g[x])
        {
          if(!intree[y]&&w<d[y])
          {
            d[y]=w;
            pq.push({y,w});
          }
        }
        
    }
    for(int i=1;i<=n;i++)
    {
      if(!intree[i])ans=-1;
    }
    cout<<ans<<'\n';

    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值