zju 2740 Message System

题目的意思就是判断一个图是否是树,一开始我的做法是先判断图是否联通,然后判断是否有环.判断联通用了O(N),然后判断环用了O(N^2),tle了.后来在网上搜了下,想起了判断环可以用并查集,并且并查集也可以判断是否是联通的,而且用最简单的并查集就行了
#include<iostream> #include <string> #include <set> using namespace std; int m, n; int p[1001], num[1001]; int find(int t) { int ot = t; if(p[t] == t) return t; while (p[t] != t) { t = p[t]; } int temp; while (p[ot] != t) { temp = p[ot]; p[ot] = t; ot = temp; } return t; } bool unit(int a, int b) { int pa = find(a); int pb = find(b); if(pa == pb) return false; if(num[pa] > num[pb]) { p[pb] = pa; ++num[pa]; } else { p[pa] = pb; ++num[pb]; } return true; } int main() { int i, j; int a, b; while (scanf("%d%d",&n,&m) != EOF) { if(m==0 && n==0) break; if(m != n-1) { for (i=0; i<m; ++i) scanf("%d%d",&a,&b); printf("No/n"); continue; } for (i=1; i<=n; ++i) { p[i] = i; num[i] = 1; } bool flag = true; for (i=0; i<m; ++i) { scanf("%d%d",&a,&b); if(!unit(a,b)) flag = false; } if(flag) printf("Yes/n") ; else printf("No/n"); } return 0; }