wannafly10(bfs序+BIT)-CFANZ编程社区

这题有点意思啊。。

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

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

考虑到兄弟之间深度相同，儿子孙子类似。。所以可以使用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的问题