【CODE[VS]】1228 苹果树 树状数组

来源:互联网 发布:质量好的衣服淘宝店铺 编辑:程序博客网 时间:2024/05/29 11:06

题目传送门

为什么我拿到这题的第一反应是树剖啊……我到底在想些什么啊

然而这题被分类在树状数组里,于是我就把想法往树状数组上靠,依然没有任何思路

于是我放弃了树状数组的解法,想用树剖暴艹这题。

却突然惊讶的发现:树剖你妹啊,这题不就是先序遍历后树状数组维护节点权值吗?

被一道傻逼题浪费了一个小时的我表示有一句mmp不知当不当讲……

附上AC代码:

#include <cstdio>#include <cctype>using namespace std;const int N=1e5+10;struct side{    int to,nt;}s[N<<1];int n,h[N],num,wz[N],sz[N],size,q,a[N],t[N],x,y;inline char nc(void){    static char ch[100010],*p1=ch,*p2=ch;    return p1==p2&&(p2=(p1=ch)+fread(ch,1,100010,stdin),p1==p2)?EOF:*p1++;}inline void read(int &a){    static char c=nc();int f=1;    for (;!isdigit(c);c=nc()) if (c=='-') f=-1;    for (a=0;isdigit(c);a=(a<<3)+(a<<1)+c-'0',c=nc());    return (void)(a*=f);}inline void add(int x,int y){    s[++num]=(side){y,h[x]},h[x]=num;    s[++num]=(side){x,h[y]},h[y]=num;}inline void so(int x,int fa){    wz[x]=++size,sz[x]=1;    for (int i=h[x]; i; i=s[i].nt)        if (s[i].to!=fa) so(s[i].to,x),sz[x]+=sz[s[i].to];    return;}#define lowbit(x) ((x)&(-x))inline void updata(int x,int w){for (int i=x; i<=n; i+=lowbit(i)) t[i]+=w; return;}inline int query(int x){int ret=0; for (int i=x; i; i-=lowbit(i)) ret+=t[i]; return ret;}int main(void){    read(n);    for (int i=1; i<n; ++i) read(x),read(y),add(x,y);    so(1,0),read(q);    for (int i=1; i<=n; ++i) a[i]=1,updata(wz[i],1);    while (q--){        char c=nc();while (c!='C'&&c!='Q') c=nc();read(x);        if (c=='C')            if (a[x]) a[x]=0,updata(wz[x],-1);            else a[x]=1,updata(wz[x],1);        else printf("%d\n",query(wz[x]+sz[x]-1)-query(wz[x]-1));    }    return 0;}
阅读全文
0 0