P1194

发表于 分类于
买礼物

买礼物

最小生成树跑一下即可。最好用 Prim，这里用了 Kruskal。

唯一坑点在于 $a$ 可能小于 $k$。

时间复杂度 $O(n\log n^2)$。

代码：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<vector>
#include<cstring>
#define ll long long
using namespace std;

const ll M=5e5,N=5e2;

ll n,a,x,ans,tot,cnt;

ll fa[N+5];

struct node{
	ll u,v,w;
	bool operator < (const node& rhs) const {
		return w<rhs.w;
	}
}edge[M+5];

ll find(ll x) {
	if(fa[x]==x) return x;
	return fa[x]=find(fa[x]);
}

void uni(ll x,ll y) {
	fa[find(x)]=find(y);
}

void kruskal() {
	for(ll i=1;i<=tot;i++) {
		if(find(edge[i].u)!=find(edge[i].v)) {
			ans+=edge[i].w;cnt++;
			uni(edge[i].u,edge[i].v);
			if(cnt==n-1) return;
		}
	}
}

inline ll read() {
	ll ret=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9') {if(ch=='-') f=-f;ch=getchar();}
	while(ch>='0'&&ch<='9') {ret=(ret<<3)+(ret<<1)+ch-'0';ch=getchar();}
	return ret*f;
}

void write(ll x) {
	static char buf[22];static ll len=-1;
	if(x>=0) {
		do{buf[++len]=x%10+48;x/=10;}while(x);
	}
	else {
		putchar('-');
		do{buf[++len]=-(x%10)+48;x/=10;}while(x);
	}
	while(len>=0) putchar(buf[len--]);
}

int main() {

	a=read();n=read();

	for(ll i=1;i<=n;i++) {
		for(ll j=1;j<=n;j++) {
			x=read();
			if(i!=j&&x==0) x=a;
			if(i>=j) continue;
			edge[++tot].u=i;edge[tot].v=j;edge[tot].w=min(a,x);
		}
	}

	for(ll i=1;i<=n;i++) fa[i]=i;

	sort(edge+1,edge+tot+1);

	ans=a;
	kruskal();

	write(ans);

	return 0;
}