hdu1879 悲劇的kruscal版本

之前已經給了幾個prim的版本，沒有注意到時間都達到800+ms，瀕臨上界，看來還是prim算法效率更高一點，不過後者可能更簡單直觀一點，一開始使用 vector+sort 使用並查集 一提交超時，檢查好幾遍 確定沒有邏輯問題，在並查集部分又優化了一點，儘量降低樹的高度直接把結點掛在根上 不過還是超時 鬱悶~~找來網上的一個C版本，看完之後自己重寫了一遍  竟然也超時 對比好幾遍 邏輯完全一樣 只是把c版本改寫成了C++版本，個人對c不太熟，後來才發現完全是Cin的原因 格式化輸入對於大量數據輸入很耗時，改成scanf就能AC，看來還得好好補補C語言的課呀，這裏最多隻有5050個數據 （可能測試用例也很多）以後要用一個大概的概念 數據量太大時用scanf

 

 

kruscal（使用priority_queue）版本：

#include <iostream>
#include <queue>
using namespace std;

typedef struct
{
 int cost;
 int v1,v2;
}node;

bool operator >(node a,node b)//重載運算符，比較規則
{
 if(a.cost>b.cost)return true;
 return false;
}

priority_queue< node,vector<node>,greater<node> >Q;

int fa[105];

int getFa(int t)
{
 if(fa[t]==t)return t;
 fa[t]=getFa(fa[t]);
 return fa[t];
}

void input(int n)
{
 node temp;
 for(int i=1;i<=n;i++)fa[i]=i;
 int m=n*(n-1)/2;

 for (i=1;i<=m;i++)
 {
  int v1,v2,cost,tag;
  scanf( "%d%d%d%d" , &v1 , &v2 , &cost , &tag );
  if(tag)fa[getFa(v2)]=v1;
  else
  {
   temp.cost=cost;
   temp.v1=v1;
   temp.v2=v2;
   Q.push(temp);
  }
 }
}

void work()
{
 int sumprice=0;
 while(!Q.empty())
 {
  node temp=Q.top();
  Q.pop();
  if(getFa(temp.v1)!=getFa(temp.v2))
  {
   sumprice+=temp.cost;
   fa[getFa(temp.v2)]=getFa(temp.v1);//直接掛在樹根更高效
  }
 }
 cout<<sumprice<<endl;
}

int main()
{
 int n;
 while(cin>>n&&n)
 {
  input(n);
  work();
 }
 return 0;
}

 

 

 

 

 

 

 

kruscal（使用vector+sort）版本：

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

int part[100];
struct node//克魯斯卡爾算法是加邊法，使用node結構表示邊信息，而不是prim的數組
{
    int from;
    int to;
    int cost;
};
vector<node> Vec;

int getFa(int n)
{
    if(part[n]==n)return n;
    part[n]=getFa(part[n]);
    return part[n];
}

bool merge(int x,int y)
{
    if(getFa(x)==getFa(y))return false;
    part[getFa(y)]=x;return true;
}
bool cmp(node x,node y)
{
    return x.cost<y.cost;
}
int main()
{
    int n,m;//n爲邊數目，m爲頂點數目
    while(cin>>n&&n)
    {
        Vec.clear();
        for (int i=1;i<=n;i++)part[i]=i;
        int from,to,cost,tag;
        node temp;
  m=n*(n-1)/2;
        while(m--)
        {
            scanf("%d%d%d%d",&from,&to,&cost,&tag);
            temp.from=from;
            temp.to=to;
            if(tag)temp.cost=0;
   else temp.cost=cost;
            Vec.push_back(temp);           
        }
        sort(Vec.begin(),Vec.end(),cmp);
        int count=n-1;
        int sum=0;
        for(vector<node>::iterator iter=Vec.begin();iter!=Vec.end();iter++)
            if(getFa(iter->from)!=getFa(iter->to))
            {
    merge(iter->from,iter->to);
                sum+=iter->cost;
                count--;
                if(!count)break;
            }
             cout<<sum<<endl;                   
    }
    return 1;
}