1. 程式人生 > >hdu——3367 並查集+最大生成樹【模板】

hdu——3367 並查集+最大生成樹【模板】

In graph theory, a pseudoforest is an undirected graph in which every connected component has at most one cycle. The maximal pseudoforests of G are the pseudoforest subgraphs of G that are not contained within any larger pseudoforest of G. A pesudoforest is larger than another if and only if the total value of the edges is greater than another one’s.

題意就是在給定的圖裡面找到一個邊權值的和最大的一個子圖,這個子圖要滿足只有一個環的要求;

思路:可以看出它本質上是要去求邊權值和最大的子圖,那麼就以為著整體上是利用克魯斯卡爾演算法得到的最大生成樹,然後在次基礎上進行一些改變,就是要得到子圖上只有一個環,那麼就可以想到,因為利用kru演算法來實現子圖 的過程中邊是按照邊權less的順序給出,那麼就意味著前一條邊和後一條邊可能不屬於同一個樹,於是就要先對在不在一個樹分析,如果在一個樹裡面,那麼如果想要這條邊連成後不會出現兩個環,只能是兩個端點都不能在環中。再對兩個端點不在一個樹上進行分析,要想這兩個端點連成後只有一個環,那麼只有一下幾種情況:兩個端點都不在環裡面,其中一個在環裡面。

程式碼:

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

using namespace std;

const int maxn=1e5+10;
const int maxm=1e4+10;
struct Edge
{
    int from,val,to;
}edge[maxn];
int pre[maxm];
int cir[maxn];//用來判定是不是在一個環裡面
int n,m;

int find(int x)
{
    if(pre[x]==x)
        return x;
    return  pre[x]=find(pre[x]);
}

bool cmp(Edge a,Edge b)
{
    return a.val>b.val;
}

long long solve()
{
    long long ans=0;
    sort(edge+1,edge+1+m,cmp);
    for(int i=1;i<=m;i++)
    {
        int fx=find(edge[i].from),fy=find(edge[i].to);
        if(fx==fy)
        {
            if(!cir[fx]&&!cir[fy])
            {
                ans+=edge[i].val;
                cir[fx]=cir[fy]=1;
            }
        }
        else
        {
            if(!cir[fx]&&!cir[fy])
            {
                ans+=edge[i].val;
                pre[fx]=fy;
            }
            else if(!cir[fx]||!cir[fy])
            {
                ans+=edge[i].val;
                pre[fx]=fy;
                cir[fx]=cir[fy]=1;
            }
        }
    }
    return ans;
}
 
void ini()
{
    memset(cir,0,sizeof(cir));
    for(int i=0;i<=n;i++)
        pre[i]=i;
}

int main()
{
    while(scanf("%d %d",&n,&m)&&(n||m))
    {
        ini();
        for(int i=1;i<=m;i++)
            scanf("%d %d %d",&edge[i].from,&edge[i].to,&edge[i].val);
        long long ans=solve();
        cout<<ans<<endl;
    }
    return 0;
}