cogs——1008. 贪婪大陆（清华巨佬代码）——树状数组

Posted 2020-09-16

1008. 贪婪大陆

★★   输入文件：greedisland.in   输出文件：greedisland.out   简单对比
时间限制：1 s   内存限制：128 MB

试题四:贪婪大陆 

【题目描述】 

 

面对蚂蚁们的疯狂进攻，小FF的Tower defense宣告失败……人类被蚂蚁们逼到了Greed Island上的一个海湾。现在，小FF的后方是一望无际的大海，前方是变异了的超级蚂蚁。 小FF还有大好前程，他可不想命丧于此， 于是他派遣手下最后一批改造SCV布置地雷以阻挡蚂蚁们的进攻。 

 

小FF最后一道防线是一条长度为N的战壕， 小FF拥有无数多种地雷，而SCV每次可以在[ L , R ]区间埋放同一种不同于之前已经埋放的地雷。 由于情况已经十万火急，小FF在某些时候可能会询问你在[ L‘ , R‘] 区间内有多少种不同的地雷， 他希望你能尽快的给予答复。 

 

【输入格式】 

 

第一行为两个整数n和m； n表示防线长度， m表示SCV布雷次数及小FF询问的次数总和。 

 

接下来有m行， 每行三个整数Q，L , R； 若Q=1 则表示SCV在[ L , R ]这段区间布上一种地雷， 若Q=2则表示小FF询问当前[ L , R ]区间总共有多少种地雷。 

 

【输出格式】 

 

对于小FF的每次询问，输出一个答案（单独一行），表示当前区间地雷总数。 

【输入样例】 

 

5 4 

 

1 1 3 

 

2 2 5 

 

1 2 4 

 

2 3 5 

 

【输出样例】 

 

1 

 

2 

 

【数据范围】 

 

对于30%的数据： 0<=n, m<=1000; 

 

对于100%的数据： 0<=n, m<=10^5.

代码:

one

#include<cstdio>

const int MAXN(100010);

int n,m,tot;
int t[MAXN<<2][2];

void insert(int x,int k,int left=1,int right=n,int p=1){
    if(left==right){
        t[p][k]++;
        return;
    }
    int mid=(left+right)>>1;
    if(x<=mid)insert(x,k,left,mid,p<<1);
    else insert(x,k,mid+1,right,p<<1|1);
    t[p][0]=t[p<<1][0]+t[p<<1|1][0];
    t[p][1]=t[p<<1][1]+t[p<<1|1][1];
}
int query(int x,int y,int k,int left=1,int right=n,int p=1){
    if(x<=left&&right<=y)
        return t[p][k];
    int mid=(left+right)>>1,ans=0;
    if(x<=mid)ans=ans+query(x,y,k,left,mid,p<<1);
    if(y>mid)ans=ans+query(x,y,k,mid+1,right,p<<1|1);
    return ans;
}

int main(){
    freopen("greedisland.in","r",stdin);
    freopen("greedisland.out","w",stdout);
    scanf("%d%d",&n,&m);
    for(int q=1;q<=m;q++){
        int ty,x,y;
        scanf("%d%d%d",&ty,&x,&y);
        if(ty==1){
            insert(x,0);
            insert(y,1);
            tot++;
        }
        else
            printf("%d\n",tot-query(1,x-1,1)-query(y+1,n,0));
    }
    return 0;
}

two

#include<cstdio>

const int MAXN(100010);

int n,m,tot;
int t[MAXN<<2],tag[MAXN<<2],mark[MAXN<<2];

void increase(int p,int k){
    t[p]+=k;
    tag[p]+=k;
    mark[p]+=k;
}

void push_down(int p){
    int& d=tag[p];
    increase(p<<1,d);
    increase(p<<1|1,d);
    d=0;
}

void insert(int x,int y,int left=1,int right=n,int p=1){
    if(x<=left&&right<=y){
        increase(p,1);
        return;
    }
    push_down(p);
    int mid=(left+right)>>1;
    if(x<=mid)insert(x,y,left,mid,p<<1);
    if(y>mid) insert(x,y,mid+1,right,p<<1|1);
    t[p]++;
    if(x<=mid&&y>mid)mark[p]++;
}
int query(int x,int y,int left=1,int right=n,int p=1){
    if(x<=left&&right<=y)
        return t[p];
    push_down(p);
    int mid=(left+right)>>1,ans=0;
    if(x<=mid)ans=ans+query(x,y,left,mid,p<<1);
    if(y>mid)ans=ans+query(x,y,mid+1,right,p<<1|1);
    if(x<=mid&&y>mid)ans-=mark[p];
    return ans;
}

int main(){
    freopen("greedisland.in","r",stdin);
    freopen("greedisland.out","w",stdout);
    scanf("%d%d",&n,&m);
    for(int q=1;q<=m;q++){
        int ty,x,y;
        scanf("%d%d%d",&ty,&x,&y);
        if(ty==1)
            insert(x,y);
        else
            printf("%d\n",query(x,y));
    }
    return 0;
}