2017年上海金马五校程序设计竞赛（网上资格赛）Problem B : Coach

一开始就是考虑直接并查集，求所有的点所在的树的深度是否都是可以整除3.

#include <iostream>
#include <map>
using namespace std;
map<int,int> p;
int par[100005],r[100005],num[100005];
void init(){
 for(int i=0;i<100005;i++){
  par[i]=i;
  num[i]=1;
 }
}
int findroot(int x){
 if(par[x]==x) return x;
 else return par[x]=findroot(par[x]);
}
void unite(int x,int y){
 x=findroot(x);
 y=findroot(y);
 if(x==y){
  return;
 }
 par[y]=x;
 num[x]+=num[y];
}
int main(){
 int n,m;
 while(scanf("%d%d",&n,&m)!=EOF){
  if(m == 0) {
   printf("No\n");
   continue;
  }
  p.clear();
  init();
  int k=1;
  for(int i=0;i<m;i++){
   int a,b;
   scanf("%d%d",&a,&b);
   if(p.find(a)==p.end()){
    p[a]=k++;
   }
   if(p.find(b)==p.end()){
    p[b]=k++;
   }
   unite(p[a],p[b]);
  }
  for(int i = 1;i<=n;i++){
   if(num[findroot(p[i])]%3!=0){
    printf("No\n");
    break;
   }
   if(i==n&&num[findroot(p[i])]%3==0){
    printf("Yes\n");
   }
  }
 }
}