kruskal模版
void kruskal(){
sort(t+1,t+1+m,cmp);
long long tot=0,ans=0;
for(int i=1;i<=m;i++){
int xx=Find(t[i].x),yy=Find(t[i].y);
if(xx==yy){
continue;
}
f[xx]=yy;
ans+=t[i].z;
tot++;
if(tot==n-1){
cout<<ans;
return ;
}
}
cout<<"orz";
return ;
}
prim模版
void prim(){
memset(vis,0,sizeof(vis));
memset(dis,0x3f,sizeof(dis));
dis[1]=0;
for(int i=1;i<=n;i++){
int t=-1;
for(int j=1;j<=n;j++){
if(!vis[j]&&(t==-1||dis[t]>dis[j])){
t=j;
}
}
if(dis[t]==0x3f3f3f3f){
cout<<"orz";
return ;
}
vis[t]=1;
ans+=dis[t];
for(int j=1;j<=n;j++){
if(dis[j]>a[t][j]&&!vis[j]){
dis[j]=a[t][j];
}
}
}
cout<<ans;
return ;
}
最小生成树
kruskal:
#include<bits/stdc++.h>
using namespace std;
int n,m,a,b,c;
int f[200005];
struct stu{
int x,y,z;
}t[200005];
bool cmp(stu a,stu b){
if(a.z==b.z){
if(a.x==b.x){
return a.y<b.y;
}
return a.x<b.x;
}
return a.z<b.z;
}
void init(){
for(int i=1;i<=n;i++){
f[i]=i;
}
}
int Find(int x){
if(f[x]==x){
return x;
}
int root=Find(f[x]);
return f[x]=root;
}
void Union(int x,int y){
int fx=Find(x);
int fy=Find(y);
if(fx==fy){
return ;
}
f[fx]=fy;
}
void kruskal(){
sort(t+1,t+1+m,cmp);
long long tot=0,ans=0;
for(int i=1;i<=m;i++){
int xx=Find(t[i].x),yy=Find(t[i].y);
if(xx==yy){
continue;
}
f[xx]=yy;
ans+=t[i].z;
tot++;
if(tot==n-1){
cout<<ans;
return ;
}
}
cout<<"orz";
return ;
}
int main(){
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
cin>>n>>m;
init();
for(int i=1;i<=m;i++){
cin>>a>>b>>c;
t[i].x=a;
t[i].y=b;
t[i].z=c;
}
kruskal();
return 0;
}
prim:
#include<bits/stdc++.h>
using namespace std;
int n,m;
int x,y,z;
int a[5005][5005];
int vis[5005],dis[5005];
int cnt=1;
long long ans;
void prim(){
memset(vis,0,sizeof(vis));
memset(dis,0x3f,sizeof(dis));
dis[1]=0;
for(int i=1;i<=n;i++){
int t=-1;
for(int j=1;j<=n;j++){
if(!vis[j]&&(t==-1||dis[t]>dis[j])){
t=j;
}
}
if(dis[t]==0x3f3f3f3f){
cout<<"orz";
return ;
}
vis[t]=1;
ans+=dis[t];
for(int j=1;j<=n;j++){
if(dis[j]>a[t][j]&&!vis[j]){
dis[j]=a[t][j];
}
}
}
cout<<ans;
return ;
}
int main(){
cin>>n>>m;
memset(a,0x3f,sizeof(a));
for(int i=1;i<=m;i++){
cin>>x>>y>>z;
a[x][y]=a[y][x]=min(a[x][y],z);
}
prim();
return 0;
}
Build?
#include<bits/stdc++.h>
using namespace std;
int n,X,Y,m;
struct stu{
int x,y,z;
}a[100005],t[200005];
int f[100005];
long long ans;
void init(){
for(int i=1;i<=n;i++){
f[i]=i;
}
}
bool cmp(stu a,stu b){
return a.x<b.x;
}
bool cmt(stu a,stu b){
return a.y<b.y;
}
bool cmy(stu a,stu b){
return a.z<b.z;
}
int Find(int x){
if(f[x]==x){
return x;
}
int root=Find(f[x]);
return f[x]=root;
}
void Union(int x,int y){
int fx=Find(x);
int fy=Find(y);
if(fx==fy){
return ;
}
f[fx]=fy;
}
void kruskal(){
sort(t+1,t+1+m,cmy);
long long tot=0,ans=0;
for(int i=1;i<=m;i++){
int xx=Find(t[i].x),yy=Find(t[i].y);
if(xx==yy){
continue;
}
f[xx]=yy;
ans+=t[i].z;
tot++;
if(tot==n-1){
cout<<ans;
return ;
}
}
}
int main(){
cin>>n;
init();
for(int i=1;i<=n;i++){
cin>>X>>Y;
a[i].x=X,a[i].y=Y;
a[i].z=i;
}
sort(a+1,a+1+n,cmp);
for(int i=1;i<n;i++){
m++;
t[m].x=a[i].z,t[m].y=a[i+1].z;
t[m].z=min(abs(a[i].x-a[i+1].x),abs(a[i].y-a[i+1].y));
}
sort(a+1,a+1+n,cmt);
for(int i=1;i<n;i++){
m++;
t[m].x=a[i].z,t[m].y=a[i+1].z;
t[m].z=min(abs(a[i].x-a[i+1].x),abs(a[i].y-a[i+1].y));
}
kruskal();
return 0;
}
建造空间站
#include<bits/stdc++.h>
using namespace std;
int n,m;
struct node{
double x,y,z,r;
}a[105];
struct stu{
int x,y;
double z;
}t[10005];
int f[105];
double ans;
void init(){
for(int i=1;i<=n;i++){
f[i]=i;
}
}
bool cmp(stu a,stu b){
return a.z<b.z;
}
int Find(int x){
if(f[x]==x){
return x;
}
int root=Find(f[x]);
return f[x]=root;
}
void Union(int x,int y){
int fx=Find(x);
int fy=Find(y);
if(fx==fy){
return ;
}
f[fx]=fy;
}
void kruskal(){
sort(t+1,t+1+m,cmp);
long long tot=0;
for(int i=1;i<=m;i++){
int xx=Find(t[i].x),yy=Find(t[i].y);
if(xx==yy){
continue;
}
f[xx]=yy;
ans+=t[i].z;
tot++;
if(tot==n-1){
printf("%.3lf\n",ans);
return ;
}
}
}
int main(){
while(cin>>n){
if(n==0){
break;
}
init();
m=0,ans=0;
for(int i=1;i<=n;i++){
double X,Y,Z,R;
cin>>X>>Y>>Z>>R;
a[i].x=X,a[i].y=Y,a[i].z=Z,a[i].r=R;
}
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){
if(i==j){
continue;
}
m++;
t[m].x=i,t[m].y=j;
t[m].z=max(sqrt(pow(a[i].x-a[j].x,2)+pow(a[i].y-a[j].y,2)+pow(a[i].z-a[j].z,2))-a[i].r-a[j].r,0.0);
}
}
kruskal();
}
return 0;
}