码迷,mamicode.com
首页 > 编程语言 > 详细

POJ--3321(dfs序+树状数组+vector防超时写法)

时间:2020-06-09 18:13:12      阅读:66      评论:0      收藏:0      [点我收藏+]

标签:amp   src   iostream   十分   ring   ret   target   数组   main   

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

题意:

树上有n个位置,初始每个位置都有苹果。以1号为总根

n-1条分叉边

m条询问:

Q:id   以id为根的子树有几个苹果

C:id    id处如果有苹果,摘掉,否则放上一个苹果。

解析:

假设输入的是这么一棵树:

技术图片

看数据,肯定要前缀和的思想,但是:

假设要询问4号为根的子树,我们要看的是4+8+9,这些序号很难通过前缀和来维护,因为输入的边并不是按顺序输的。

所以,考虑能不能把这些序号重新编排一下,使其能通过前缀和来求?

这里就引入了dfs序

st[],en[],来记录 i 在dfs序里的开始和结束位置

比如以此图为例:

dfs的遍历顺序:1     2   5    6   3     7   4     8    9

(st[i],en[i]):  (1,9)  (2,4)   (3,3)  (4,4)  (5,6)   (6,6)  (7,9)   (8,8)  (9,9)

很好理解,结合树,比如4:(7,9)它在dfs序中以第7个开始,第9个结束。

所以有:

void dfs(int n)
{
    num++;
    st[n]=num;
    for(int i=0;i<vv[n].size();i++)
    {
        dfs(vv[n][i]);
    }
    en[n]=num;
}

剩下的就交给树状数组维护前缀和了,查询一个点,只需要:getsum(en[id])-getsum(st[id]-1)

vector防超时的写法:vector<vector<int> > vv(maxn);

总的代码:

#include <cstdio>
#include<cstring>
#include<map>
#include<vector>
#include<iostream>
using namespace std;
const int maxn =1e5+10;
int st[maxn],en[maxn],vis[maxn],c[maxn];
vector<vector<int> > vv(maxn);
int num=0;
int lowbit(int x)
{
    return x&(-x);
}
void update(int id,int x)
{
    for(int i=id;i<maxn;i+=lowbit(i))
    {
        c[i]+=x;
    }
    return ;
}
int getsum(int id)
{
    int sum=0;
    for(int i=id;i>0;i-=lowbit(i))
        sum+=c[i];
    return sum;
}
void dfs(int n)
{
    num++;
    st[n]=num;
    for(int i=0;i<vv[n].size();i++)
    {
        dfs(vv[n][i]);
    }
    en[n]=num;
}
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n-1;i++)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        vv[a].push_back(b);
    }
    
    dfs(1);
    for(int i=1;i<=n;i++)
    {
        update(i,1);
        vis[i]=1;
    }
    int m;
    scanf("%d",&m);
    while(m--)
    {    
        char ch[12];
        scanf("%s",ch);
        if(ch[0]==Q)
        {
            int id;
            scanf("%d",&id);        
            cout<<getsum(en[id])-getsum(st[id]-1)<<endl;
        }
        else
        {
            int id;
            scanf("%d",&id);
            if(vis[id])
            {
                vis[id]=0;
                update(st[id],-1);
            }
            else
            {
                vis[id]=1;
                update(st[id],1);
            }
        }
    }
}

参考自:https://www.cnblogs.com/FrankChen831X/p/10800106.html  十分感谢!

POJ--3321(dfs序+树状数组+vector防超时写法)

标签:amp   src   iostream   十分   ring   ret   target   数组   main   

原文地址:https://www.cnblogs.com/liyexin/p/13074188.html

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