洛谷P1194 买礼物

洛谷P1194 买礼物
在买了第一次之后，你还要买 n-1次，把所有点都连起来，这样做一个最小生成树
就行了，然后要注意题目中说了如果a[ i ][ j ]==0 则表示两个点是不相连的，赋值 inf
最后还要注意一下一开连第一条边是需要A的

#include <bits/stdc++.h> 
#define LL long long 
#define For(i,j,k) for(int i=j;i<=k;i++)  
using namespace std ; 

const int N = 511,inf = 1e9 ; 
int A,n,sum ; 
int dist[N],a[N][N] ; 
bool visit[N] ; 

inline LL read() 
{
    LL x = 0 , f = 1 ; 
    char ch = getchar() ; 
    while(ch<'0'||ch>'9') { if(ch=='-') f = -1 ; ch = getchar() ; } 
    while(ch>='0'&&ch<='9') { x = x * 10+ch-48 ; ch = getchar() ; } 
    return x * f ;  
}

inline void prim() 
{
    For(i,1,n) visit[ i ] = 0 , dist[ i ] = inf ; 
    dist[ 1 ] = 0 ;  dist[ 0 ] = inf ; 
    int k ; 
    For(i,1,n) {
        k = 0 ; 
        For(j,1,n) 
            if(!visit[j]&&dist[j] < dist[ k ] ) k = j ; 
        visit[ k ] = 1 ; 
        For(j,1,n) 
            if(!visit[j]&&a[k][j] < dist[ j ]) dist[ j ] = a[ k ][ j ] ; 
    }
    For(i,1,n) sum+=dist[ i ] ; 
    sum+=A ; 
    printf("%d
",sum) ; 
}

int main() 
{
    A = read() ; n = read() ; 
    For(i,1,n) 
        For(j,1,n) {
            a[ i ][ j ] = read() ; 
            if(a[ i ][ j ] == 0) a[ i ][ j ] = inf ; 
        }
    prim() ; 
    return 0 ; 
}