1021 Deepest Root (25 分)

68 篇文章 0 订阅
27 篇文章 0 订阅

DFS
似乎找到了PAT的一些规律,如果考的是数据结构里面的东西,就很少有坑。
而如果考的模拟或者简单的题,就有很多边界条件。
使用了两个全局变量,记录每个起点遍历的最大深度和所有顶点的最大深度。
一次性AC,还是比较开心的!

版本1

只有入度为1的点才可能是最深的根节点,已经尽力优化了,但是还是过不了牛客网的同名题。
PAT提交从原来的1100ms提升到了663ms不知道还能怎么优化。

#include<cstdio>
#include<vector>
#include<cstring>
using namespace std;
const int maxn=1e4+5;
vector<int> G[maxn];
vector<int> p;
int maxd,maxD=-1;
int vis[maxn],in[maxn];
void DFS(int x,int deep){
	vis[x]=1;
	for(int i=0;i<G[x].size();i++){
		if(vis[G[x][i]]==0) {
			vis[G[x][i]]=1;
			if(deep > maxd) {
				maxd = deep;
			}
			DFS(G[x][i],deep+1);
		}
	}
}

int main(){
	int n,u,v;
	scanf("%d",&n);
	memset(in,0,sizeof(in));
	for(int i=0;i<n-1;i++){
		scanf("%d%d",&u,&v);
		G[u].push_back(v);
		G[v].push_back(u);
		in[u]++;
		in[v]++;
	} 
	

	for(int k=1;k<=n;k++){
		memset(vis,0,sizeof(vis));
		maxd=0;
		if(k==1){
			int cnt=0;
			for(int i=1;i<=n;i++){
				if(vis[i]==0){
					DFS(i,1);
					cnt++;
				}
			}
			if(cnt>1) {
				printf("Error: %d components",cnt);
				return 0;
			}
		}else{
			if(in[k]!=1) continue;
			DFS(k,1);
		}
		if(maxd > maxD){
			p.clear();
			p.push_back(k);
			maxD=maxd;
		}else if(maxd == maxD){
			p.push_back(k);
		}
	}
	for(int i=0;i<p.size();i++){
		printf("%d\n",p[i]);
	}

	return 0;
}

版本2

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e4+5;
int vis[maxn];
vector<int> G[maxn],ans;
int opt=-1;
struct node{
	int val,layer;
};
int bfs(int r){
	int dep=-1;
	queue<node> q;
	node root;
	root.layer = 0;
	root.val = r;
	q.push(root);
	vis[r]=1;
	while(!q.empty()){
		node tmp = q.front();
		q.pop();
		int u = tmp.val;
		for(int i=0;i<G[u].size();i++){
			int v = G[u][i];
			root.val = v;
			root.layer = tmp.layer + 1;
			if(root.layer > dep) dep = root.layer;
			if(vis[v]==0){
				q.push(root);
				vis[v]=1;
			}
		}
	} 
	return dep;
}
int main(){
	int n,u,v,d;
	scanf("%d",&n);
	for(int i=0;i<n-1;i++){
		scanf("%d%d",&u,&v);
		G[u].push_back(v);
		G[v].push_back(u);
	}
	int cnt=0;
	for(int i=1;i<=n;i++){
		memset(vis,0,sizeof(vis));
		if(i==1){
			for(int j=1;j<=n;j++){ //求连通分量的个数 
				if(vis[j]==0){  //深搜一遍后,没有标记的点继续搜 
					d = bfs(j);
					cnt++; 
				}
			}
			if(cnt>1){
				printf("Error: %d components",cnt);
				return 0;
			} 
		}else d = bfs(i);
		
		if(d > opt){
			opt = d;
			ans.clear();
			ans.push_back(i);
		}else if(d == opt){
			ans.push_back(i);
			}
	}
	for(auto it: ans){
		printf("%d\n",it);
	}
	return 0;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值