- 模版1:kruskal算法.并查集+贪心.
#include<iostream>
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll summ=0;
const int N=101000;
int fa[N];
int n,m;
int ans=0;
struct Node{
int x,y,t;
}way[N];
bool operator <(Node a,Node b) {
return a.t<b.t;
}//第一种排序方法.
int cmp(const Node &a,const Node &b)
{
return a.t<b.t;
}//第二种排序方法.
void init(int n)
{
for(int i=1;i<=n;i++)
fa[i]=i;
}
int find(int x)
{
if(fa[x]==x)
return x;
return fa[x]=find(fa[x]);
}//压缩路径.
void hd(int a,int b)
{
int x=find(a);
int y=find(b);
fa[x]=y;
}
int main()
{
cin>>n>>m;
init(n);
for(int i=1;i<=m;i++)
{
int a,b,t;
cin>>a>>b>>t;
way[i].x=a,way[i].y=b,way[i].t=t;
}
sort(way+1,way+m+1,cmp);
for(int i=1;i<=m;i++)
{
int fx=find(way[i].x);
int fy=find(way[i].y);
if(fa[fx]!=fa[fy])
{
//最小生成树就是找权值最小的依次连接成树.
fa[fx]=fy;
ans+=way[i].t;
n--;
}
if(n==1)
{
cout<<ans<<endl;
return 0;
}
}
cout<<-1<<endl;
return 0;
}
//最小生成树.
3.primes算法(普里姆算法)
#include<iostream>
#include<algorithm>
#include<bits/stdc++.h>
using namespace std;
typedef pair<int,int> PII;
const int N=2e3+10;
const int M=1e5+10;
int n,m;
int head[N],nxt[N],cnt;
int dis[N];
bool vis[N];
int v1[M],w1[M];
priority_queue< PII,vector<PII>,greater<PII> >heap;//定义一个小根堆。
void add(int a,int b,int c)
{
v1[cnt]=b;
w1[cnt]=c;
nxt[cnt]=head[a];
head[a]=cnt++;
}//链式前向星存图.
int prime()
{
memset(dis,0x3f,sizeof dis);
dis[1]=0;
heap.push({dis[1],1});
int num=n,res=0;
while(heap.size()&&num)
{
auto t=heap.top();
heap.pop();
int v=t.second,d=t.first;
if(vis[v])
continue;
vis[v]=true;
res+=d;
num--;
for(int i=head[v];i!=-1;i=nxt[i])
{
int j=v1[i];
if(!vis[j]&&dis[j]>w1[i])
dis[j]=w1[i];
heap.push({dis[j],j});
}
}
return num?-1:res;//所有点都用过以后返回.
}
int main()
{
cin>>n>>m;
memset(head,-1,sizeof head);
for(int i=0;i<m;i++)
{
int a,b,c;
cin>>a>>b>>c;
add(a,b,c);
add(b,a,c);
}
int t=prime();
t==-1?cout<<"NO":cout<<t;
return 0;
}
就是图的存储+贪心算法.//这里是无向图.