Fork me on GitHub

2014-2015 Codeforces Trainings Season 2 Episode 7 G Gophers --线段树

题意: 有n个地鼠,m个CD碟,每个CD碟有一个影响范围,范围内的地鼠都会被吵到,每次有一个操作就是移动CD碟,然后求每次被影响的地鼠有多少只。

解法: 线段树做。我们只关注地鼠有没有被吵到就可以了,之前我还去把所有可能移到的位置都存下来离散化一下,然后维护也维护错了。一堆bug,真是想多了。

线段树维护两个值:  cntS[rt]表示该段被多少个区间所覆盖, NOG[rt]表示此区间内没有被影响到的地鼠有多少个。

那么我们更新到区间,然后直接pushup即可, 因为更新到区间的时候已经可以确定NOG[rt]了: 

如果 l == r : 说明是叶子节点, 那么如果cntS[rt] > 0,NOG[rt]就为0,否则为1

否则,如果cntS[rt] = 0 , 那么 NOG[rt] = NOG[2*rt]+NOG[2*rt+1],  否则说明这段被区间覆盖,NOG[rt] = 0.

代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
using namespace std;
#define N 501007

int NOG[4*N],cntS[4*N];
struct Shift{
    int L1,R1,L2,R2;
}sf[510009];
int x[510009],b[2510009];
int seg[510009][2];

void pushup(int l,int r,int rt)
{
    if(l == r)
    {
        if(cntS[rt]) NOG[rt] = 0;
        else         NOG[rt] = 1;
        return;
    }
    if(!cntS[rt]) NOG[rt] = NOG[rt<<1] + NOG[rt<<1|1];
    else          NOG[rt] = 0;
}

void build(int l,int r,int rt)
{
    cntS[rt] = 0;
    if(l == r)
    {
        NOG[rt] = 1;
        return;
    }
    int mid = (l+r)/2;
    build(l,mid,2*rt);
    build(mid+1,r,2*rt+1);
    pushup(l,r,rt);
}

void update(int l,int r,int aa,int bb,int val,int rt)
{
    if(aa <= l && bb >= r)
    {
        cntS[rt] += val;
        pushup(l,r,rt);
        return;
    }
    int mid = (l+r)/2;
    if(aa <= mid)
        update(l,mid,aa,bb,val,2*rt);
    if(bb > mid)
        update(mid+1,r,aa,bb,val,2*rt+1);
    pushup(l,r,rt);
}

int main()
{
    int n,m,d,l,i,j,val,p,r;
    while(scanf("%d%d%d%d",&n,&m,&d,&l)!=EOF)
    {
        x[1] = 0;
        for(i=2;i<=n;i++)
            scanf("%d",&x[i]);
        for(i=1;i<=m;i++)
        {
            scanf("%d",&val);
            seg[i][0] = max(0,val-l);
            seg[i][1] = min(1000000000,val+l);
        }
        for(i=1;i<=d;i++)
        {
            scanf("%d%d",&p,&r);
            sf[i].L1 = max(0,p-l);
            sf[i].R1 = min(1000000000,p+l);
            sf[i].L2 = max(0,r-l);
            sf[i].R2 = min(1000000000,r+l);
        }
        build(1,n,1);
        for(i=1;i<=m;i++)
        {
            int L = lower_bound(x+1,x+n+1,seg[i][0])-x;
            int R = upper_bound(x+1,x+n+1,seg[i][1])-x;
            R--;
            update(1,n,L,R,1,1);
        }
        printf("%d\n",n-NOG[1]);
        for(i=1;i<=d;i++)
        {
            int L1 = lower_bound(x+1,x+n+1,sf[i].L1)-x;
            int R1 = upper_bound(x+1,x+n+1,sf[i].R1)-x-1;
            int L2 = lower_bound(x+1,x+n+1,sf[i].L2)-x;
            int R2 = upper_bound(x+1,x+n+1,sf[i].R2)-x-1;
            update(1,n,L1,R1,-1,1);
            update(1,n,L2,R2,1,1);
            printf("%d\n",n-NOG[1]);
        }
    }
    return 0;
}
View Code

 

posted @ 2014-11-13 23:43  whatbeg  阅读(190)  评论(0编辑  收藏  举报