题意:
一个图,不止边有权值,点也有权值
Input:
输入的第一行包含一个整数t,它指示数据集的数量。
第二行中有T个数据集。
在单个数据集中,第一行包含一个整数n,表示qs的数目。
第2行包含n个整数,表示每个qs最喜欢的网络适配器的价格。
在第N+2行的第3行中,包含一个矩阵,指示Qs对之间电缆的价格。
约束:
输入中的所有整数都是非负的且不超过1000。
Output:
对于每个数据集,在一行中输出最小成本。不需要额外的空行。
注意:
if (i != j)//这个条件是必须的,不然自环边也有了权重。 Map[i][j] = Map[j][i] = x + price[i] + price[j];
Prim:
一开始maxn设置太大导致 段错误
#include<iostream>
#include<string>
#include<cmath>
#include<ctype.h>
#include<memory.h>
#include<string.h>
#include<algorithm>
#include<map>
#include<iomanip>
#include<set>
#include<list>
#include<vector>
#include<stack>
#include<queue>
#define ll long long int
using namespace std;
const int INF = 0x3f3f3f3f;
const int maxn = 1010;
int t;
int n;
int price[maxn];
int Map[maxn][maxn];
int dis[maxn];
int vis[maxn];
void prim()
{
memset(vis, 0, sizeof(vis));
for (int i = 1; i <= n; i++)
dis[i] = Map[1][i];
vis[1] = 1;
for (int i = 1; i <= n; i++)
{
int Min = INF, min_index = -1;
for (int j = 1; j <= n; j++)
{
if (dis[j] < Min && !vis[j])
{
Min = dis[j];
min_index = j;
}
}
vis[min_index] = 1;
for (int j = 1; j <= n; j++)
{
if (dis[j] > Map[min_index][j] && !vis[j])
dis[j] = Map[min_index][j];
}
}
}
int main()
{
cin >> t;
while (t--)
{
cin >> n;
for (int i = 1; i <= n; i++)
for (int j = 1; j <= n; j++)
Map[i][j] = i == j ? 0 : INF;
for (int i = 1; i <= n; i++)
cin >> price[i];
for (int i = 1; i <= n; i++)
for (int j = 1; j <= n; j++)
{
int x;
cin >> x;
if (i != j)//这个条件是必须的,不然自环边也有了权重。
Map[i][j] = Map[j][i] = x + price[i] + price[j];
}
prim();
int ans = 0;
for (int i = 1; i <= n; i++)
ans += dis[i];
cout << ans << endl;
}
return 0;
}
Kruskal:
#include<iostream>
#include<string>
#include<cmath>
#include<ctype.h>
#include<memory.h>
#include<string.h>
#include<algorithm>
#include<map>
#include<iomanip>
#include<set>
#include<list>
#include<vector>
#include<stack>
#include<queue>
#define ll long long int
using namespace std;
const int INF = 0x3f3f3f3f;
const int maxn = 999999;
struct edge
{
int u, v, w;
};
bool cmp(edge a, edge b)
{
return a.w < b.w;
}
edge a[maxn];
int price[maxn];
int par[maxn];
int n, t;
void ini()
{
for (int i = 0; i <= n; i++)
par[i] = i;
}
int find(int x)
{
if (par[x] == x)
return par[x];
return par[x] = find(par[x]);
}
int main()
{
cin >> t;
while (t--)
{
cin >> n;
ini();
for (int i = 1; i <= n; i++)
cin >> price[i];
int cnt = 0;
for (int i = 1; i <= n; i++)
{
for (int j = 1; j <= n; j++)
{
int x;
cin >> x;
if (i != j)
{
a[cnt].u = i;
a[cnt].v = j;
a[cnt].w = x + price[i] + price[j];
cnt++;
}
}
}
sort(a, a + cnt, cmp);
int ans = 0;
for (int i = 0; i < cnt; i++)
{
int fa = find(a[i].u);
int fb = find(a[i].v);
if (fa != fb)
{
par[fa] = fb;
ans += a[i].w;
}
}
cout << ans << endl;
}
return 0;
}