P1550 [USACO08OCT]打井Watering Hole

Posted 2021-01-16 lance1ot

P1550 [USACO08OCT]打井Watering Hole

---

题目比较简单，却提醒了图论建图的的重要性。

很多时候建图不只是要将给定的信息建立，还需要添加辅助点。

不过这题是真水

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
using std::sort;
const int maxn=500;
struct node
{
    int A,B;
    int val;
    bool operator <(const node &a)const{ return val<a.val; }
};
node line[maxn*maxn];
int f[maxn],tail;
int find(int x)
{
    return f[x]=(f[x]==x?x:find(f[x]));
}
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)   f[i]=i;
    int a;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a);
        line[++tail].A=0;
        line[tail].B=i;
        line[tail].val=a;
    }
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
        {
            scanf("%d",&a);
            line[++tail].A=i;
            line[tail].B=j;
            line[tail].val=a;
        }
    sort(line+1,line+1+tail);
    int get=0,pass=1,ans=0;
    while(get<n)
    {
        int F1=line[pass].A,F2=line[pass].B;
        F1=find(F1);F2=find(F2);
        if(F1!=F2)
        {
            get++;
            f[F1]=F2;
            ans+=line[pass].val;
        }
        pass++;
    }
    printf("%d",ans);
}