克魯斯卡爾算法+並查集

算法要點：Kruskal算法的最難點在於怎樣判斷加入邊(x,y)后是否形成了環。

問題可化為：判斷邊(x,y)的兩個頂點x,y在圖（實際是森林）mst中最否已經連通。如果已經連通，加入邊將形成環；否則，不形成環。

 

在kruskal算法中，要用到並查集的合並和查找

並查集：

int getfa(int k)        //找到最祖先
{
    if(fa[k]==k)    return k;
    fa[k]=getfa(fa[k]);
    return fa[k];
}

void merge(int x,int y)       //合並祖先
{
    int fx=getfa(x);
    int fy=getfa(y);
    fa[fx]=fy;
}

bool judge(int x, int y)       //判斷是不是一個祖先
{
    int fx=getfa(x);
    int fy=getfa(y);
    return fx==fy;
}

 

 kruskal算法核心：

for(int i=1;i<=n;i++)
        fa[i]=i;
sort(e+1,e+1+len,mys);
int cal=0;
for(int i=1;i<=len;i++)
{
    int v=getfa(e[i].x);
    int u=getfa(e[i].y);
    if(v!=u)
    {
        merge(v,u);
                //根據題意添加
        if(++cal==n-1)
        {
            break;
        }
    }
}

 

輸入：

int fa[maxn];
int len=0;
struct node
{
    int x,y,v;
}e[maxn];

void init(int xx,int yy,int vv)
{
    if(yy<0||yy>n*m)    return;
    e[++len].y=yy;
    e[len].x=xx;e[len].v=vv;
}

int main()
{
    memset(e,0,sizeof(e));
    cin>>n>>m;
    for(int i=1;i<=m;i++)
    {
        int xx,yy,vv;
        cin>>xx>>yy>>vv;
        init(xx,yy,vv);
        init(yy,xx,vv);
    }