P2184 贪婪大陆 线段树 + 区间覆盖

题目链接

题意:每次给定一个区间,有两个操作
(1)区间炸弹的种类数 + 1
(2)查询区间的不同炸弹种类数

仔细想想可以想到,可以转换成求一个区间内不同区间覆盖的最大次数,以前做过很多类似的题,大体思想就是把 l 的位置 + 1 ,r + 1 的位置 - 1 ,这个题也是类似,用线段树维护区间左端起点的值和右端起点的值,对于每次添加的区间,l 和 r 的位置都 + 1 。每次查询的区间,用 lsum ( 1 ~ r ) - rsum ( 1 ~ ( l -1 ) ) ,也就是起点减去终点,就可以得到区间内最大的覆盖次数。


#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#define X first
#define Y second
#define L (u<<1)
#define R (u<<1|1)
#define Mid (tr[u].l+tr[u].r>>1)
#define Len(u) (tr[u].r-tr[u].l+1)
using namespace std;

typedef long long LL;
typedef pair<int,int> PII;

const int N=100010,mod=1e9+7,INF=0x3f3f3f3f;
const double eps=1e-6;

int n,m;
struct Node
{
    int l,r;
    int lsum,rsum;
}tr[N<<2];

void pushup(int u)
{
    tr[u].lsum=tr[L].lsum+tr[R].lsum;
    tr[u].rsum=tr[L].rsum+tr[R].rsum;
}

void build(int u,int l,int r)
{
    tr[u]={l,r,0,0};
    if(l==r) return;
    build(L,l,Mid),build(R,Mid+1,r);
}

void modify(int u,int l,int r,int c)
{
    if(tr[u].l>=l&&tr[u].r<=r)
    {
        if(c==1) tr[u].lsum+=1;
        else tr[u].rsum+=1;
        return;
    }
    if(l<=Mid) modify(L,l,r,c);
    else modify(R,l,r,c);
    pushup(u);
}

int query(int u,int l,int r,int op)
{
    if(tr[u].l>=l&&tr[u].r<=r)
    {
        if(op==1) return tr[u].lsum;
        else return tr[u].rsum;
    }
    int t=0;
    if(l<=Mid) t+=query(L,l,r,op);
    if(r>Mid) t+=query(R,l,r,op);
    return t;
}

int main()
{
//	ios::sync_with_stdio(false);
//	cin.tie(0);

    cin>>n>>m;;
    build(1,1,n);
    while(m--)
    {
        int op,l,r;
        scanf("%d%d%d",&op,&l,&r);
        if(op==1) modify(1,l,l,1),modify(1,r,r,2);
        else printf("%d\n",query(1,1,r,1)-query(1,1,l-1,2));
    }






	return 0;
}












你可能感兴趣的:(线段树)