APIO2008 免费道路

传送门


%%%\(\sf{ZYD\;\;TQL}\)


感觉这个题的题意有点难懂,简单概括一下就是:
\(n\)个村庄,\(m\)条道路,道路分为水泥路和鹅卵石路,选择其中\(n-1\)条路使得所有村庄联通,并且选择的路中恰好有\(k\)条鹅卵石路。

我们可以先对所有的水泥路跑一遍\(Kruskal\),把它们全加入生成树,然后再对鹅卵石路跑一遍,此时加进生成树里的鹅卵石路是必须要加的,因为没有它们,是无法让图联通的

之后我们将图复原,现将必须要加的鹅卵石路加进去,再添加鹅卵石路,直至总数达到\(k\),最后再加水泥路。

无解的情况有:

  • 找不到生成树
  • 必须添加的鹅卵石路大于\(k\)
  • 鹅卵石路的数量小于\(k\)

代码:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int fa[20010];
struct zzz{
    int f,t; bool fl;
}e1[100010],e2[100010],imp[20010],ans[20010];
int find(int x){
    return fa[x]==x? x:fa[x]=find(fa[x]);
}
int cnt,tot1,tot2,tot3;
int read(){
    int k=0; char c=getchar();
    for(;c<'0'||c>'9';) c=getchar();
    for(;c>='0'&&c<='9';c=getchar())
      k=k*10+c-48;
    return k;
}
int main(){
    int n=read(),m=read(),k=read();
    for(int i=1;i<=n;i++) fa[i]=i;
    for(int i=1;i<=m;i++){
        int x=read(),y=read();
        if(read()) e2[++tot2].f=x,e2[tot2].t=y;
        else e1[++tot1].f=x,e1[tot1].t=y;
    }
    if(tot1<k){
        printf("no solution"); return 0;
    }
    for(int i=1;i<=tot2;i++){
        int x=find(e2[i].f),y=find(e2[i].t);
        if(fa[x]!=fa[y]) fa[x]=fa[y],cnt++;
    }
    for(int i=1;i<=tot1;i++){
        int x=find(e1[i].f),y=find(e1[i].t);
        if(x!=y){
            fa[x]=fa[y],cnt++;
            imp[++tot3]=e1[i];
        }
    }
    if(cnt<n-1||tot3>k){
        printf("no solution"); return 0;
    }
    
    cnt=0;
    for(int i=1;i<=n;i++) fa[i]=i;
    for(int i=1;i<=tot3;i++){
        int x=find(imp[i].f),y=find(imp[i].t);
        if(fa[x]!=fa[y]){
            fa[x]=fa[y]; cnt++;
            ans[cnt]=imp[i]; ans[cnt].fl=0;
        }
    }
    for(int i=1;i<=tot1;i++){
        int x=find(e1[i].f),y=find(e1[i].t);
        if(fa[x]!=fa[y]){
            fa[x]=fa[y]; cnt++;
            ans[cnt]=e1[i]; ans[cnt].fl=0;
        }
        if(cnt==k) break;
    }
    if(cnt<k){
        printf("no solution"); return 0;
    }
    for(int i=1;i<=tot2;i++){
        int x=find(e2[i].f),y=find(e2[i].t);
        if(fa[x]!=fa[y]){
            fa[x]=fa[y]; cnt++;
            ans[cnt]=e2[i]; ans[cnt].fl=1;
        }
        if(cnt==n-1) break;
    }
    for(int i=1;i<=n-1;i++)
      printf("%d %d %d\n",ans[i].f,ans[i].t,ans[i].fl);
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/wxl-Ezio/p/9508062.html