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);        }    }}