线段树题集合

统计颜色

题意:
n个桶按顺序排列,我们用1~n给桶标号。有两种操作:
1 l r c 区间[l,r]中的每个桶中都放入一个颜色为c的球 (1≤l,r ≤n,l≤r,0≤c≤60)
2 l r   查询区间[l,r]的桶中有多少种不同颜色的球     (1≤l,r ≤n,l≤r)

解析:

二进制优化,线段树或操作

最多61种求,转化为一个LL数字

每次加一个c球,就对区间(l,r)或上1LL<<c

最后求query返回的数中各位1的个数

wa:1<<c是错误写法,正确写法是:1ll<<c  !!!

ac:

#include<bits/stdc++.h>
#define ll long long
#define MAXN 400005
#define ls l,m,rt<<1
#define rs m+1,r,rt<<1|1
using namespace std;
ll sum[MAXN],add[MAXN];

void pushup(int rt)
{
    sum[rt]=sum[rt<<1]|sum[rt<<1|1];
}

void pushdown(int rt)
{
    if(add[rt])
    {
        add[rt<<1]|=add[rt];
        add[rt<<1|1]|=add[rt];
        sum[rt<<1]|=add[rt];
        sum[rt<<1|1]|=add[rt];
        add[rt]=0;
    }
}

void update(int a,int b,ll v,int l,int r,int rt)
{
    if(a<=l&&b>=r)
    {
        add[rt]|=v;
        sum[rt]|=v;
        return ;
    }
    pushdown(rt);
    int m=(l+r)>>1;
    if(a<=m)
        update(a,b,v,ls);
    if(m<b)
        update(a,b,v,rs);
    pushup(rt);
}

ll query(int a,int b,int l,int r,int rt)
{
    if(a<=l&&b>=r)
    {
        return sum[rt];
    }
    pushdown(rt);
    int m=(l+r)>>1;
    ll res=0;
    if(a<=m)
        res|=query(a,b,ls);
    if(m<b)
        res|=query(a,b,rs);
    return res;
}

int getc(ll g)
{
    int ans=0;
    while(g)
    {
        ans+=g%2;
        g=g/2;
    }
    return ans;
}

int main()
{
    int n,m,x,l,r;
    ll c;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        memset(add,0,sizeof(add));
        memset(sum,0,sizeof(sum));
        for(int i=1;i<=m;i++)
        {
            scanf("%d",&x);
            if(x==1)
            {
                scanf("%d%d%lld",&l,&r,&c);
                c=1LL<<c;
                update(l,r,c,1,n,1);
            }
            else{
                scanf("%d%d",&l,&r);
                printf("%d\n",getc(query(l,r,1,n,1)));
            }
        }
    }
    return 0;
}

https://codeforces.com/contest/1249/problem/D2

题意:

给定n个区间,满足重叠次数不超过k次的情况下选择最多的区间,问需要删除哪些区间

解析:

区间按r从小到大排序,然后区间+1,然后判断是否>k,如果>k就删除这个区间

ac:

#include<bits/stdc++.h>
#define ls l,m,rt<<1
#define rs m+1,r,rt<<1|1
#define MAXN 200005
using namespace std;
struct node
{
    int l,r,id;
    friend bool operator <(node a,node b)
    {
        return a.r<b.r;
    }
}ee[MAXN];
int maxs[MAXN<<2]={0};
int add[MAXN<<2]={0};

void push_up(int rt)
{
    maxs[rt]=max(maxs[rt<<1],maxs[rt<<1|1]);
}

void pushdown(int l,int r,int rt)
{
    if(add[rt]!=0){
        add[rt<<1]+=add[rt];
        add[rt<<1|1]+=add[rt];
        maxs[rt<<1]+=add[rt];
        maxs[rt<<1|1]+=add[rt];
        add[rt]=0;
    }
}

void build(int l,int r,int rt)
{
    if(l==r){
        maxs[rt]=0;
        return ;
    }
    int m=(l+r)>>1;
    build(ls);
    build(rs);
}

void update(int a,int b,int c,int l,int r,int rt)
{
    if(a<=l&&r<=b)
    {
        add[rt]+=c;
        maxs[rt]+=c;
        return ;
    }
    int m=(l+r)>>1;
    pushdown(l,r,rt);
    if(a<=m)
        update(a,b,c,ls);
    if(b>m)
        update(a,b,c,rs);
    push_up(rt);
}

int query(int a,int b,int l,int r,int rt)
{
    if(a<=l&&r<=b)
    {
        return maxs[rt];
    }
    int m=(l+r)>>1;
    pushdown(l,r,rt);
    int ans=0;
    if(a<=m)
        ans=max(ans,query(a,b,ls));
    if(b>m)
        ans=max(ans,query(a,b,rs));
    return ans;
}
int vis[MAXN];

int main()
{
    int n,k;
    scanf("%d%d",&n,&k);
    int maxs=0;
    for(int i=1;i<=n;i++){
        scanf("%d%d",&ee[i].l,&ee[i].r);
        ee[i].id=i;
        maxs=max(maxs,max(ee[i].l,ee[i].r));
    }
    sort(ee+1,ee+n+1);
    build(1,maxs,1);
    for(int i=1;i<=n;i++)
    {
        update(ee[i].l,ee[i].r,1,1,maxs,1);
        if(query(1,ee[i].r,1,maxs,1)>k){
            update(ee[i].l,ee[i].r,-1,1,maxs,1);
        }
        else{
            vis[ee[i].id]=1;
        }
    }
    int ans=0;
    for(int i=1;i<=n;i++)
        if(vis[i]==0)
            ans++;
    printf("%d\n",ans);
    for(int i=1;i<=n;i++)
        if(vis[i]==0)
            printf("%d ",i);
    return 0;
}

2.链接:https://ac.nowcoder.com/acm/contest/946/D
来源:牛客网
时间限制:C/C++ 1秒,其他语言2秒
空间限制:C/C++ 32768K,其他语言65536K
64bit IO Format: %lld

题目描述

筱玛是一个热爱线段树的好筱玛。

筱玛的爷爷马爷在游戏中被筱玛吊打了,于是他恼羞成怒,决定给筱玛出这样一道数据结构题:

给定一个长度为nn的数组AA,刚开始每一项的值均为00。

支持以下两种操作,操作共mm次:

1 l r1 l r:将Al∼ArAl∼Ar的每一项的值加上11。

2 l r2 l r:执行操作编号在[l,r][l,r]内的所有操作各一次,保证rr小于当前操作的编号。

mm次操作结束后,你要告诉马爷AA数组变成什么样子了。

由于答案可能会很大,你只需要输出数组AA中的每个数在模109+7109+7意义下的值。

输入描述:

第一行两个数n,mn,m,分别表示数组长度及操作次数。
接下来mm行,每行三个数opt,l,ropt,l,r,表示一次操作。

输出描述:

输出一行共nn个数,表示mm次操作结束后,A1∼AnA1∼An的值。解析:

当前操作仅对以前的操作有效

所以从后往前遍历

两个线段树

1个记录操作的叠加,即2产生的效果

1个记录操作1产生的效果,即数字的改变

ac:

#include<bits/stdc++.h>
#define ls l,m,rt<<1
#define rs m+1,r,rt<<1|1
#define lowbit(x)  (x)&(-x)
#define MAXN 100005
#define ll long long
#define mod 1000000007
using namespace std;
int n;

ll sum[MAXN<<2],add[MAXN<<2];

void pushup(int rt)//往上传
{
    sum[rt]=(sum[rt<<1]+sum[rt<<1|1])%mod;
}

void pushdown(int l,int r,int rt)//往下传
{
    if(add[rt])
    {
        int m=(l+r)>>1;
        add[rt<<1]=(add[rt<<1]+add[rt])%mod;//下传懒惰标记
        add[rt<<1|1]=(add[rt<<1|1]+add[rt])%mod;
        sum[rt<<1]=(sum[rt<<1]+(add[rt]*(m-l+1))%mod)%mod;//下传修改儿子结点值
        sum[rt<<1|1]=(sum[rt<<1|1]+(add[rt]*(r-(m+1)+1))%mod)%mod;
        add[rt]=0;//该点懒惰标记清零
    }
}

void build(int l,int r,int rt)
{
    if(l==r)
    {
        sum[rt]=0;
        return ;
    }
    int m=(l+r)>>1;
    build(ls);
    build(rs);
    pushup(rt);//更新父节点,往上更新
}

void update(int a,int b,int v,int l,int r,int rt)
{
    if(a<=l&&b>=r)//完全被包围
    {
        sum[rt]=(sum[rt]+((r-l+1)*v)%mod)%mod;
        add[rt]=(add[rt]+v)%mod;
        return ;
    }
    pushdown(l,r,rt);//要访问儿子结点,要把懒标记下传了
    int m=(l+r)>>1;
    if(a<=m)//包含左边,改左边
        update(a,b,v,ls);
    if(b>m)//包含右边,改右边
        update(a,b,v,rs);
    pushup(rt);
}

ll query(int a,int b,int l,int r,int rt)
{
    if(a<=l&&b>=r)
    {
        return sum[rt];
    }
    pushdown(l,r,rt);
    ll res=0;
    int m=(l+r)>>1;
    if(a<=m)
        res=(res+query(a,b,ls))%mod;
    if(b>m)
        res=(res+query(a,b,rs))%mod;
    return res;
}

int query3(int x,int l,int r,int rt)
{
    if (l==r)
    {
        return sum[rt];
    }
    pushdown(l,r,rt);
    int m=(l+r)>>1;
    if (x<=m)
        return query3(x,ls);
    else
        return query3(x,rs);
}

ll sum2[MAXN<<2],add2[MAXN<<2];

void pushup2(int rt)//往上传
{
    sum2[rt]=(sum2[rt<<1]+sum2[rt<<1|1])%mod;
}

void pushdown2(int l,int r,int rt)//往下传
{
    if(add2[rt])
    {
        int m=(l+r)>>1;
        add2[rt<<1]=(add2[rt<<1]+add2[rt])%mod;//下传懒惰标记
        add2[rt<<1|1]=(add2[rt<<1|1]+add2[rt])%mod;
        sum2[rt<<1]=(sum2[rt<<1]+(add2[rt]*(m-l+1))%mod)%mod;//下传修改儿子结点值
        sum2[rt<<1|1]=(sum2[rt<<1|1]+(add2[rt]*(r-(m+1)+1))%mod)%mod;
        add2[rt]=0;//该点懒惰标记清零
    }
}

void build2(int l,int r,int rt)
{
    if(l==r)
    {
        sum2[rt]=0;
        return ;
    }
    int m=(l+r)>>1;
    build2(ls);
    build2(rs);
    pushup2(rt);//更新父节点,往上更新
}

void update2(int a,int b,int v,int l,int r,int rt)
{
    if(a<=l&&b>=r)//完全被包围
    {
        sum2[rt]=(sum2[rt]+((r-l+1)*v)%mod)%mod;
        add2[rt]=(add2[rt]+v)%mod;
        return ;
    }
    pushdown2(l,r,rt);//要访问儿子结点,要把懒标记下传了
    int m=(l+r)>>1;
    if(a<=m)//包含左边,改左边
        update2(a,b,v,ls);
    if(b>m)//包含右边,改右边
        update2(a,b,v,rs);
    pushup2(rt);
}

ll query2(int a,int b,int l,int r,int rt)
{
    if(a<=l&&b>=r)
    {
        return sum2[rt];
    }
    pushdown2(l,r,rt);
    ll res=0;
    int m=(l+r)>>1;
    if(a<=m)
        res=(res+query2(a,b,ls))%mod;
    if(b>m)
        res=(res+query2(a,b,rs))%mod;
    return res;
}

int query4(int x,int l,int r,int rt)
{
    if (l==r)
    {
        return sum2[rt];
    }
    pushdown2(l,r,rt);
    int m=(l+r)>>1;
    if (x<=m)
        return query4(x,ls);
    else
        return query4(x,rs);
}

struct node
{
    int a,l,r;
}q[MAXN];

int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
        scanf("%d%d%d",&q[i].a,&q[i].l,&q[i].r);
    build(1,m,1);
    for(int i=m;i>=1;i--)
    {
        if(q[i].a==2)
        {
            int v=query3(i,1,m,1)+1;
            update(q[i].l,q[i].r,v,1,m,1);
        }
    }
    build2(1,n,1);
    for(int i=m;i>=1;i--)
    {
        if(q[i].a==1)
        {
            int v=query3(i,1,m,1)+1;
            update2(q[i].l,q[i].r,v,1,n,1);
        }
    }
    for(int i=1;i<=n;i++)
    {
        printf("%d ",query4(i,1,n,1));
    }
    return 0;
}

链接:https://ac.nowcoder.com/acm/contest/949/F
来源:牛客网
时间限制:C/C++ 1秒,其他语言2秒
空间限制:C/C++ 262144K,其他语言524288K
64bit IO Format: %lld

题目描述

小石有 n 个妹子,每个妹子都有一个细心程度 aiai 和一个热心程度 bibi,
小石想给她们一个重要程度 titi(重要程度为 1 表示最重要,重要程度越小表示越重要)。
如果一个妹子 i 的细心程度和热心程度都比妹子 j 大,那么妹子 i 的重要程度要大于妹子 j 的重要程度,即妹子 i 比妹子 j 重要。
流程如下:
每次从所有没有重要程度的妹子中,找到若干妹子。对于这些妹子的任意一个,需要保证没有其他妹子比她更重要。然后把她们的重要程度标为 1 。下一次再从剩下没有重要程度的妹子中找到若干妹子,依然符合上述条件,然后把她们的重要程度标为 2,……,重复直到所有妹子都有自己的重要程度。
由于妹子太多,小石忙不过来,请你帮帮他。

输入描述:

第一行输入一个正整数 n,表示妹子的数量。
接下来 n 行,每行两个正整数 ai,biai,bi,描述每个妹子的细心程度和热心程度。 
保证所有的 aiai 两两不等,所有的 bibi 两两不等。 

输出描述:

共 n 行,第 i 行输出一个正整数 titi 表示第 i 个妹子的重要程度。

备注:

1≤n≤105,1≤ai,bi≤109

解析:

对于第i个妹子,她的重要程度是她的排名.aj>ai&&bj>bi

我们要找到max(rank(j))+1,ai<aj

我们按ai排序,用线段树维护区间极大值,然后查找(bi+1,n)范围内最大的rk,rki=max(rk(bi+1,n))+1

随后将rki加入到线段树中

ac:

#include<bits/stdc++.h>
using namespace std;
#define mid (l+r)/2
const int N=100005;
int n,len,b[N],ans[N],tree[N*4];
struct node{
    int x,y,id;
}a[N];
#define gc getchar
inline int read()
{
    int res=0,f=0;char c;c=gc();
    while(!isdigit(c)){if(c=='-')f=1;c=gc();}
    while(isdigit(c)){res=res*10+c-48;c=gc();}
    if(f)return -res;return res;
}
bool cmp(node a,node b)
{
    return a.x>b.x;
}
void change(int nod,int l,int r,int x,int val)
{
    if(l==r)
    {
        tree[nod]=val;
        return;
    }
    if(x<=mid)change(nod*2,l,mid,x,val);
    else change(nod*2+1,mid+1,r,x,val);
    tree[nod]=max(tree[nod*2],tree[nod*2+1]);
}
int find(int nod,int l,int r,int L,int R)
{
    if(l==L&&r==R)return tree[nod];
    if(R<=mid)return find(nod*2,l,mid,L,R);
    else if(L>mid)return find(nod*2+1,mid+1,r,L,R);
    else return(max(find(nod*2,l,mid,L,mid),find(nod*2+1,mid+1,r,mid+1,r)));
}
int main()
{
    n=read();
    for(int i=1;i<=n;i++)
    {
        a[i].x=read();a[i].y=read();a[i].id=i;
        b[i]=a[i].y;
    }
    sort(b+1,b+n+1);
    len=unique(b+1,b+n+1)-b-1;
    for(int i=1;i<=n;i++)
        a[i].y=lower_bound(b+1,b+len+1,a[i].y)-b;
    sort(a+1,a+n+1,cmp);
    for(int i=1;i<=n;i++)
    {
        ans[a[i].id]=find(1,1,n,a[i].y,n)+1;
        change(1,1,n,a[i].y,ans[a[i].id]);
    }
    for(int i=1;i<=n;i++)printf("%d\n",ans[i]);
    return 0;
}

链接:https://ac.nowcoder.com/acm/contest/949/H
来源:牛客网
时间限制:C/C++ 1秒,其他语言2秒
空间限制:C/C++ 524288K,其他语言1048576K
64bit IO Format: %lld

题目描述

小阳手中一共有 n 个贝壳,每个贝壳都有颜色,且初始第 i 个贝壳的颜色为 colicoli 。现在小阳有 3 种操作:
1 l r x:给 [l,r][l,r] 区间里所有贝壳的颜色值加上 xx 。
2 l r:询问 [l,r][l,r] 区间里所有相邻贝壳 颜色值的差(取绝对值) 的最大值(若 l=rl=r 输出 0)。
3 l r :询问 [l,r][l,r] 区间里所有贝壳颜色值的最大公约数。

输入描述:

第一行输入两个正整数 n,mn,m,分别表示贝壳个数和操作个数。
第二行输入 nn 个数 colicoli,表示每个贝壳的初始颜色。
第三到第 m+2m+2 行,每行第一个数为 optopt,表示操作编号。接下来的输入的变量与操作编号对应。

输出描述:

共 m 行,对于每个询问(操作 2 和操作 3)输出对应的结果。
输入

5 6
2 2 3 3 3
1 2 3 3
2 2 4
3 3 5
1 1 4 2
3 2 3
2 3 5

输出

3
3
1
3

备注:

1≤n,m≤105,1≤coli,x≤103,1≤opt≤3,1≤l≤r≤n

解析:
1 l r x:给 [l,r][l,r] 区间里所有贝壳的颜色值加上 xx 。

2 l r:询问 [l,r][l,r] 区间里所有相邻贝壳 颜色值的差(取绝对值) 的最大值(若 l=rl=r 输出 0)。

3 l r :询问 [l,r][l,r] 区间里所有贝壳颜色值的最大公约数。

1很好处理,关键是3怎么区间修改

gcd(a,b)=gcd(a,b-a)

gcd(a,b,c,d)=gcd(a,b-a,c-b,d-c);

gcd(a+1,b+1,c+1,d)=gcd(a+1,b-a,c-b,d-c-1);

查询gcd(ai,...an)=gcd(ai,bi+1,....bn);

ai=(1,i)sum(bi)

gcd(ai,...an)=gcd(sum(b1,...,bi),bi+1,bn);

如上例:a+1,b+1,c+1, [1,3]修改+1,中间的不用变,仅仅l修改+1,[r+1] 修改-1

我们建立一个差分数组维护gcd和max

区间修改维护差分单点修改即可,max就是差分数组的绝对值最大

ac:

#include<bits/stdc++.h>
#define ll long LONG_MAX
#define ls l,m,rt<<1
#define rs m+1,r,rt<<1|1
#define MAXN 400005
using namespace std;

int maxs[MAXN],sum[MAXN],a[MAXN],Gcd[MAXN];

int gcd(int a,int b)
{
    return b==0?a:gcd(b,a%b);
}

void pushup(int rt)
{
    maxs[rt]=max(maxs[rt<<1],maxs[rt<<1|1]);
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
    Gcd[rt]=gcd(Gcd[rt<<1],Gcd[rt<<1|1]);
}

void build(int l,int r,int rt)
{
    if(l==r)
    {
        sum[rt]=a[l];
        Gcd[rt]=maxs[rt]=abs(a[l]);
        return ;
    }
    int m=(l+r)>>1;
    build(ls);
    build(rs);
    pushup(rt);
}

void update(int x,int v,int l,int r,int rt)
{
    if(l==r)
    {
        a[l]+=v;
        sum[rt]=a[l];
        maxs[rt]=Gcd[rt]=abs(a[l]);
        return ;
    }
    int m=(l+r)>>1;
    if(x<=m)
        update(x,v,ls);
    else
        update(x,v,rs);
    pushup(rt);
}

int query_sum(int a,int b,int l,int r,int rt)
{
    if(a<=l&&b>=r)
    {
        return sum[rt];
    }
    int m=(l+r)>>1;
    int res=0;
    if(a<=m)
        res+=query_sum(a,b,ls);
    if(b>m)
        res+=query_sum(a,b,rs);
    return res;
}

int query_max(int a,int b,int l,int r,int rt)
{
    if(a<=l&&b>=r)
    {
        return abs(maxs[rt]);
    }
    int m=(l+r)>>1;
    int res=0;
    if(a<=m)
        res=max(res,query_max(a,b,ls));
    if(b>m)
        res=max(res,query_max(a,b,rs));
    return res;
}

int query_gcd(int a,int b,int l,int r,int rt)
{
    if(a<=l&&b>=r)
    {
        return Gcd[rt];
    }
    int m=(l+r)>>1;
    int res=0;
    if(a<=m)
        res=gcd(res,query_gcd(a,b,ls));
    if(b>m)
        res=gcd(res,query_gcd(a,b,rs));
    return res;
}

int main()
{
    int n,m,op,l,r,x;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    for(int i=n;i>=1;i--)
        a[i]-=a[i-1];
    build(1,n,1);
    while(m--)
    {
        scanf("%d",&op);
        if(op==1)
        {
            scanf("%d%d%d",&l,&r,&x);
            update(l,x,1,n,1);
            if(r<n)
                update(r+1,-x,1,n,1);
        }
        else if(op==2)
        {
            scanf("%d%d",&l,&r);
            printf("%d\n",query_max(l+1,r,1,n,1));
        }
        else{
            scanf("%d%d",&l,&r);
            int res=query_sum(1,l,1,n,1);
            printf("%d\n",gcd(res,query_gcd(l+1,r,1,n,1)));
        }
    }
    return 0;
}

 

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值