布线问题
时间限制:1000 ms | 内存限制:65535 KB
难度:4
描述
南阳理工学院要进行用电线路改造,现在校长要求设计师设计出一种布线方式,该布线方式需要满足以下条件:
1、把所有的楼都供上电。
2、所用电线花费最少
输入
第一行是一个整数n表示有n组测试数据。(n<5)
每组测试数据的第一行是两个整数v,e.
v表示学校里楼的总个数(v<=500)
随后的e行里,每行有三个整数a,b,c表示a与b之间如果建铺设线路花费为c(c<=100)。(哪两栋楼间如果没有指明花费,则表示这两栋楼直接连通需要费用太大或者不可能连通)
随后的1行里,有v个整数,其中第i个数表示从第i号楼接线到外界供电设施所需要的费用。( 0 < e < v*(v-1)/2 )
(楼的编号从1开始),由于安全问题,只能选择一个楼连接到外界供电设备。
数据保证至少存在一种方案满足要求。
输出
每组测试数据输出一个正整数,表示铺设满足校长要求的线路的最小花费。
样例输入
1
4 6
1 2 10
2 3 10
3 1 10
1 4 1
2 4 1
3 4 1
1 3 5 6
样例输出
4
[分析]
比较简单的最小生成树问题。
直接用prim或kruskal做,然后取最小 外界供电设备 相加得ans。
[代码]
Kruskal
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define MAX 505
struct UnionFind
{
int fa[MAX];
int num[MAX];
UnionFind()
{
make_set();
}
void make_set()//初始化
{
for (int i = 0; i < MAX; i++)
{
fa[i] = i;
num[i] = 0;
}
}
int find_set(int x)//查
{
return x == fa[x] ? x : fa[x] = find_set(fa[x]);
}
void union_set(int a, int b)//合并
{
int x = find_set(a);
int y = find_set(b);
if (x != y)
{
fa[y] = x;
num[x] += num[y];
}
}
};
struct Node
{
int a, b, c;
}node[150000];
bool cmp(Node A, Node B)
{
return A.c < B.c;
}
int main()
{
int t;
scanf("%d", &t);
while (t--)
{
UnionFind uf;
int v, e;
scanf("%d%d", &v, &e);
for (int i = 0;i < e;i++)
{
scanf("%d%d%d", &node[i].a, &node[i].b, &node[i].c);
}
sort(node, node + e, cmp);
int minn = 0x3f3f3f3f;
int x;
for (int i = 0;i < v;i++)
{
scanf("%d", &x);
minn = min(minn, x);
}
int ans=0;
for (int i = 0;i < e;i++)
{
if (uf.find_set(node[i].a)!= uf.find_set(node[i].b))
{
uf.union_set(node[i].a, node[i].b);
ans += node[i].c;
if (uf.num[uf.find_set(node[i].a)]==v)break;
}
}
printf("%d\n", ans + minn);
}
}
prim
#include<cstdio>
#include<cstring>
#define inf 0x3f3f3f3f
int n;
int dis[505];
int vis[505];
int map[505][505];
void Prim() {
int i, j, k, tmp;
for (i = 1;i <= n;i++)//n为楼数
{
dis[i] = inf;//初始化
}
dis[1] = 0;
for (i = 1;i <= n;i++)
{
tmp = inf;
for (j = 1;j <= n;j++)
{
if (!vis[j] && tmp>dis[j])
{
tmp = dis[j];
k = j;
}//找出最小距离的节点
}
vis[k] = 1;//把访问的节点做标记
for (j = 1;j <= n;j++) {
if (!vis[j] && dis[j]>map[k][j])
dis[j] = map[k][j];//更新最短距离
}
}
}
int main()
{
int t;
int a, b, c;
scanf("%d", &t);
while (t--)
{
int v, e;
memset(vis, 0, sizeof(vis));
memset(map, inf, sizeof(map));
scanf("%d%d", &v, &e);
n = v;
for (int i = 0;i < e;i++)
{
scanf("%d%d%d", &a, &b, &c);
map[a][b] = map[b][a] = c;
}
int minn = inf;
int x;
for (int i = 0;i < v;i++)
{
scanf("%d", &x);
if (x < minn)minn = x;
}
Prim();
int ans = 0;
for (int i = 1;i <= v;i++)
{
ans += dis[i];
}
printf("%d\n", ans + minn);
}
}