时间限制: 1 Sec 内存限制: 128 MB
题目描述
小粉兔用集训队的奖金买下了一片地。
这片地上有 n 个房子,有些房子之间有道路,有些房子之间则是杂草。
她可以花费一定的代价拆毁一条道路,或是啃光一片草使得两个房子间可以通行(大雾)。
她喜欢生成树,所以她要让所有道路形成一棵生成树。
求最小花费。
输入
第一行一个数,n。
接下来 n 行,每行 n 个数,代表 ai,j。如果为正数,说明它们之间没有道路,需要 ai,j的花费来修建;如果为负数,说明它们之间有道路,需要 −ai,j的花费来拆毁。
输出
一行一个数,代表最小花费。
样例输入
3
0 1 -3
1 0 5
-3 5 0
样例输出
1
提示
对于 100% 的数据,5≤n≤1000,1≤|ai,j(i≠j)|≤1000,保证所有 ai,i=0且 ai,j=aj,i 。
思路
这是一道最小生成树的题目,平时做的都是只有建立的,没有拆的,(所以一开始我不出意料的WA了) 。其实也 没有什么特殊的,既然有拆的过程,那么你就先拆(拆掉多余的)在建。
一开始先把正数,负数,分开存,正数是建路要用的,负数是拆路要用的,然后都排序
对于负数(有路),从大到小枚举每条路,如果这条路的两点没有联通就联通它们,如果已经联通了,那么就拆掉这条路(有点贪心的意思)
对于正数(无路),就是正常程序了,从小到大枚举,生成树,结束。
最后答案相加就是最小代价
(最后感谢ybl大佬的帮助!!!)
代码:
#include<iostream>
#include<string>
#include<map>
#include<queue>
#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
#include<iomanip>
#include<cmath>
//#include<unordered_map>
#define X first
#define Y second
#define INF 0x3f3f3f3f
using namespace std;
typedef long long ll;
const double eps=1e-6;
const int N=1e6+10;
int n,mp[1010][1010],pre[1010],ans;
struct node
{
int x;
int y;
int l;
}s[N],d[N];
bool cmp(node a,node b)
{
return a.l<b.l;
}
int find(int x)
{
return x==pre[x]?x:(pre[x]=find(pre[x]));
}
int Union(int x,int y)
{
int x1=find(x);
int x2=find(y);
if(x1!=x2)
{
pre[x1]=x2;
return 0;
}
else return 1;
}
int main( )
{
ios::sync_with_stdio(false);
cin>>n;
for(int i=1;i<=n;i++) pre[i]=i;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
cin>>mp[i][j];
}
}
int k=0,t=0;
for(int i=1;i<=n;i++)
{
for(int j=i+1;j<=n;j++)
{
if(mp[i][j]<0)
{
d[t].x=i;
d[t].y=j;
d[t].l=-mp[i][j];
t++;
}
else
{
s[k].x=i;
s[k].y=j;
s[k].l=mp[i][j];
k++;
}
}
}
sort(s,s+k,cmp);
sort(d,d+t,cmp);
for(int i=t-1;i>=0;i--)
{
if(Union(d[i].x,d[i].y)==1)
ans+=d[i].l;
}
for(int i=0;i<k;i++)
{
if(Union(s[i].x,s[i].y)==0)
ans+=s[i].l;
}
cout<<ans;
return 0;
}