wannafly10(bfs序+BIT)

这题有点意思啊。。

考虑到要修改的除自己外还有儿子，孙子，父亲，祖父，别忘了兄弟。。

父亲祖父还比较好搞。。然而兄弟儿子什么就有一大堆了，所以需要一种方法映射到区间上才能整段修改。。

考虑到兄弟之间深度相同，儿子孙子类似。。所以可以使用bfs序，这样同龄人之间就可以堆一块修改了。。

其实还用不着线段树，因为是单点询问，所以差分后直接上BIT就行。。。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#include<map>
#include<set>
#define inc(i,l,r) for(int i=l;i<=r;i++)
#define dec(i,l,r) for(int i=l;i>=r;i--)
#define link(x) for(edge *j=h[x];j;j=j->next)
#define mem(a) memset(a,sizeof(a), 0)
#define inf 1e9
#define succ(x) (1<<x)
#define lowbit(x) (x&(-x))
#define NM 750005
#define nm 4000000
using namespace std;
int read(){
	int x=0,f=1;char ch=getchar();
	while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
	while(isdigit(ch))x=x*10+ch-'0',ch=getchar();
	return f*x;
}



struct edge{
	int t;edge*next;
}e[nm],*h[NM],*o=e;
void add(int x,int y){
	o->t=y;o->next=h[x];h[x]=o++;
}
int n,m,a[NM],c[NM],_x,_y,_t,in[NM],out[NM],f[NM],id[NM],tot;
queue<int>q;
void mod(int x,int t){
	for(;x<=n;x+=lowbit(x))c[x]+=t;
}
int sum(int x,int s=0){
	for(;x;x-=lowbit(x))s+=c[x];return s;
}

void bfs(){
	f[1]=id[1]=tot=1;q.push(1);
	while(!q.empty()){
		int t=q.front();q.pop();in[id[t]]=tot+1;
		link(t)if(!f[j->t]){
			f[j->t]=t;q.push(j->t);id[j->t]=++tot;
		}
		out[id[t]]=tot;
	}
}


int main(){
//	freopen("data.in","r",stdin);
	n=read();m=read();
	inc(i,1,n-1){
		_x=read();_y=read();add(_x,_y);add(_y,_x);
	}
	bfs();
//	inc(i,1,n)printf("%d ",out[i]);putchar('\n');
	while(m--){
		_t=_x=read();
		if(_t==1){
			_y=1;mod(_y,1);mod(_y+1,-1);
		}else{
			_y=id[f[_x]];
			mod(_y,1);mod(_y+1,-1);
			_x=in[_y];_y=out[_y];
			mod(_x,1);mod(_y+1,-1);
			if(f[_t]!=1){
				_y=id[f[f[_t]]];
				mod(_y,1);mod(_y+1,-1);
			}
		}
		_x=in[id[_t]];_y=out[id[_t]];
		if(_x<=_y){
			mod(_x,1);mod(_y+1,-1);
			_x=in[_x];_y=out[_y];
			if(_x<=_y){
				mod(_x,1);mod(_y+1,-1);
			}
		}
		printf("%d\n",sum(id[_t]));
	}
	return 0;
}

链接：https://www.nowcoder.com/acm/contest/72/C
来源：牛客网

C-小H和游戏

时间限制：C/C++ 2秒，其他语言4秒
空间限制：C/C++ 262144K，其他语言524288K
64bit IO Format: %lld

题目描述

    小H正在玩一个战略类游戏，她可以操纵己方的飞机对敌国的N座城市(编号为1~N)进行轰炸
    敌国的城市形成了一棵树，小H会依次进行Q次轰炸，每次会选择一个城市A进行轰炸，和这座城市距离不超过2的城市都会受损(这里距离的定义是两点最短路径上的边数)，轰炸结束后，小H还想知道当前城市A受损的次数
    作为游戏的开发者之一，你有义务回答小H的问题

输入描述:

第1行，两个整数N(1≤N≤750000)、Q(1≤Q≤750000)
第2~N行，每行两个整数表示树上的一条边
第N+1~N+Q行，每行一个整数，表示小H这次轰炸的城市

输出描述:

输出Q行，每行一个整数表示这一次轰炸的城市在此次轰炸后共计受损几次

示例1

输入

4 4 
1 2 
2 3 
3 4 
1 
2 
3 
4

输出

1 
2 
3 
3