#PTA甲級#1021 Deepest Root

自己寫出來發現比搜到的都要短一些,發出來存一下好哈哈哈哈哈哈

#include <iostream>
#include <vector>
#include <string.h>

using namespace std;

const int N = 10100;

vector<int> g[N];
int height[N];//高度
bool vis[N];
int mmax=-1;

int n;

int dfs(int s) {
	vis[s]=1;
	
	int inmax=-1;
	int height=0;
	for(int i=0; i<g[s].size(); i++) {
		if(!vis[g[s][i]]) {
			int tmp = dfs(g[s][i]);			
			if(tmp>inmax)inmax=tmp;			
		}
	}

	return inmax+2;
}
int main() {
	scanf("%d",&n);

	int a=0,b=0;
	for(int i=1; i<n; i++) {
		scanf("%d %d",&a,&b);
		g[a].push_back(b);
		g[b].push_back(a);
	}

	
	for(int i=1; i<=n; i++) {
		memset(vis,0,sizeof(vis));		
		height[i] = dfs(i);
		int comp=1;
		
		if(height[i]>mmax)mmax=height[i];

		for(int j=1; j<=n; j++) {
			if(!vis[j]){			 
				dfs(j);
				comp++;
			}
		}

		if(comp>1) {
			printf("Error: %d components",comp);
			return 0;
		}
	}
	
	for(int i=1;i<=n;i++){
		if(height[i]==mmax){
			printf("%d\n",i);
		}
	}
	return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章