最小生成树练习3

​​畅通道路​​放图祭命运共同体

祭祀完毕！

下面来看这道题。

这输入方式……不明摆着让我们用kruscal吗？

只是这道题与之前两道不一样的地方是，有部分道路是事先修好的。

那么我们就稍微变通一下即可。

我们回想一下，我们怎么记录生成树的？并查集！

所以我们只需要在读入时，把现成的边在并查集中连在一起即可，剩下的这个边就可以直接扔掉，不需要放在kruscal的排序序列中。剩下的按kruscal做一遍即可。

每加一条边，把边权加在最终结果即可。

代码如下：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
struct edge{
  int u,v,w;
};

int cmp(edge a,edge b){
  return a.w<b.w;
}
int n,m;
edge e[10005];
int p[105];
int ans;
int find(int cur){
  return cur==p[cur]?cur:p[cur]=find(p[cur]);
}
void work(){
  ans=0;
  sort(e,e+m,cmp);
  for(int i=0;i<m;i++){
    int x=find(e[i].u),y=find(e[i].v);
    if(x!=y){
      p[x]=y;
      ans+=e[i].w; 
    }
  }
}

int main(){
  while(1){
    scanf("%d",&n);
    if(n==0){
      break;
    }
    for(int i=1;i<=n;i++){
      p[i]=i;
    }
    m=0;
    for(int i=0;i<n*(n-1)/2;i++){
      int u,v,w,vis;
      scanf("%d%d%d%d",&u,&v,&w,&vis);
      if(vis){
        p[find(u)]=find(v);
      }else{
        e[m].u=u;
        e[m].v=v;
        e[m].w=w;
        m++;
      }
    }
    work();
    printf("%d\n",ans);
  }
  return 0;
}

PS：这两天在忙着填志愿，可能做题比较少，以后会加大题量的。