bzoj 1103: [POI2007]大都市meg 树状数组

来源：互联网发布：淘宝圣手魔方编辑：程序博客网时间：2024/05/18 03:55

一句话题意：将边权修改为0（初始为1）和查询某个点到1的距离。

一开始预处理没有问题吧，然后修改边权只对它管辖的子树有影响。相当于区间修改。然后查分一下用树状数组即可。

AC代码如下：

#include<iostream>#include<cstdio>#include<cstring>#define N 250005using namespace std;int n,m,tot,dfsclk,c[N],bg[N],ed[N],fst[N],pnt[N],nxt[N];int read(){int x=0; char ch=getchar();while (ch<'0' || ch>'9') ch=getchar();while (ch>='0' && ch<='9'){ x=x*10+ch-'0'; ch=getchar(); }return x;}void add(int x,int y){pnt[++tot]=y; nxt[tot]=fst[x]; fst[x]=tot;}void ins(int x,int t){for (; x<=n; x+=x&-x) c[x]+=t;}int getsum(int x){int t=0; for (; x; x-=x&-x) t+=c[x]; return t;}void dfs(int x){bg[x]=++dfsclk; int p;for (p=fst[x]; p; p=nxt[p]) dfs(pnt[p]);ed[x]=dfsclk;}int main(){n=read(); int i,x,y;for (i=1; i<n; i++){x=read(); y=read(); add(x,y);}dfs(1); m=read()+n-1; char ch;for (i=2; i<=n; i++){ ins(bg[i],1); ins(ed[i]+1,-1); }while (m--){ch=getchar(); while (ch<'A' || ch>'Z') ch=getchar();if (ch=='W') printf("%d\n",getsum(bg[read()])); else{x=read(); y=read();ins(bg[y],-1); ins(ed[y]+1,1);}}return 0;}

by lych

2016.3.15

0 0