P1194 买礼物 （最小生成树）

题目描述
又到了一年一度的明明生日了，明明想要买BB样东西，巧的是，这BB样东西价格都是AA元。

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

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

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

输入格式
第一行两个整数，A,BA,B。

接下来BB行，每行BB个数，第II行第JJ个为K_{I,J}K
I,J
​
。

我们保证K_{I,J}=K_{J,I}K
I,J
​
=K
J,I
​
并且K_{I,I}=0K
I,I
​
=0。

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

输出格式
一个整数，为最小要花的钱数。

输入输出样例
输入 #1复制
1 1
0

输出 #1复制
1
输入 #2复制
3 3
0 2 4
2 0 2
4 2 0
输出 #2复制
7

#include
#include
#include
using namespace std;
int a,b,cnt,f[650];
const int inf=0x3f3f3f3f;
struct node{
  int u,v,w;
}e[250005];
void build(int u,int v,int w){
    e[++cnt].u=u;
    e[cnt].v=v;
    e[cnt].w=w;
}
bool cmp(node a,node b){
    return a.w<b.w;
}
int find(int x){
    if(x==f[x]){
        return x;
    }
    else return f[x]=find(f[x]);
}
void merge(int x,int y){
    int rx=find(x);
    int ry=find(y);
    if(ry!=rx){
        f[ry]=rx;
    }
}
int main(){
  cin>>a>>b;
  int x;
  for(int i=1;i<=b;i++){
      for(int j=1;j<=b;j++){
          cin>>x;
          if(x!=0&&i<=j){
              build(i,j,x);
          }
      }
  }
  for(int i=1;i<=b;i++){
      f[i]=i;
  }
  int ans=0;
  sort(e+1,e+cnt+1,cmp);
  for(int i=1;i<=cnt;i++){
      int x=e[i].u;
      int y=e[i].v;
      if(find(x)!=find(y)){
          merge(x,y);
          if(e[i].w<a)ans+=e[i].w;
          else ans+=a;
      }
  }
  cout<<ans+a<<endl;
}