计蒜客-青出于蓝胜于蓝 dfs+树状数组

 题目描述: 

武当派一共有 n人,门派内 n 人按照武功高低进行排名,武功最高的人排名第 1,次高的人排名第 2,... 武功最低的人排名第 n。现在我们用武功的排名来给每个人标号,除了祖师爷,每个人都有一个师父,每个人可能有多个徒弟。

我们知道,武当派人才辈出,连祖师爷的武功都只能排行到 p。也就是说徒弟的武功是可能超过师父的,所谓的青出于蓝胜于蓝。

请你帮忙计算每个人的所有子弟(包括徒弟的徒弟,徒弟的徒弟的徒弟....)中,有多少人的武功超过了他自己。

输入格式
输入第一行两个整数 n,p(1≤n≤100000,1≤p≤n)n, p(1 <= n <= 100000, 1 <= p <= n)n,p(1≤n≤100000,1≤p≤n)。

接下来 n−1n-1n−1 行,每行输入两个整数 u,v(1≤u,v≤n)u, v(1 <= u, v <= n)u,v(1≤u,v≤n),表示 u 和 v 之间存在师徒关系。

输出格式
输出一行 n 个整数,第 i 个整数表示武功排行为 i 的人的子弟有多少人超过了他。

行末不要输出多余的空格。
样例输入
10 5
5 3
5 8
3 4
3 1
2 1
6 7
8 7
9 8
8 10
样例输出
0 0 2 0 4 0 1 2 0 0

题解参考https://blog.csdn.net/wjhshuai/article/details/79546381

 题解:

如下图所示,对一颗树进行dfs,先搜索自己,然后是所有孩子节点。(以二叉树为例)

节点旁边的编号表示搜索的时序,从根节点开始搜,第一次搜的是1节点,然后搜2、4、8、5.....(图上只标注了一部分)

对于节点2来说,其子节点为搜索时序3,4,5的节点。

所以dfs的时候,每搜索到一个节点时,使用数组l[]标记其搜索时序,将子节点搜索完后,r[]数组标记回溯时的时序,这样就可以判断每个节点子节点的搜索时序范围。比如对2来说,dfs时序为2,搜完子节点回溯时时序为5,所以3-5时序的就是其子节点。

题目要求每一个节点子节点比自己小的个数,所以就可以将节点从1-n遍历,每次计算区间l[i]-r[i]的区间和sum(sum表示区间内比当前节点小的节点数),l[i]-r[i]区间就确保一定是i节点的子节点,因为要统计的节点一定比i小,所以每次遍历一个节点时,就将其时序l[i]的地方加1,这样后面求区间和时如果包含了之前加1的部分,那就可以统计进去。。(说得有点绕,具体看代码)

 

#include <bits/stdc++.h> 
#define MAXN 100005
using namespace std;
// 从1开始计数 
int C[MAXN]; // 树状数组

int head[MAXN];// i号节点的第一个子节点在结构体数组的下标为head[i]
int l[MAXN],r[MAXN];// l数组:记录dfs搜到i点时的次序,r数组:记录dfs回溯回i点时的时序 
int n,f;// 输入
int times;// 深度搜索的次序 
struct Node {
	int val,next;// 节点编号,下一个兄弟节点在node数组的下标	
}node[MAXN]; 
int num;// node数组元素数目 


// 取x最低位的1 
int lowbit(int x)  {
	return x & -x;
}

// pos下标处增加delta
void update(int pos,int delta) {	
	for(int i = pos;i <= n;i += lowbit(i)) {
		C[i] += delta;		
	}	
}

// 区间查询,查询前x项和 
int query(int x)  {
	int cnt = 0;
	for(int i = x;i >= 1;i -= lowbit(i)) {
		cnt += C[i];
	}	
	
	return cnt;
}

// 往图里加点
void add(int u,int v)  {
	node[num].next = head[u];
	node[num].val = v;	
	head[u] = num++;	
}

// 深度优先搜索,先搜完孩子节点,再搜本节点 
// cur:当前节点编号,father: 父节点编号(用来避免重复搜索) 
void dfs(int cur,int father) {
	l[cur] = ++times;
	// 遍历所有子节点 
	for(int i = head[cur];i != -1;i = node[i].next) {
		int son = node[i].val;// 子节点编号 
		// 深搜
		if(son != father) 
			dfs(son,cur);
	} 
	
	// 回溯时标记
	r[cur] = times;
}

int main() {
	scanf("%d%d",&n,&f);
	for (int i = 0;i <= n;i++) head[i] = -1;// 初始化	 			
	int u,v;
	for(int i = 1;i < n;i++) {
		scanf("%d%d",&u,&v);// u是v的师父 
		add(u,v);
		add(v,u);
	}
	dfs(f,-1);// 从根节点开始搜索 
	// 从第一名到最后一名遍历武林高手 
	for(int i = 1;i <= n;i++) {
		printf("%d\n",query(r[i])-query(l[i]));
		update(l[i],1);
	}
	
			
	return 0;
}

 

没有更多推荐了,返回首页

私密
私密原因:
请选择设置私密原因
  • 广告
  • 抄袭
  • 版权
  • 政治
  • 色情
  • 无意义
  • 其他
其他原因:
120
出错啦
系统繁忙,请稍后再试