code forces E. Danil and a Part-time Job

本文提供了一种解决CodeForces 877E问题的方法,通过使用深度优先搜索(DFS序)和线段树实现区间更新及区间求和操作。代码中详细展示了如何构建线段树,并进行查询和更新等关键步骤。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

题目链接:

http://codeforces.com/contest/877/problem/E

题解:

dfs序+线段树区间更新和区间求和

代码:

#include <cmath>
#include <cstdio>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define met(a,b) memset(a,b,sizeof(a))
#define inf 0x3f3f3f3f
#define lchild root<<1,l,mid
#define rchild root<<1|1,mid+1,r
const int maxn = 2e5+10;
int L[maxn],R[maxn];
int tot;
vector<int>p[maxn];
int tree[maxn<<2],lazy[maxn<<2];
int n;

//dfs序
void dfs(int u,int fa)
{
    L[u]=++tot;
    for(int i=0;i<p[u].size();i++)
    {
        int v=p[u][i];
        if(v==fa)
            continue;
        dfs(v,u);
    }
    R[u]=tot;
}


void push_up(int root)
{
    tree[root]=tree[root<<1]+tree[root<<1|1];
}

void push_down(int root,int l,int r)
{
    if(lazy[root])
    {
        int mid=(l+r)>>1;
        tree[root<<1]=mid-l+1-tree[root<<1];
        lazy[root<<1]^=1;
        tree[root<<1|1]=r-(mid+1)+1-tree[root<<1|1];
        lazy[root<<1|1]^=1;
        lazy[root]=0;
    }
}

void build(int root=1,int l=1,int r=n)
{
    if(l==r)
    {
        //        tree[root]=room[L[l]];
        tree[root]=0;
        return;
    }
    int mid=(l+r)>>1;
    build(lchild),build(rchild);
    push_up(root);
}

void updata(int L,int R,int root=1,int l=1,int r=n)
{
    if(L<=l&&r<=R)
    {
        tree[root]=(r-l+1)-tree[root];
        lazy[root]^=1;
        return;
    }
    if(lazy[root])
        push_down(root,l,r);
    int mid=(l+r)>>1;
    if(L<=mid)
        updata(L,R,lchild);
    if(R>mid)
        updata(L,R,rchild);
    push_up(root);
}

int query(int L,int R,int root=1,int l=1,int r=n)
{
    if(L<=l&&r<=R)
        return tree[root];
    if(lazy[root])
        push_down(root,l,r);
    int mid=(l+r)>>1,res=0;
    if(L<=mid)
        res+=query(L,R,lchild);
    if(R>mid)
        res+=query(L,R,rchild);
    return res;
}

int main()
{
    scanf("%d",&n);
    for(int i=2;i<=n;i++)
    {
        int x;
        scanf("%d",&x);
        p[x].push_back(i);
        p[i].push_back(x);
    }
    met(tree,0),met(lazy,0);
    met(L,0),met(R,0);
    tot=0;
    dfs(1,-1);
    build();
    for(int i=1;i<=n;i++)
    {
        int x;
        scanf("%d",&x);
        if(x==1)
            updata(L[i],L[i]);
    }
    int m;
    scanf("%d",&m);
    while(m--)
    {
        string s;
        int num;
        cin>>s>>num;
        if(s=="get")
            printf("%d\n",query(L[num],R[num]));
        else
            updata(L[num],R[num]);
    }

}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值