问题描述
栋栋居住在一个繁华的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
这道题目我用的是最小生成树的prim算法。
建两个数组d,d2分别表示陆路和水路的离最小生成树的团的距离。首先把它们全部赋值为无穷大。
然后找到建码头花费最小的点作为最初的起点。把它的d,d2设为0。其他点到这个团的最短水路就是各自码头的花费和这个起点的花费的和。
采用prim算法,找到离团最近的点,找一遍陆路再找一遍水路。如果下一个点是水路的话,就把s标识设为1,这样在总距离加和的时候就能区分水路或者是陆路。但是更重要的是你在团里建了一个码头之后就能把其他点的水路距离仅变成它码头的花费,这里设置了一个FLAG变量,标识你现在的团子里面有没有码头。需要特别注意的是题目给的数据两个点之间的道路可能不止一条,用Prim算法的时候用的是最小的距离。然而另外的距离也不是多余的,因为题目要求赚钱的道路一定要修。所以刚开始把所有负的距离都加到总距离里面。之后负的距离就不加了。
“`
include
include
using namespace std;
const int maxn=10010;
const int inf=10000000;
int n,m;
int dis[maxn][maxn];
int d[maxn];
int d2[maxn];
int w[maxn];
int vis[maxn]= {0};
int flag=0;
int prim(int x)
{
fill(d,d+maxn,inf);
d[x]=0;
d2[x]=0;
int ans=0;
for(int i=0; i<n; i++)
{
int s=0;
int sel=-1,min=inf;
for(int i=1; i<=n; i++)
{
if(min>d[i]&&vis[i]==0)
{
min=d[i];
sel=i;
}
}
for(int i=1; i<=n; i++)
{
if(min>d2[i]&&vis[i]==0)
{
min=d2[i];
sel=i;
s=1;
}
}
if(sel==-1)return -1;
vis[sel]=1;
if(s==0)
{
if(d[sel]>0)ans+=d[sel];
}
else
{
ans+=d2[sel];
if(flag==0)
{
for(int i=1; i<=n; i++)
{
if(w[i]!=-1)d2[i]=w[i];
}
}
flag=1;
}
for(int i=1; i<=n; i++)
{
if(dis[sel][i]!=inf&&dis[sel][i]<d[i]&&vis[i]==0)
{
d[i]=dis[sel][i];
}
}
}
return ans;
}
int main()
{
freopen(“d://jin.txt”,”r”,stdin);
cin>>n>>m;
fill(dis[0],dis[0]+maxn*maxn,inf);
int a,b,c;
int ans2=0;
for(int i=0; i