1488:新的开始
时间限制: 1000 ms 内存限制: 65536 KB
提交数: 773 通过数: 401
【题目描述】
发展采矿业当然首先得有矿井,小 FF 花了上次探险获得的千分之一的财富请人在岛上挖了 n 口矿井,但他似乎忘记考虑的矿井供电问题……
为了保证电力的供应,小 FF 想到了两种办法:
在这一口矿井上建立一个发电站,费用为 v(发电站的输出功率可以供给任意多个矿井)。
将这口矿井与另外的已经有电力供应的矿井之间建立电网,费用为 p。
小 FF 希望身为「NewBe_One」计划首席工程师的你帮他想出一个保证所有矿井电力供应的最小花费。
【输入】
第一行一个整数 n,表示矿井总数。
第 2∼n+1 行,每行一个整数,第 i 个数 vi 表示在第 i 口矿井上建立发电站的费用。
接下来为一个 n×n 的矩阵 p,其中 pi,j 表示在第 i 口矿井和第 j 口矿井之间建立电网的费用(数据保证有pi,j=pj,i ,且 pi,i=0。
【输出】
输出仅一个整数,表示让所有矿井获得充足电能的最小花费。
【输入样例】
4
5
4
4
3
0 2 2 2
2 0 3 3
2 3 0 4
2 3 4 0
【输出样例】
9
【提示】
样例解释
小 FF 可以选择在 4 号矿井建立发电站然后把所有矿井都不其建立电网,总花费是 3+2+2+2=9。
数据范围:
对于 30% 的数据:1≤n≤50;
对于 100% 的数据:1≤n≤300,0≤vi,pi,j≤105 。
思路:当边权特别大,不取这条边,分别修两个电站才是最优的。
正确的解法是在原图的基础上虚拟一个点,原图的每一个节点都与这 个虚拟点有一条边,边权为在该点上修建电站的花费。
再对新图做最小生成树。
#include <bits/stdc++.h>
#define inf 1<<30
#define ll long long
using namespace std;
int n,x,fa[305],num=0;
struct edge
{
int u,v,w;
bool operator<(const edge &x)const
{
return w<x.w;
}
}a[305*305];
int Find(int x)
{
if(fa[x]!=x)
fa[x]=Find(fa[x]);
return fa[x];
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
cin >> n;
for(int i=1;i<=n;i++)
{
cin >> x;
a[num].u=i,a[num].v=n+1,a[num++].w=x;
fa[i]=i;
}
fa[n+1]=n+1;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
cin >> x;
if(i<j)
a[num].u=i,a[num].v=j,a[num++].w=x;
}
}
sort(a,a+num);
int cnt=0;
ll s=0;
for(int i=0;i<num;i++)
{
int xx=Find(a[i].u),yy=Find(a[i].v);
if(xx!=yy)
fa[xx]=yy,s+=a[i].w;
if(cnt==n)
break;
}
cout << s << endl;
return 0;
}