upc 兔

时间限制: 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;
}