#1197491 (C++17) No.3792 XOR^2 Sum Query

• 問題: No.3792 XOR^2 Sum Query
• ユーザー: 👑 potato167
• 提出日時: 2026-11-06 23:05:16
• 言語: C++17 (gcc 15.3.0 + boost 1.92.0 + ACL)
• 結果: TLE
• 実行時間: -
• コード長: 4,761 bytes
• コンパイル時間: 1,685 ms
• コンパイル使用メモリ: 238,100 KB
• 実行使用メモリ: 138,360 KB
• 最終ジャッジ日時: 2026-11-06 23:07:14
• 合計ジャッジ時間: 82,761 ms
• ジャッジサーバーID （参考情報）: judge4_0 / judge1_0

ファイルパターン	結果
sample	AC * 1
other	AC * 28 WA * 4 TLE * 3

ソースコード

#line 1 "b.cpp"
// https://github.com/potato167/po167_library
#include <bits/stdc++.h>
#line 4 "b.cpp"
using namespace std;
using ll=long long;
const ll ILL=2167167167167167167;
const int INF=2100000000;
#define rep(i,a,b) for (int i=(int)(a);i<(int)(b);i++)
#define all(p) p.begin(),p.end()
template<class T> using pq_ = priority_queue<T, vector<T>, greater<T>>;
template<class T> int LB(vector<T> &v,T a){return lower_bound(v.begin(),v.end(),a)-v.begin();}
template<class T> int UB(vector<T> &v,T a){return upper_bound(v.begin(),v.end(),a)-v.begin();}
template<class T> bool chmin(T &a,T b){if(b<a){a=b;return 1;}else return 0;}
template<class T> bool chmax(T &a,T b){if(a<b){a=b;return 1;}else return 0;}
template<class T> void So(vector<T> &v) {sort(v.begin(),v.end());}
template<class T> void Sore(vector<T> &v) {sort(v.begin(),v.end(),[](T x,T y){return x>y;});}
bool yneos(bool a,bool upp=false){if(a){cout<<(upp?"YES\n":"Yes\n");}else{cout<<(upp?"NO\n":"No\n");}return a;}
template<class T> void vec_out(vector<T> &p,int ty=0){
    if(ty==2){cout<<'{';for(int i=0;i<(int)p.size();i++){if(i){cout<<",";}cout<<'"'<<p[i]<<'"';}cout<<"}\n";}
    else{if(ty==1){cout<<p.size()<<"\n";}for(int i=0;i<(int)(p.size());i++){if(i) cout<<" ";cout<<p[i];}cout<<"\n";}}
template<class T> T vec_min(vector<T> &a){assert(!a.empty());T ans=a[0];for(auto &x:a) chmin(ans,x);return ans;}
template<class T> T vec_max(vector<T> &a){assert(!a.empty());T ans=a[0];for(auto &x:a) chmax(ans,x);return ans;}
template<class T> T vec_sum(vector<T> &a){T ans=T(0);for(auto &x:a) ans+=x;return ans;}
int pop_count(long long a){int res=0;while(a){res+=(int)(a&1),a>>=1;}return res;}
template<class T> T square(T a){return a * a;}

#include <atcoder/modint>
using mint = atcoder::modint998244353;

#include <atcoder/segtree>
int sum_mint(int a, int b) {
    return a ^ b;
}
int e_mint() {
    return 0;
}
int op(int a, int b) {
    return (a | b);
}
int e() {
    return 0;
}
random_device seed_gen;
mt19937 rng(seed_gen());
// return [l, r)
long long rand_long(long long l,long long r){
    return uniform_int_distribution<long long>(l,r-1)(rng);
}


void solve();
// DEAR MYSTERIES / TOMOO
int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int t = 1;
    // cin >> t;
    rep(i, 0, t) solve();
}

void solve(){
    int N, Q;
    cin >> N >> Q;
    vector<int> A(N);
    rep(i, 0, N) cin >> A[i];
    vector<int> hash(N);
    rep(i, 0, N) hash[i] = rand_long(167, 1 << 30);
    atcoder::segtree<int, op, e> seg(A);
    const int D = 30;
    vector<mint> p2(N + 100);
    p2[0] = 1;
    rep(i, 1, 100 + N) p2[i] = p2[i - 1] * 2;
    vector<atcoder::segtree<int, sum_mint, e_mint>> H;
    rep(i, 0, D) {
        vector<int> init(N);
        rep(j, 0, N) if (A[j] & (1 << i)) init[j] = hash[j];
        H.push_back(atcoder::segtree<int, sum_mint, e_mint>(init));
    }
    vector<int> order(D);
    rep(d, 0, D) order[d] = d;
    vector<mint> val(D);
    mint inv4 = (mint(4)).inv();
    while (Q--) {
        int t;
        cin >> t;
        if (t == 1) {
            int i, x;
            cin >> i >> x;
            i--;
            seg.set(i, x);
            rep(d, 0, D) {
                if (x & (1 << d)) H[d].set(i, hash[i]);
                else H[d].set(i, 0);
            }
        }
        else {
            int l, r;
            cin >> l >> r;
            l--;
            auto tmp = seg.prod(l, r);
            mint ans = (ll(tmp)) * (ll(tmp));
            ans *= 2;
            rep(d, 0, D) val[d] = H[d].prod(l, r);
            sort(all(order), [&](int a, int b) {
                return val[a].val() < val[b].val();
            });
            mint sum = 0;
            // cout << ans.val() << " ";
            for (int a = 0, b = 0; a < D; a = b) {
                mint sum2 = 0;
                while (b != D && val[order[a]] == val[order[b]]) {
                    sum2 += (1 << order[b]);
                    b++;
                }
                if (val[order[a]] == 0) continue;
                ans -= sum * sum2 * 2;
                sum += sum2;
            }
            // cout << ans.val() << "\n";
            ans *= inv4;
            ans *= p2[r - l];
            cout << ans.val() << "\n";
        }
    }
}


/*
 * 区間に対する、総 xor の 2 乗
 * 単体 bit の2乗は、単純に1個以上あるなら半分の確率でok
 * 2つのやつは、結局基底があるかいなかになる
 * 01, 10, 11 のうち、1種類しかないならば、
 * 11 だけなら 1/2
 * それ以外は 0
 * 他は 1/4
 * 01, 10 だけというのは簡単に検知ができる
 * 11 だけがむずい
 * まあ各桁でhashでもなんでも持っておけばいいんだけど
 * データの保持多すぎ
 * 010
 * 011
 * 100
 */