2017小米服务端开发笔试(求二叉树高度)

、本题的思路:dfs#include #include #include #include #include #include #include #include #include #include #include #include #include #include #i
摘要由CSDN通过智能技术生成

本题的思路:

dfs

#include<stdio.h>    
#include<iostream>    
#include<math.h>    
#include<stdlib.h>    
#include<ctype.h>    
#include<algorithm>    
#include<vector>    
#include<string.h>    
#include<queue>    
#include<stack>    
#include<set>    
#include<map>    
#include<sstream>    
#include<time.h>    
#include<utility>    
#include<malloc.h>    
#include<stdexcept>    
#include<iomanip>    
#include<iterator>    
#include<string>

using namespace std;

int n, ans;

vector<int> a[101000];
void dfs(int u, int num)
{
	ans = max(ans, num);
	for (int i = 0;i < a[u].size();i++)
	{
		int v = a[u][i];
		dfs(v, num + 1);
	}
}

int main()
{
	while (scanf("%d", &n) != EOF)
	{
		int u, v;
		for (int i = 0;i < 1010;i++)
			a[i].clear();

		for (int i = 1;i < n;i++)
		{
			scanf("%d%d", &u, &v);
			a[u].push_back(v);
		}
		ans = 0;
		for (int i = 0;i < n;i++)
		{
			dfs(i, 1);
		}
		printf("%d\n", ans);
	}
	return 0;
}

第二种写法 其实一样

#include <iostream>
#include <vector>
#include <cstdio>
#include <cstring>
#include <map>
using namespace std;
struct E {
	int v;
	E *next;
};

struct G {
	E *h[1010], e[1010], *re;
	void init(int n) {
		memset(h, 0, sizeof(h[0])*n);
		re = e;
	}
	void add(int u, int v) {
		re->v = v;
		re->next = h[u];
		h[u] = re++;
	}
	void dfs(int u, int dep, int &ans) {
		ans = max(ans, dep);
		for (E *i = h[u]; i; i = i->next)
			dfs(i->v, dep + 1, ans);
	}
} g;


int indeg[1010];

int main() {
	int n, u, v, r;
	scanf("%d", &n);
	g.init(n);
	for (int i = 1; i < n; ++i) {
		scanf("%d%d", &u, &v);
		g.add(u, v);
		++indeg[v];
	}
	int ans = 0;
	for (int i = 0; i < n; ++i)
		g.dfs(i, 1, ans);
	printf("%d", ans);
	return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值