Caterpillar POJ - 3310

题目链接：https://cn.vjudge.net/problem/POJ-3310

题目

判断一个图是否是一棵树（是否全部连通并且不存在环），并且满足条件，存在一条链，使得所有点要么在链上要么与链上的点距离为1。

分析

树的直径？反正我没用树的直径做，下面是我的做法。先用并查集判断是否连通以及是否有环，然后我们可以发现，如果满足条件的话，当我们把所有度为1的点全部拿出来之后（当然，与这些点相连的点的度也要发生变化），那么剩下的点的度一定小于等于2，所以只判断剩下的点当中有没有度大于2的点就可以了。

代码

#include<iostream>
#include<cstdio>
#include<string.h>
#include<vector>
#include<queue>
#include<cmath>
using namespace std;
int n,m,deg[110],num[110],pre[110];
vector<int> v[110];
void init()
{
	for(int i=0;i<110;i++)
	{
		deg[i] = 0;
		num[i] = 1;
		pre[i] = i;
		v[i].clear();
	}
}
int find(int x)
{
	int r = x;
	while(r != pre[r])
		r = pre[r];
	int tmp,a=x;
	while(a != r)
	{
		tmp = pre[a];
		pre[a] = r;
		a = tmp;
	}
	return r;
}
void unionn(int x,int y)
{
	int rx = find(x);
	int ry = find(y);
	if(rx != ry)
	{
		pre[ry] = rx;
		num[rx] += num[ry];
	}
}
int solve()
{
	int cnt = 0;
	queue<int> q;
	for(int i=1;i<=n;i++)
	{
		if(deg[i] == 1)
		{
			cnt++;
			q.push(i);
		}
	}
	while(!q.empty())
	{
		int t = q.front();
		q.pop();
		for(int i=0;i<v[t].size();i++)
		{
			int num = v[t][i];
			deg[num]--;
		}
	}
	for(int i=1;i<=n;i++)
	{
		if(deg[i] > 2)
			return 0;
	}
	return 1;
}
int main()
{
	int cas = 1;
	while(~scanf("%d",&n))
	{
		if(n==0) break;
		scanf("%d",&m);
		init();
		int flag = 1;
		for(int i=0;i<m;i++)
		{
			int a,b;
			scanf("%d%d",&a,&b);
			deg[a]++;
			deg[b]++;
			v[a].push_back(b);
			v[b].push_back(a);
			if(find(a) == find(b))
				flag = 0;
			unionn(a,b);
		}
		int rt = find(1);
		if(!flag || num[rt] != n)
		{
			printf("Graph %d is not a caterpillar.\n",cas++);
			continue;
		}
		int ans = solve();
		if(!ans) printf("Graph %d is not a caterpillar.\n",cas++);
		else printf("Graph %d is a caterpillar.\n",cas++);
	}
	return 0;
}