洛谷P1194 买礼物

题目描述

又到了一年一度的明明生日了，明明想要买B样东西，巧的是，这B样东西价格都是A元。

但是，商店老板说最近有促销活动，也就是：

如果你买了第I样东西，再买第J样，那么就可以只花K[I,J]元，更巧的是，K[I,J]竟然等于K[J,I]。

现在明明想知道，他最少要花多少钱。

输入输出格式

输入格式：

第一行两个整数，A,B。

接下来B行，每行B个数，第I行第J个为K[I,J]。

我们保证K[I,J]=K[J,I]并且K[I,I]=0。

特别的，如果K[I,J]=0，那么表示这两样东西之间不会导致优惠。

输出格式：

仅一行一个整数，为最小要花的钱数。

输入输出样例

输入样例#1：

【样例输入1】
1 1
0
【样例输入2】
3 3
0 2 4
2 0 2
4 2 0

输出样例#1：

【样例输出1】
1
【样例输出2】
7

说明

样例解释2

先买第2样东西，花费3元，接下来因为优惠，买1，3样都只要2元，共7元。

（同时满足多个“优惠”的时候，聪明的明明当然不会选择用4元买剩下那件，而选择用2元。）

数据规模

对于30%的数据,1<=B<=10。

对于100%的数据,1<=B<=500，0<=A,K[I,J]<=1000。

最小生成树，只是有些要注意的地方。。。

附代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#define MAXN 510
using namespace std;
int n,m,k=1,fa[MAXN];
struct node{
       int u,v,w;
}a[MAXN*MAXN];
inline int read(){
       int date=0,w=1;char c=0;
       while(c<'0'||c>'9'){if(c=='-')w=-1;c=getchar();}
       while(c>='0'&&c<='9'){date=date*10+c-'0';c=getchar();}
       return date*w;
}
bool cmp(const node &x,const node &y){
     return x.w<y.w;
}
int find(int x){return fa[x]==x?x:fa[x]=find(fa[x]);}
void uniun(int x,int y){x=find(x);y=find(y);if(x!=y)fa[y]=x;}
void kruskal(){
     int s=1,ans=0;
     for(int i=1;i<=k;i++)
     if(find(a[i].u)!=find(a[i].v)){
                                    uniun(a[i].u,a[i].v);
                                    ans+=a[i].w;
                                    s++;
                                    if(s==m)break;
                                    }
     printf("%d\n",ans+n);
}
int main(){
    int x;
    n=read();m=read();
    for(int i=0;i<=m;i++)fa[i]=i;
    for(int i=1;i<=m;i++)
    for(int j=1;j<=m;j++){
            x=read();
            if(j>=i||x==0)continue;
            a[k].u=i;a[k].v=j;a[k].w=x;
            k++;
            }
    for(int i=1;i<=m;i++){
            a[k].u=0;a[k].v=i;a[k].w=n;
            k++;
            }
    k--;
    sort(a+1,a+k+1,cmp);
    kruskal();
    return 0;
}