Kruskal(缩点) - 联络员 - AcWing 1143

Kruskal(缩点) - 联络员 - AcWing 1143

题意：

$给定一个n个点，m条无向边构成的图。$

$边分为两种。$

$①、必选边，这类边必须要选择连接(即使是重边)。$

$②、可选边，这类边可选可不选。$

$现要选择部分可选边，使得整个图连通，求最小代价。$

输入格式

第一行两个整数 n，m 表示Tyvj一共有 n 个管理员，有 m 个通信渠道;

第二行到 m+1 行，每行四个非负整数，p,u,v,w 当 p=1 时，表示这个通信渠道为必选通信渠道；当 p=2 时，表示这个通信渠道为选择性通信渠道；u,v,w 表示本条信息描述的是 u，v 管理员之间的通信渠道，u 可以收到 v 的信息，v 也可以收到 u 的信息，w 表示费用。

输出格式

一个整数，表示最小的通信费用。

数据范围

$1\le n\le 2000,1\le m\le 10000$

输入样例：

5 6
1 1 2 1
1 2 3 1
1 3 4 1
1 4 1 1
2 2 5 10
2 2 5 5

输出样例：

9

---

分析：

$我们可将由必选边连接的点构成的连通块整体视作一个点，这样问题就转化为最小生成树问题。$

$每个连通块的代表点为该连通块的祖宗节点，用并查集维护。$

具体落实：

$①、若输入的是必选边，就将点加入到对应的连通块中去，同时累加权重。$

$②、若输入的是可选边，就将这条边先存下来。然后在连通块之间跑kruskal。$

代码：

#include<iostream>
#include<cstring>
#include<algorithm>

using namespace std;

const int N=2010, M=10010;

int n,m;
int res,idx;
int p[N];
struct edge
{
    int u,v,w;
    bool operator < (const edge &t) const
    {
        return w<t.w;
    }
}E[M];

int Find(int x)
{
    if(p[x]!=x) return p[x]=Find(p[x]);
    return p[x];
}

void kruskal()
{
    sort(E,E+idx);
    
    int pu,pv;
    for(int i=0;i<idx;i++)
    {
        pu=Find(E[i].u),pv=Find(E[i].v);
        if(pu!=pv)
        {
            res+=E[i].w;
            p[pu]=pv;
        }
    }
}

int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++) p[i]=i;
    
    int t,u,v,w,pu,pv;
    while(m--)
    {
        cin>>t>>u>>v>>w;
        pu=Find(u),pv=Find(v);
        if(t==1)    //必选边
        {
            p[pu]=pv;
            res+=w;
        }
        else E[idx++]={u,v,w};
    }
    
    kruskal();
    
    cout<<res<<endl;
    
    return 0;
}