NOI2015 软件包管理器 树剖+线段树

题目背景是一个安装与卸载软件的系统，软件之间关系是一棵树，以0为根节点。当你安装一个软件时需要先安装它的所有祖先，卸载时则要先卸载整个子树。对每一次操作询问改变状态的软件数目。

思路：dfs序做树剖，用线段树维护数据。

#include<iostream>
#include<vector>
#include<stdio.h>
#define ls (id*2)
#define rs (id*2+1)
using namespace std;
vector<int> lin[100005];
int in[100005],size[100005],fa[100005],top[100005],n,num,ans;
struct node
{
int val;
int lazy;
}t[100005*4];
int abs(int a)
{
return a>=0?a:-a;
}
void push_down(int id,int l,int mid,int r)
{
if (t[id].lazy==-1)return;
t[ls].lazy=t[id].lazy;
t[rs].lazy=t[id].lazy;
t[ls].val=(mid-l+1)*t[id].lazy;
t[rs].val=(r-mid)*t[id].lazy;
t[id].lazy=-1;
}
void push_up(int id)
{
t[id].val=t[ls].val+t[rs].val;
}
void build(int id,int l,int r)
{
if(l==r)
{
t[id].val=0;
t[id].lazy=-1;//改成-1！切记！
return;
}
int mid=(l+r)/2;
build(ls,l,mid);
build(rs,mid+1,r);
push_up(id);
}
void insert(int id,int l,int r,int L,int R,int V)
{
if(r<L||l>R) return;
if(r<=R && l>=L)
{
ans+=abs(V*(r-l+1)-t[id].val);
t[id].val=V*(r-l+1);
t[id].lazy=V;
return;
}
int mid=(l+r)/2;
push_down(id,l,mid,r);
insert(ls,l,mid,L,R,V);
insert(rs,mid+1,r,L,R,V);
push_up(id);
}
void dfs1(int x,int pre)
{
in[x]=++num;
fa[x]=pre;
for (int i=0;i<lin[x].size();i++)
{
int nex=lin[x][i];
if (nex==pre)   continue;
dfs1(nex,x);
}
size[x]=num-in[x]+1;//必须+1，否则dfs2中判断叶子节点很麻烦。
}
void dfs2(int x,int t)
{
in[x]=++num;
top[x]=t;
if (size[x]==1) return;//叶子节点直接返回，否则会死循环(不能用size[x]=0做!)
int msize=0,heavy=0;//都要赋初始值！
for (int i=0;i<lin[x].size();i++)
{
int nex=lin[x][i];
if (nex==fa[x]) continue;
if (size[nex]>msize)
{
msize=size[nex];
heavy=nex;
}
}
dfs2(heavy,t);//先走重儿子，其他随意
for (int i=0;i<lin[x].size();i++)
{
int nex=lin[x][i];
if (nex==fa[x] || nex==heavy) continue;
dfs2(nex,nex);
}
}
int main()
{
cin>>n;
for (int i=1;i<n;i++)
{
int a;
cin>>a;//编号都+1，防止爆线段树
lin[a+1].push_back(i+1);
lin[i+1].push_back(a+1);
}
num=0;  dfs1(1,0);
num=0;  dfs2(1,1);
int q;
cin>>q;
char s[15];
while (q--)
{
int a;
scanf("%s",s);//一定不要用cin！！！！T了半个小时...
scanf("%d",&a);
a++;
if (s[0]=='i')
{
ans=0;
while (a!=0)
{
insert(1,1,n,in[top[a]],in[a],1);
a=fa[top[a]];
}
printf("%d\n",ans);
}
if (s[0]=='u')
{
ans=0;
insert(1,1,n,in[a],in[a]+size[a]-1,0);
printf("%d\n",ans);
}
}
}