问题描述
栋栋居住在一个繁华的C市中,然而,这个城市的道路大都年久失修。市长准备重新修一些路以方便市民,于是找到了栋栋,希望栋栋能帮助他。
C市中有n个比较重要的地点,市长希望这些地点重点被考虑。现在可以修一些道路来连接其中的一些地点,每条道路可以连接其中的两个地点。另外由于C市有一条河从中穿过,也可以在其中的一些地点建设码头,所有建了码头的地点可以通过河道连接。
栋栋拿到了允许建设的道路的信息,包括每条可以建设的道路的花费,以及哪些地点可以建设码头和建设码头的花费。
市长希望栋栋给出一个方案,使得任意两个地点能只通过新修的路或者河道互达,同时花费尽量小。
输入格式
输入的第一行包含两个整数n, m,分别表示C市中重要地点的个数和可以建设的道路条数。所有地点从1到n依次编号。
接下来m行,每行三个整数a, b, c,表示可以建设一条从地点a到地点b的道路,花费为c。若c为正,表示建设是花钱的,如果c为负,则表示建设了道路后还可以赚钱(比如建设收费道路)。
接下来一行,包含n个整数w_1, w_2, …, w_n。如果w_i为正数,则表示在地点i建设码头的花费,如果w_i为-1,则表示地点i无法建设码头。
输入保证至少存在一个方法使得任意两个地点能只通过新修的路或者河道互达。
输出格式
输出一行,包含一个整数,表示使得所有地点通过新修道路或者码头连接的最小花费。如果满足条件的情况下还能赚钱,那么你应该输出一个负数。
样例输入
5 5
1 2 4
1 3 -1
2 3 3
2 4 5
4 5 10
-1 10 10 1 1
样例输出
9
样例说明
建设第2、3、4条道路,在地点4、5建设码头,总的花费为9。
数据规模和约定
对于20%的数据,1<=n<=10,1<=m<=20,0<=c<=20,w_i<=20;
对于50%的数据,1<=n<=100,1<=m<=1000,-50<=c<=50,w_i<=50;
对于70%的数据,1<=n<=1000;
对于100%的数据,1 <= n <= 10000,1 <= m <= 100000,-1000<=c<=1000,-1<=w_i<=1000,w_i≠0。
AC的C++程序如下:
#include<iostream>
#include<queue>
#include<string>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=10000+5;
int cost[maxn];//每个码头的费用
int father[maxn];
void init(int n)
{
for(int i=0;i<=n;i++)
{
father[i]=i;
}
}
int find(int n)
{
return n==father[n]?n:father[n]=find(father[n]);
}
void merge(int a,int b)
{
a=find(a);
b=find(b);
if(a==b) return;
else
{
father[a]=b;
return;
}
}
struct edge
{
int u,v;
int len;
edge(){
}
edge(int u1,int v1,int len1)
{
u=u1,v=v1,len=len1;
}
bool operator<(const edge &s) const
{
return len>s.len;
}
};
int n,m;
int main()
{
priority_queue<edge> q1,q2;//分别保存有=无码头和有码头的道路信息
cin>>n>>m;
init(n);
for(int i=1;i<=m;i++)
{
int u,v,len;
cin>>u>>v>>len;
q1.push(edge(u,v,len));
q2.push(edge(u,v,len));
}
for(int i=1;i<=n;i++)
{
cin>>cost[i];
if(cost[i]!=-1) q2.push(edge(0,i,cost[i]));
}
int sum1=0,sum2=0;//分别保存无码头和有码头的总花费
while(!q1.empty())
{
edge top=q1.top();
q1.pop();
if(find(top.u)!=find(top.v)||top.len<0) //若不在同一个连通分量中或是赚钱的道路则合并
{
merge(top.u,top.v);
sum1+=top.len;
}
}
int root=find(1),flag=1;
for(int i=1;i<=n;i++)
{
if(find(i)!=root)
{
flag=0;//若只走陆路不能连通所有城市,则将flag置为0
break;
}
}
init(n);
while(!q2.empty())
{
edge top=q2.top();
q2.pop();
if(find(top.u)!=find(top.v)||top.len<0) //若不在同一个连通分量中或是赚钱的道路则合并
{
merge(top.u,top.v);
sum2+=top.len;
}
}
if(flag==0) cout<<sum2<<endl;
else cout<<min(sum1,sum2)<<endl;
}