hdu 4313 Matrix (kruskal模板题,cmp倒着取边)

版权声明:本文为博主原创文章,转载请注明出处~~ https://blog.csdn.net/hxc2101/article/details/82108972

题目:http://acm.hdu.edu.cn/showproblem.php?pid=4313

题意:有n个城市,有n-1条边连通,其中的k个城市有破坏机器,要求切断部分路,使得这k个城市之间不能连通,每条路对应不同的花费时间,求最少花费。

思路:以含“破坏机器”的城市作为根结点,分为k棵树,每加入一条边,判断该边的两个端点所对应的祖先是不是都含“机器”,若都含不连这条边直接在ans上加,其他情况的只要在join函数中改下条件即可,两者若其中之一含机器,那么不含“机器”的点接在含“机器”的点下,若都不含机器,就随便接好了。

注意 1.因为要使去掉的边代价最小,即越后面加入的越有可能不符合条件,即需要按边从大到小排序(即cmp倒着来)

        2.ans 是 long long , kruskal() 函数的返回类型也要改为long long(一开始WA在这里)

#include <bits/stdc++.h>
#pragma GCC optimize(3)
//#pragma GCC optimize("unroll-loops")
//#pragma comment(linker, "/stack:200000000")
//#pragma GCC optimize("Ofast,no-stack-protector")
//#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
#define fio ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)
#define pb push_back
#define mkp(a,b) make_pair(a,b)
#define PII pair<int,int>
#define PLL pair<ll,ll>
#define fi first
#define se second
#define lc (d<<1) //d*2
#define rc (d<<1|1) //d*2+1
#define eps 1e-9
#define dbg(x) cerr << #x << " = " << x << "\n";
#define mst(a,val) memset(a,val,sizeof(a))
#define stn(a) setprecision(a)//小数总有效位数
#define stfl setiosflags(ios::fixed)//点后位数:cout<<stfl<<stn(a);
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const double PI=3.1415926535897932;
const int MAXN=1e5+10;
const ll mod=1e9+7;
ll inline mpow(ll a,ll b){ll ans=1;a%=mod;while(b){if(b&1)ans=(ans*a)%mod;a=(a*a)%mod,b>>=1;}return ans;}
int inline sgn(double x){return (x>-eps)-(x<eps);} //a<b:sgn(a-b)<0
priority_queue<int,vector<int>,greater<int> > qu; //up
priority_queue<int,vector<int>,less<int> > qd; //dn
const int inf = 0x3f3f3f3f; //9
const ll inff = 0x3f3f3f3f3f3f3f3f; //18

int T;
int n,k;
int x,y,z;
int id[100010];

//并查集
int pre[100010];
inline void init(int n) {for(int i=0;i<=n;i++) pre[i]=i;}
inline int find(int x)
{
    int r=x,i=x,j;
    while(pre[r]!=r) r=pre[r];
    while(i!=r) {j=pre[i];pre[i]=r;i=j;}
    return r;
}
inline void join(int x,int y)
{
    int a=find(x),b=find(y);
    if(id[a]) pre[b]=a;
    else pre[a]=b;
}
//kruskal
int Ver,E;
struct Edg {int u,v,d;}edg[100010];
inline bool cmp(Edg a,Edg b)  {return a.d>b.d;}
inline ll kruskal()
{
    sort(edg,edg+E,cmp);
    ll ans=0;
    for(int i=0;i<E;i++)
    {
        Edg tmp=edg[i];
        int a=find(tmp.u),b=find(tmp.v);
        if(id[a]&&id[b])
            ans+=(ll)tmp.d;
        else join(a,b);
    }
    return ans;
}

int main()
{
    fio;
    cin>>T;
    while(T--)
    {
        cin>>n>>k;
        init(n); Ver=n; E=n-1;
        mst(id,0);
        for(int i=0;i<n-1;i++) cin>>edg[i].u>>edg[i].v>>edg[i].d;
        for(int i=0;i<k;i++) {int tmp; cin>>tmp; id[tmp]=1;}
        ll ans=kruskal();
        cout<<ans<<endl;
    }
}

猜你喜欢

转载自blog.csdn.net/hxc2101/article/details/82108972