买礼物 洛谷P1194 最小生成树

题目描述

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

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

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

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

输入格式

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

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

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

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

输出格式

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

输入输出样例

输入 #1复制

1 1
0

输出 #1复制

1

输入 #2复制

3 3
0 2 4
2 0 2
4 2 0

输出 #2复制

7

说明/提示

样例解释 22。

先买第 22 样东西，花费 33 元，接下来因为优惠，买 1,31,3 样都只要 22 元，共 77 元。

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

数据规模

对于 30\%30% 的数据，1\le B\le_101≤B≤1​0。

对于 100\%100% 的数据e500,0\le A,K_{I,J}\le10001≤B≤500,0≤A,KI,J​≤1000。

prim算法 当K i j 等于零或者大于没有优惠前的价格 就让K i j 等于 优惠前的价格.

输出的时候要加上买第一件物品的钱

#include <bits/stdc++.h>
using namespace std;
const int MAXN = 1010;const int MAX = 0x3f3f3f3f;
int net[MAXN][MAXN],weight[MAXN],v,n,w;
void prime()
{
	int begin = 1,sum = 0;
	for(int i = 1;i <= n; i++)
		weight[i] = net[begin][i];
	for(int k = 1;k < n; k++){
		int v = 0,min = MAX;
		for(int i = 1;i <= n; i++){
			if(weight[i] > 0 && weight[i] < min){
				v = i;
				min = weight[i];
			}
		}
		sum += weight[v];
		weight[v] = 0;
		for(int i = 1;i <= n; i++){
			if(weight[i] > 0 && net[v][i] < weight[i])
				weight[i] = net[v][i];
		}
	}
	cout << sum+v;
}
int main()
{
	cin >> v >> n;
	for(int i = 1;i <= n; i++)
		for(int j = 1;j <= n; j++){
			cin >> w;
			if(i == j) continue;
			if(w > v || w == 0) net[i][j] = v;
			else net[i][j] = w;
		}
	prime();
	return 0;
}

kruskal

#include <bits/stdc++.h>
using namespace std;
const int INF = 0x3f3f3f3f;
struct info{
	int s,e,w;
	bool operator < (info & a) const
	{
		return w < a.w;
	}
}node[250010];
int pre[510],v,n,c;
int Find(int a)
{
	if(pre[a] == a) return a;
	else return pre[a] = Find(pre[a]);
}
void Union(int a,int b)
{
	int _a = Find(a),_b = Find(b);
	if(a == b) return;
	else{
		pre[_a] = _b;
	}
}
void init()
{
	for(int i = 1;i < 510; i++)
		pre[i] = i;
}
void kruskal(int cnt)
{
	int t = 0,sum = 0;
	for(int i = 0;i < cnt; i++){
		if(Find(node[i].s) != Find(node[i].e)){
			t++;
			Union(node[i].s,node[i].e);
			sum += node[i].w;
		}
		if(t == n-1) break;
	}
	cout << sum + v;
}
int main()
{
	int cnt = 0;
	cin >> v >> n;
	for(int i = 1;i <= n; i++)
		for(int j = 1;j <=n; j++){
			cin >> c;
			if(i == j) c = INF;
			else if(c == 0 || c > v) c = v;
			node[cnt].s = i;node[cnt].e = j;node[cnt].w = c;
			cnt++;
		}
	sort(node,node+cnt-1);
	init();
	kruskal(cnt);
	return 0;
}