Gym-103145-problem-D. Lowbit (吉司机线段树)

链接:D. Lowbit

题意：
给一个长度为n的数组a,有以下两种操作:

1. 给出l , r 把区间[l , r]内的每一个ai变成 a[i] + lowbit(a[i]).
2. 查询区间[l , r]的权值和。

思路:
对于某个数加上它的lowbit，如果这个数是 1000000(2)的形式，它的lowbit等于本身，那么等价于乘2。所以我们对区间打一个标记，如果全是上述形式，那么直接区间乘，否则就暴力更新。

代码:

#include <iostream>
#include <cstdio>
#include <queue>
#include <math.h>
#include <map>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
const int maxn = 1e6 + 7;
typedef long long ll;
const int mod=998244353;
int T , n , q;
int vis[maxn];
ll sum[maxn],pre[maxn],la[maxn];
ll a[maxn];
ll lowbit(ll x){
    return x & ( - x);
}
void pushup(int rt){
    sum[rt] = (sum[rt << 1] + sum[rt << 1 | 1]) % mod;
    vis[rt] = min(vis[rt << 1] , vis[rt << 1 | 1]);
}
void pushdown(int rt){
    if(la[rt] == 1) return ;
    sum[rt << 1] = sum[rt << 1] * la[rt] % mod;
    sum[rt << 1 | 1] = sum[rt << 1 | 1] * la[rt] % mod;
    la[rt << 1] = la[rt << 1] * la[rt] % mod;
    la[rt << 1 | 1] = la[rt << 1 | 1] * la[rt] % mod;
    la[rt] = 1;
}
void build(int l,int r,int rt){
    la[rt] = 1;
    if(l == r){
        sum[rt] = a[l];
        vis[rt] = 0;
        return ;
    }
    int mid = (l + r) / 2;
    build(l , mid , rt << 1);
    build(mid + 1 , r , rt << 1 | 1);
    pushup(rt);
}
void update(int L,int R,int l , int r,int rt){
    if(L <= l && R >= r && vis[rt] == 1){
        sum[rt] = sum[rt] * 2 % mod;
        la[rt] = la[rt] * 2 % mod;
        return ;
    }
    if(l == r){
        sum[rt] = sum[rt] + lowbit(sum[rt]);
        if(sum[rt] == lowbit(sum[rt])) vis[rt] = 1;
        return;
    }
    pushdown(rt);
    int mid = (l + r) / 2;
    if(L <= mid) update(L , R , l , mid , rt << 1);
    if(R > mid) update(L , R , mid + 1 , r , rt << 1 | 1);
    pushup(rt);
}
ll query(int L,int R,int l, int r , int rt){
    if(L <= l && R >= r){
        return sum[rt];
    }
    pushdown(rt);
    ll ans = 0;
    int mid = (l + r) / 2;
    if(L <= mid) ans = (ans +  query(L , R , l , mid , rt << 1)) % mod;
    if(R > mid)  ans = (ans +  query(L , R , mid + 1 , r , rt << 1 | 1)) % mod;
    return ans;
}
int main(){
    scanf("%d",&T);
    while(T--){
        scanf("%d",&n);
        for(int i = 1; i <= n; i ++){
            scanf("%lld",&a[i]);
        }
        build(1 , n , 1);
        int op , l , r;
        scanf("%d",&q);
        while(q--){
            scanf("%d%d%d",&op,&l,&r);
            if(op == 1){
                update(l , r , 1 ,n , 1);
            }
            else{
                ll ans = query(l , r , 1 , n , 1);
                ans = ans % mod;
                printf ("%lld\n",ans);
            }
        }
    }
}