码迷,mamicode.com
首页 > 其他好文 > 详细

BZOJ_4196&&UOJ_128_[NOI2015]_软件包管理器

时间:2016-05-10 12:50:45      阅读:199      评论:0      收藏:0      [点我收藏+]

标签:

描述


http://www.lydsy.com/JudgeOnline/problem.php?id=4196

给出一棵树,树上点权为0或1.u权值为1的条件是从根节点到u路径上的所有点权值都为1.u权值为0的条件为u的子树中所有节点权值都为0,进行如下两种操作:

1.install u.将u改为1.

2.uninstall u.将u改为0.

每次操作输出执行此操作需要改动的点的个数,并进行改动操作.

 

分析


两种操作:路径操作和子树操作.树链剖分用线段树维护.

路径操作:直接向上一直走到根节点即可.

子树操作:注意到树链剖分结束后,对于节点u,u的子树中的所有节点的tid值都大于tid[u],因为它们是在u之后访问的,并且子树中所有点的tid值是连续的,因为从u点开始,直到返回u点,访问整个子树的时候,tid值每次+1,故值时连续的.所以可以记录子树中的tid的最大值,即将整个子树转化为区间后的区间右端点(左端点为u).那进行子树操作的时候进行区间操作就可以了.比路径操作更简单.

 

注意:

1.线段树操作的时候要注意l<=r(tid[top[u]]要卸载tid[u]的前面,因为tid[top[u]]<=tid[u])(好像这个地方错了不止一次了= =).

 

ps.

1.从这次开始都用静态链表来代替vector了,讲真比vector快= =.做的时候结构体中放原本放在vector中的参量,再多放一个next,结构体数组g的大小为边的数量,再开一个大小为点的数量的head数组,head[u]用来代表从u出发的边中的第一条边在结构体数组g中编号.

#include <cstdio>
#include <algorithm>

using namespace std;

const int maxn=100000+5;

int n,q;
int dep[maxn];
int prt[maxn];
int size[maxn];
int son[maxn];
int top[maxn];
int tid[maxn];
int rev_tid[maxn];
int r[maxn];

struct Edge{
    int head[maxn];
    struct edge{
        int to,next;
        edge(){}
        edge(int a,int b):to(a),next(b){}
    }g[maxn];

    void insert(int from,int to,int id){
        g[id]=edge(to,head[from]);
        head[from]=id;
    }
}G;

struct Segment_Tree{
    struct node{
        int l,r,x,s;
    }a[4*maxn];
    
    void build_tree(int l,int r,int k){
        a[k].l=l; a[k].r=r; a[k].x=0; a[k].s=0;
        if(l==r) return;
        int mid=l+(r-l)/2;
        build_tree(l,mid,2*k); build_tree(mid+1,r,2*k+1);
    }
    
    inline void push_down(int k)
    {
        if(a[k].s!=-1){
            a[2*k].s=a[k].s; a[2*k].x=a[k].s?a[2*k].r-a[2*k].l+1:0;
            a[2*k+1].s=a[k].s; a[2*k+1].x=a[k].s?a[2*k+1].r-a[2*k+1].l+1:0;
        }
    }
    
    inline void push_up(int k)
    {
        a[k].x=a[2*k].x+a[2*k+1].x;
        a[k].s=(a[2*k].s==a[2*k+1].s&&a[2*k].s!=-1)?a[2*k].s:-1;
    }
    
    int search(int l,int r,int k,int t){
        if(a[k].l==l&&a[k].r==r){
            int ans=t?a[k].x:r-l+1-a[k].x;
            a[k].s=t?0:1;
            a[k].x=t?0:r-l+1;
            return ans;
        }
        int mid=a[k].l+(a[k].r-a[k].l)/2;
        push_down(k);
        int ans;
        if(r<=mid) ans=search(l,r,2*k,t);
        else if(l>mid) ans=search(l,r,2*k+1,t);
        else ans=search(l,mid,2*k,t)+search(mid+1,r,2*k+1,t);
        push_up(k);
        return ans;
    }
}T;
            

void find_h_e(int u,int p,int d){
    prt[u]=p; dep[u]=d; size[u]=1;
    int max_size=0;
    for(int i=G.head[u];i;i=G.g[i].next){
        int v=G.g[i].to;
        find_h_e(v,u,d+1);
        size[u]+=size[v];
        if(size[v]>max_size){
            max_size=size[v];
            son[u]=v;
        }
    }
}

void conect_h_e(int u,int anc,int &k){
    top[u]=anc; tid[u]=++k; rev_tid[k]=u;
    if(son[u]) conect_h_e(son[u],anc,k);
    for(int i=G.head[u];i;i=G.g[i].next){
        int v=G.g[i].to;
        if(v!=son[u]){
            conect_h_e(v,v,k);
        }
    }
    r[u]=rev_tid[k];
}

int get_sum1(int u){
    int sum_now=0;
    while(top[u]!=0){
        sum_now+=T.search(tid[top[u]],tid[u],1,0);    
        u=prt[top[u]];
    }
    sum_now+=T.search(1,tid[u],1,0);
    return sum_now;
}

int get_sum2(int u){return T.search(tid[u],tid[r[u]],1,1);}

void solve(){
    scanf("%d",&q);
    for(int i=1;i<=q;i++){
        char c[15]; scanf("%s",c);
        int u; scanf("%d",&u);
        if(c[0]==i) printf("%d\n",get_sum1(u));
        else printf("%d\n",get_sum2(u));
        
    }
}

void init(){
    scanf("%d",&n);
    for(int i=1;i<n;i++){
        int from;
        scanf("%d",&from);
        G.insert(from,i,i);
    }
}

int main(){
#ifndef ONLINE_JUDGE
    freopen("3.in","r",stdin);
    freopen("3.out","w",stdout);
#endif
    init();
    find_h_e(0,0,0);
    int k=0;
    conect_h_e(0,0,k);
    T.build_tree(1,n,1);
    solve();
#ifndef ONLINE_JUDGE
    fclose(stdin);
    fclose(stdout);
    system("3.out");
#endif
    return 0;
}

 

 

BZOJ_4196&&UOJ_128_[NOI2015]_软件包管理器

标签:

原文地址:http://www.cnblogs.com/Sunnie69/p/5477191.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!