P2146 [NOI2015]软件包管理器

题目大意:1、求1条链点权为1的个数,并将路径上所有点权赋0,2、求节点u子树中点权为0的个数,并把子树点权赋1。

思路:一道裸模板,有个小技巧,我们可以在跳链时边跳边把权值赋0,这样既不会影响答案,还能少些几个函数,而这题我调了2个多小时,在一番对拍下,我终于发现,我树链剖分的模板打错了!!!(怪不得我树剖跑那么慢。在求重儿子时,nmax不能设成全局变量!!!血一般的教训emm;

代码如下:

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#include<cstdlib>
using namespace std;
const int MaxN=200000;
struct edge{
    int to,next;
}e[MaxN*2];
struct tree{
    int lb,rb,add,sum;
}tr[MaxN*4];
int n,q,last[MaxN],tot;
int dis[MaxN],fa[MaxN],son[MaxN],dfn[MaxN],top[MaxN],siz[MaxN],id,namx;
void add_edge(int x,int y){
    e[++tot].to=y;e[tot].next=last[x];last[x]=tot;
    return;
}
void dfs1(int x,int f){
    int nmax=-0x3f3f3f3f;//!!!2小时啊啊啊啊啊
    siz[x]=1;dis[x]=dis[f]+1;fa[x]=f;
    for(int i=last[x];i;i=e[i].next){
        int u=e[i].to;
        if(u==f) continue;
        dfs1(u,x);siz[x]+=siz[u];
        if(nmax<siz[u]) nmax=siz[u],son[x]=u;
    }
    return;
}
void dfs2(int x,int tp){
    dfn[x]=++id;top[x]=tp;
    if(!son[x]) return;dfs2(son[x],tp);
    for(int i=last[x];i;i=e[i].next){
        int u=e[i].to;
        if(u==fa[x]||u==son[x]) continue;//!
        dfs2(u,u);
    }
    return;
}
void downlag(int x){
    if(tr[x].add==-1) return;
    tr[x<<1].sum=tr[x].add*(tr[x<<1].rb-tr[x<<1].lb+1);tr[x<<1|1].sum=(tr[x<<1|1].rb-tr[x<<1|1].lb+1)*tr[x].add;
    tr[x<<1].add=tr[x<<1|1].add=tr[x].add;tr[x].add=-1;return;
}
void build(int now,int l,int r){
    tr[now].lb=l;tr[now].rb=r;tr[now].add=-1;
    if(l==r){tr[now].sum=1;return;}
    int mid=(l+r)>>1;
    build(now<<1,l,mid);build(now<<1|1,mid+1,r);
    tr[now].sum=tr[now<<1].sum+tr[now<<1|1].sum;
    return;
}
void add(int now,int l,int r,int w){
    if(tr[now].lb>=l&&tr[now].rb<=r){tr[now].sum=w*(tr[now].rb-tr[now].lb+1);tr[now].add=w;return;}
    downlag(now);
    int mid=(tr[now].lb+tr[now].rb)>>1;
    if(mid>=l) add(now<<1,l,r,w);if(mid<r) add(now<<1|1,l,r,w);
    tr[now].sum=tr[now<<1].sum+tr[now<<1|1].sum;
    return;
}
int query(int now,int l,int r){
    if(tr[now].lb>=l&&tr[now].rb<=r) return tr[now].sum;
    downlag(now);
    int mid=(tr[now].lb+tr[now].rb)>>1,ans=0;
    if(mid>=l) ans+=query(now<<1,l,r);if(mid<r) ans+=query(now<<1|1,l,r);
    return ans;
}
int wrk(int x){
    int ans=0;
    while(top[x]!=1){
        ans+=query(1,dfn[top[x]],dfn[x]);
        add(1,dfn[top[x]],dfn[x],0);
        x=fa[top[x]];
    }
    ans+=query(1,dfn[1],dfn[x]);add(1,dfn[1],dfn[x],0);
    return ans;
}
int main(){
    char c[10];int x,y;
    scanf("%d",&n);
    for(int i=2;i<=n;++i) scanf("%d",&x),++x,add_edge(x,i),add_edge(i,x);
    scanf("%d",&q);
    dfs1(1,0);dfs2(1,1);build(1,1,n);
    while(q--){
        scanf("%s",c);
        if(c[0]=='i'){
            scanf("%d",&x);printf("%d\n",wrk(x+1));
        }
        if(c[0]=='u'){
            scanf("%d",&x);
            printf("%d\n",siz[x+1]-query(1,dfn[x+1],dfn[x+1]+siz[x+1]-1));
            add(1,dfn[x+1],dfn[x+1]+siz[x+1]-1,1);
        }
    }
}

 

上一篇:python os模块


下一篇:P1955 [NOI2015]程序自动分析 离散化