牛客—— 红球进黑洞 (线段树+位运算)

简介: 牛客—— 红球进黑洞 (线段树+位运算)

牛客—— 红球进黑洞 (线段树+位运算)

铭宇巨巨推荐的题!


原题链接


题意:


给定一个序列,两种操作,一是区间求和,二是将区间里的每个数都异或x。


思路:


一眼就线段树,关键是怎么维护第二个操作。


借助最小异或生成树的思想以及异或题的常见套路,我们可以把每个数都进行二进制拆分,用线段树分别维护每一位上的0和1。


对于操作二,我们只需要分别维护每一位的值即可;对于操作一,计算区间里1的个数乘以对应的位数,就相当于是二进制转化为十进制的过程。


代码:

#pragma GCC optimize(3)
#pragma GCC optimize("Ofast","unroll-loops","omit-frame-pointer","inline")
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll,ll>PLL;
typedef pair<int,int>PII;
typedef pair<double,double>PDD;
#define I_int ll
#define x first
#define y second
inline ll read()
{
    ll x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
char F[200];
inline void out(I_int x) {
    if (x == 0) return (void) (putchar('0'));
    I_int tmp = x > 0 ? x : -x;
    if (x < 0) putchar('-');
    int cnt = 0;
    while (tmp > 0) {
        F[cnt++] = tmp % 10 + '0';
        tmp /= 10;
    }
    while (cnt > 0) putchar(F[--cnt]);
    //cout<<" ";
}
ll ksm(ll a,ll b,ll p){ll res=1;while(b){if(b&1)res=res*a%p;a=a*a%p;b>>=1;}return res;}
const int inf=0x3f3f3f3f,mod=1e9+7;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const int maxn=1e5+10,maxm=3e6+7;
const double PI = atan(1.0)*4;
struct node{
    int l,r,laz,sum;
}tr[25][maxn*4];
int a[maxn];
int n,m,k;
void pushup(int cnt,int u){
    tr[cnt][u].sum=tr[cnt][u<<1].sum+tr[cnt][u<<1|1].sum;
    tr[cnt][u].laz=0;
    return ;
}
void pushdown(int cnt,int u){
    if(tr[cnt][u].laz){
        tr[cnt][u].laz=0;
        tr[cnt][u<<1].laz^=1;tr[cnt][u<<1|1].laz^=1;
        tr[cnt][u<<1].sum=tr[cnt][u<<1].r-tr[cnt][u<<1].l+1-tr[cnt][u<<1].sum;
        tr[cnt][u<<1|1].sum=tr[cnt][u<<1|1].r-tr[cnt][u<<1|1].l+1-tr[cnt][u<<1|1].sum;
    }
    ///异或后1的个数 = 原来0的个数 = 区间长度(总个数)- 异或前1的个数
}
void build(int cnt,int u,int l,int r){
    tr[cnt][u]={l,r,0,0};
    if(l==r){
        tr[cnt][u].sum=(a[l]>>cnt)&1;
        return ;
    }
    int mid=(l+r)>>1;
    build(cnt,u<<1,l,mid);build(cnt,u<<1|1,mid+1,r);
    pushup(cnt,u);
}
///update(i,1,ql,qr,k&1);
void update(int cnt,int u,int ql,int qr,int x){
    if(tr[cnt][u].l>=ql&&tr[cnt][u].r<=qr){
        tr[cnt][u].sum=tr[cnt][u].r-tr[cnt][u].l+1-tr[cnt][u].sum;
        tr[cnt][u].laz^=x;
        return ;
    }
    pushdown(cnt,u);
    int mid=(tr[cnt][u].l+tr[cnt][u].r)>>1;
    if(ql<=mid) update(cnt,u<<1,ql,qr,x);
    if(qr>mid) update(cnt,u<<1|1,ql,qr,x);
    pushup(cnt,u);
}
///qask(i,1,ql,qr);
ll qask(int cnt,int u,int ql,int qr){
    if(tr[cnt][u].l>=ql&&tr[cnt][u].r<=qr) return tr[cnt][u].sum;
    pushdown(cnt,u);
    ll res=0;
    int mid=(tr[cnt][u].l+tr[cnt][u].r)>>1;
    if(ql<=mid) res+=qask(cnt,u<<1,ql,qr);
    if(qr>mid) res+=qask(cnt,u<<1|1,ql,qr);
    return res;
}
int main(){
    n=read(),m=read();
    for(int i=1;i<=n;i++) a[i]=read();
    for(int i=0;i<=20;i++) build(i,1,1,n);
    while(m--){
        int op=read(),ql=read(),qr=read();
        if(op==1){
            ll res=0;
            for(int i=0;i<=20;i++) res+=(1<<i)*qask(i,1,ql,qr);
            out(res);puts("");
        }
        else{
            int k=read();
            for(int i=0;i<=20;i++){
                if(k&1) update(i,1,ql,qr,k&1);
                k>>=1;
            }
        }
    }
    return 0;
}

参考博客

目录
相关文章
|
6月前
|
机器学习/深度学习 算法
刷题记录:牛客-WY49数对 | 以数学分析来破解暴力搜索的时间复杂度问题 2023/1/11
这是一个关于编程题解的文章摘要,讨论了一道名为&quot;WY49 数对&quot;的问题。文章指出,暴力搜索的解决方案在大规模问题中效率低下,然后转向通过数学分析来寻找更优解。作者解释了如何根据除数和余数的关系,以及余数的周期性变化来计算满足条件的数对数量。通过将数对中的其中一个数(被模数x)按除数y划分区间,并分析每个区间的余数分布,得出一个公式来计算总数。最后,提供了两种不同的代码实现来展示这个思路,这些代码具有O(n)的时间复杂度和O(1)的空间复杂度。文章强调了理解数学方法在解决此类问题中的重要性,特别是对于优化算法性能的作用。
93 3
|
6月前
【错题集-编程题】最长上升子序列(二)(贪心 + 二分)
【错题集-编程题】最长上升子序列(二)(贪心 + 二分)
|
6月前
|
算法 C++ Java
C/C++每日一练(20230421) 位1的个数、递归和非递归求和、俄罗斯套娃信封问题
C/C++每日一练(20230421) 位1的个数、递归和非递归求和、俄罗斯套娃信封问题
42 0
C/C++每日一练(20230421) 位1的个数、递归和非递归求和、俄罗斯套娃信封问题
|
6月前
|
算法 vr&ar
1611F - ATM and Students详细题解(*1800,线段树维护前缀和;双指针算法(思维))
1611F - ATM and Students详细题解(*1800,线段树维护前缀和;双指针算法(思维))
46 0
|
11月前
|
算法 测试技术 C#
一题三解(暴力、二分查找算法、单指针):鸡蛋掉落
一题三解(暴力、二分查找算法、单指针):鸡蛋掉落
|
算法
代码随想录Day28 贪心03 LeetCode T1005 K次取反后最大化的数组和 LeetCode T134 加油站 LeetCode T135 分发糖果
代码随想录Day28 贪心03 LeetCode T1005 K次取反后最大化的数组和 LeetCode T134 加油站 LeetCode T135 分发糖果
35 0
|
算法 C++
【每日算法Day 104】偷电瓶的周某今天放出来了,还不赶紧做这道题防范一下!
【每日算法Day 104】偷电瓶的周某今天放出来了,还不赶紧做这道题防范一下!
|
算法
【每日挠头算法题】Acwing 756. 蛇形矩阵 —— 巧妙解法
【每日挠头算法题】Acwing 756. 蛇形矩阵 —— 巧妙解法
140 0
【每日挠头算法题】Acwing 756. 蛇形矩阵 —— 巧妙解法
|
算法 Java C++
代码随想录刷题|LeetCode 1005.K次取反后最大化的数组和 134. 加油站 135. 分发糖果
代码随想录刷题|LeetCode 1005.K次取反后最大化的数组和 134. 加油站 135. 分发糖果
【刷穿 LeetCode】371. 两整数之和 : 使用「位运算」求解的两种方式
【刷穿 LeetCode】371. 两整数之和 : 使用「位运算」求解的两种方式