【树链剖分（区间线段树）】BZOJ4196-[NOI2015]软件包管理

【题目大意】

如果软件包A依赖软件包B，那么安装软件包A以前，必须先安装软件包B。同时，如果想要卸载软件包B，则必须卸载软件包A。而且，由于你之前的工作，除0号软件包以外，在你的管理器当中的软件包都会依赖一个且仅一个软件包，而0号软件包不依赖任何一个软件包。依赖关系不存在环。求出在安装和卸载某个软件包时，实际上会改变多少个软件包的安装状态（即安装操作会安装多少个未安装的软件包，或卸载操作会卸载多少个已安装的软件包。（注意，安装一个已安装的软件包，或卸载一个未安装的软件包，都不会改变任何软件包的安装状态，即在此情况下，改变安装状态的软件包数为0）

【思路】

裸的树剖...然而我发现我写错了区间覆盖的线段树，要设置两个标记，一个记录该区间是否需要修改，另一个记录该区间覆盖的值。

话说BZOJ要用printf否则会RE，我怎么不长记性呢………

  1 #include<iostream>
  2 #include<cstdio>
  3 #include<cstring>
  4 #include<algorithm>
  5 #include<vector>
  6 #define lson l,m,rt<<1
  7 #define rson m+1,r,rt<<1|1
  8 using namespace std;
  9 const int MAXN=100000+50;
 10 const int rt=0;
 11 vector<int> E[MAXN];
 12 int n;
 13 int fa[MAXN],dep[MAXN],hson[MAXN],size[MAXN];
 14 int cnt=0,top[MAXN],pos[MAXN];
 15 int sum[MAXN<<3],add[MAXN<<3],change[MAXN<<3];
 16
 17 //树链剖分部分
 18 void addedge(int u,int v)
 19 {
 20     E[u].push_back(v);
 21 }
 22
 23 void dfs1(int u,int father,int depth)
 24 {
 25     fa[u]=father;
 26     dep[u]=depth;
 27     size[u]=1;
 28     hson[u]=-1;
 29     for (int i=0;i<E[u].size();i++)
 30     {
 31         int to=E[u][i];
 32         dfs1(to,u,depth+1);
 33         size[u]+=size[to];
 34         if (hson[u]==-1 || size[to]>size[hson[u]]) hson[u]=to;
 35     }
 36 }
 37
 38 void dfs2(int u,int t)
 39 {
 40     pos[u]=++cnt;
 41     top[u]=t;
 42     if (hson[u]!=-1) dfs2(hson[u],t);
 43     for (int i=0;i<E[u].size();i++)
 44     {
 45         int to=E[u][i];
 46         if (to!=hson[u]) dfs2(to,to);
 47     }
 48 }
 49
 50 //线段树部分
 51 void build()
 52 {
 53     memset(sum,0,sizeof(sum));
 54     memset(add,0,sizeof(add));
 55 }
 56
 57 void pushup(int rt)
 58 {
 59     sum[rt]=sum[rt<<1]+sum[rt<<1|1];
 60 }
 61
 62 void pushdown(int rt,int m)
 63 {
 64     if (change[rt])
 65     {
 66         change[rt<<1]=change[rt<<1|1]=1;
 67         add[rt<<1]=add[rt];
 68         add[rt<<1|1]=add[rt];
 69         sum[rt<<1]=add[rt]*(m-(m>>1));
 70         sum[rt<<1|1]=add[rt]*(m>>1);
 71         add[rt]=change[rt]=0;
 72     }
 73 }
 74
 75 int query_sum(int L,int R,int l,int r,int rt)
 76 {
 77     if (L<=l && r<=R) return sum[rt];
 78     pushdown(rt,r-l+1);
 79     int m=(l+r)>>1;
 80     int ret=0;
 81     if (m>=L) ret+=query_sum(L,R,lson);
 82     if (m<R) ret+=query_sum(L,R,rson);
 83     pushup(rt);
 84     return ret;
 85 }
 86
 87 void modify(int L,int R,int l,int r,int rt,int x)
 88 {
 89     if (L<=l && r<=R)
 90     {
 91         change[rt]=1;
 92         add[rt]=x;
 93         sum[rt]=(r-l+1)*x;
 94         return;
 95     }
 96     pushdown(rt,r-l+1);
 97     int m=(l+r)>>1;
 98     if (m>=L) modify(L,R,lson,x);
 99     if (m<R) modify(L,R,rson,x);
100     pushup(rt);
101 }
102
103 //树链剖分查询部分
104 int install(int x,int y)
105 {
106     int ret=dep[x],f1=top[x],f2=top[y];
107     while (f1!=f2)
108     {
109         ret-=query_sum(pos[f1],pos[x],1,n,1);
110         modify(pos[f1],pos[x],1,n,1,1);
111         x=fa[f1];
112         f1=top[x];
113     }
114     ret-=query_sum(pos[y],pos[x],1,n,1);
115     modify(pos[y],pos[x],1,n,1,1);
116     return (ret);
117 }
118
119 int uninstall(int x)
120 {
121     int ret=query_sum(pos[x],pos[x]+size[x]-1,1,n,1);
122     modify(pos[x],pos[x]+size[x]-1,1,n,1,0);
123     return ret;
124 }
125
126
127 //读入部分
128 void init()
129 {
130     scanf("%d",&n);
131     for (int i=1;i<n;i++)
132     {
133         int tmp;
134         scanf("%d",&tmp);
135         addedge(tmp,i);
136     }
137     dfs1(0,0,1);
138     dfs2(0,0);
139 }
140
141 void get_ans()
142 {
143     memset(sum,0,sizeof(sum));
144     memset(change,0,sizeof(change));
145     memset(add,0,sizeof(add));
146     int q;
147     scanf("%d",&q);
148     for (int i=0;i<q;i++)
149     {
150         char str[25];
151         int x;
152         scanf("%s%d",str,&x);
153         if (str[0]==‘i‘) printf("%d\n",install(x,rt));
154             else if (str[0]==‘u‘) printf("%d\n",uninstall(x));
155     }
156 }
157
158 int main()
159 {
160     init();
161     build();
162     get_ans();
163     return 0;
164 }