这题有点意思啊。。
考虑到要修改的除自己外还有儿子,孙子,父亲,祖父,别忘了兄弟。。
父亲祖父还比较好搞。。然而兄弟儿子什么就有一大堆了,所以需要一种方法映射到区间上才能整段修改。。
考虑到兄弟之间深度相同,儿子孙子类似。。所以可以使用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
空间限制:C/C++ 262144K,其他语言524288K
64bit IO Format: %lld
题目描述
小H正在玩一个战略类游戏,她可以操纵己方的飞机对敌国的N座城市(编号为1~N)进行轰炸
敌国的城市形成了一棵树,小H会依次进行Q次轰炸,每次会选择一个城市A进行轰炸,和这座城市距离不超过2的城市都会受损(这里距离的定义是两点最短路径上的边数),轰炸结束后,小H还想知道当前城市A受损的次数
作为游戏的开发者之一,你有义务回答小H的问题
敌国的城市形成了一棵树,小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