poj之旅:3321 Apple Tree

来源:互联网 发布:渲染软件 编辑:程序博客网 时间:2024/06/08 08:48

http://poj.org/problem?id=3321

这道题涉及到怎么把多叉树的查询转化为树状数组的前缀和查询
思想:通过某种办法将x转化为一个区间[L,R];
一个连续的区间可以代表一棵子树

代码块语法遵循标准markdown代码,例如:

#include <iostream>#include <cstdio>#include <vector>using namespace std;/*id[i]编号为i的树杈对应的后序遍历次序size[i]编号为i的树杈对应的子树规模转化为BIT问题,i->[L,R]:R=id[i],L=R-size[i]+1x[id[i]]代表是否有果前缀和sum(id[i])代表所有后序遍历在id[i]之前的减少的苹果数量*/int id[100010]={0},size[100010]={0};int BIT[100010]={0},x[100010]={0};typedef vector<int> ve;void postOrder(int subroot,vector<ve> &adj,bool book[],int& subsize,int &order){    //cout<<"traverse subroot "<<subroot<<endl;    //subsize当前子树规模    bool leave=1;//当前树是否为叶子    for(int i=0;i<adj[subroot].size();i++){        if(!book[adj[subroot][i]]){            leave=0;            book[adj[subroot][i]]=1;            int tempsize=0;//子树返回的规模            postOrder(adj[subroot][i],adj,book,tempsize,order);            //cout<<"tempsize "<<adj[subroot][i]<<": "<<tempsize<<endl;            subsize+=tempsize;        }    }    if (leave)    {        subsize=1;        size[subroot]=1;    }    else{        subsize++;        size[subroot]=subsize;    }    id[subroot]=++order;    //cout<<"traverse subroot "<<subroot<<" end"<<endl;}void add(int,int,int);int Sum(int);int main(){    int n;    cin>>n;    //邻接表用于进行后序遍历    //vector<vector<int>> adj(n+1);    vector<ve> adj(100002);     for(int i=1;i<n;i++){        int f1,f2;        scanf("%d%d",&f1,&f2);        adj[f1].push_back(f2);        adj[f2].push_back(f1);    }    //后序遍历    bool book[100010]={0};    book[1]=1;    int tsize=0,order=0;    postOrder(1,adj,book,tsize,order);    /*    for(int i=1;i<=n;i++){        cout<<"id "<<i<<": "<<id[i]<<endl;        cout<<"size "<<i<<": "<<size[i]<<endl;    }    */    int m,fork;    char ins;    cin>>m;    while(m--){        cin>>ins;        cin>>fork;        int postid=id[fork];        if(ins=='C'){            if(x[postid]==0){                x[postid]=1;                add(postid,1,n);            }else{                x[postid]=0;                add(postid,-1,n);            }            /*            for(int i=1;i<=n;i++){            cout<<"x["<<i<<"] "<<x[i]<<endl;            cout<<"BIT["<<i<<"] "<<BIT[i]<<endl;            }            */        }else{            int sumR=Sum(postid);            //cout<<"sumR: "<<sumR<<endl;            int sumL=Sum(postid-size[fork]);            //cout<<"sumL: "<<sumL<<endl;            cout<<size[fork]-sumR+sumL<<endl;        }    }}void add(int i,int num,int n){    while(i<=n){        BIT[i]+=num;        i=i+(-i&i);    }}int Sum(int i){    int res=0;    while(i>0){        res+=BIT[i];        i=i-(-i&i);    }    return res;}

不懂的地方:

有一个地方会卡超时:

vector<vector<int>> adj(100010);

这样就会超时,不知道为什么,有知道为什么的人可以解答一下。

0 0