tree 用點、邊的算法

由於江老師說上次那個算法是以前的，所以我就用了新學的計算邊數與點數的算法……

程序：

#include <iostream>

#include <fstream>

#include <algorithm> 

using namespace std;

/*

ifstream fin("tree.in");

ofstream fout("tree.out");

#define cin fin

#define cout fout

*/

int fa[100001];

int line[100001],point[100001]; //記錄邊數與點數

int u_f(int x)

{

   int root=x;

   while(fa[root]!=root) root=fa[root]; 

   while(fa[x]!=x) 

   {

      int y=fa[x];

      fa[x]=root;

      x=y;

   }

 

   return root;

}

 

int main()

{

   int n,m;

   cin>>n>>m;

   for(int i=1;i<=n;i++)

   {

          fa[i]=i;

          line[i]=0;

          point[i]=1;     //數初始爲零，點數初始爲一

   } 

   for(int i=0;i<m;i++)

   {

      int x,y;

      cin>>x>>y;

      int fx=u_f(x);

      int fy=u_f(y);

      if(fx!=fy)

      {

       fa[fy]=fx;

       point[fx]+=point[fy];  //點數相加

       line[fx]+=line[fy]+1;  //邊數相加再加 1

     }

     else line[fx]++; 

//如果要連接已經相連的兩個連通塊，則爲環形，邊數加一

   }

   int sum=0;

   for(int i=1;i<=n;i++)

      if(fa[i]==i && line[i]-point[i]==-1) sum++;

    cout<<sum<<endl; 

   return 0; 

}