[题解]洛谷P1194 买礼物

原题:传送门

如果两个物品之间存在优惠关系就连一条边,注意值为0不要连边

然后再加一个节点0,往点1~B连权为A的边

然后求一遍最小生成树即可

代码:

#include
#include
#include
#include
using namespace std;

int a,b,family[510];

struct edge{
    int u,v,w;
}e[134750];

int cmp(edge aa,edge bb){
    return aa.w<bb.w;
}

int m=0;
void build(int u,int v,int w){
    m++;
    e[m].u=u;e[m].v=v;e[m].w=w;
}

int find(int x){
    if(family[x]==x)return x;
    return family[x]=find(family[x]);
}

int ans=0,tot=0;
int main(){
    scanf("%d%d",&a,&b);
    for(int i=1;i<=b;i++)
        for(int j=1;j<=b;j++){
            int k;
            scanf("%d",&k);
            if(k!=0&&i<j)build(i,j,k);
        }
    for(int i=1;i<=b;i++)
        build(0,i,a);
    for(int i=0;i<=b;i++)
        family[i]=i;
    sort(e+1,e+m+1,cmp);
    for(int i=1;i<=m;i++){
        int f1=find(e[i].u),f2=find(e[i].v);
        if(f1!=f2){
            family[f1]=f2;
            ans+=e[i].w;
            tot++;
            if(tot==b)
                break;
        }
    }
    printf("%d",ans);
    return 0;
}

 

转载于:https://www.cnblogs.com/sjrb/p/10351838.html

你可能感兴趣的:([题解]洛谷P1194 买礼物)