結果

問題 No.879 Range Mod 2 Query
コンテスト
ユーザー vwxyz
提出日時 2024-04-11 01:57:35
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,039 bytes
コンパイル時間 3,190 ms
コンパイル使用メモリ 257,428 KB
実行使用メモリ 15,488 KB
最終ジャッジ日時 2024-10-02 21:09:52
合計ジャッジ時間 7,264 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 3 WA * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <atcoder/lazysegtree>
#include <cstdio>
#include <bits/stdc++.h>

using namespace std;
using namespace atcoder;


struct S {
    long long s;
    long long ce;
    long long co;
};

struct F {
    long long p;
    long long d;
    long long x;
};

S op(S l, S r) {
	return S{l.s+r.s, l.ce+r.ce, l.co+r.co};
}

S e() {
	return S{0ll, 0ll, 0ll};
}

F composition(F l, F r) {
	if (l.d){
        return F({(r.p+r.x+l.p)%2,1ll,l.x});
	}
	else if (r.d){
        return F({r.p,1,r.x+l.x});
	}
    else{
        return F({0,0,r.x+l.x});
    }
}

S mapping(F l, S r) {
    long long s,ce,co;
    if (l.d){
        if(l.p){
            ce=r.co;
            co=r.ce;
        }
        else{
            ce=r.ce;
            co=r.co;
        }
        s=co+l.x*(ce+co);
        if(l.x%2){
            ce=r.co;
            co=r.ce;
        }
        else{
            ce=r.ce;
            co=r.co;
        }
    }
    else{
        s=r.s+l.x*(r.ce+r.co);
        if(l.x%2){
            ce=r.co;
            co=r.ce;
        }
        else{
            ce=r.ce;
            co=r.co;            
        }
    }
    return S({s,ce,co});
}

F id() {
	return F({0ll, 0ll, 0ll});
}

int main() {
    int N, Q;
    cin >> N >> Q;
    vector<long long> A(N);
    for(int i=0;i<N;i++){
        cin>>A[i];
    }
    vector<S> a(N);
    for(int i=0;i<N;i++){
        if(A[i]%2){
            a[i]=S({A[i],0ll,1ll});
        }
        else{
            a[i]=S({A[i],1ll,0ll});
        }
    }
    lazy_segtree<S, op, e, F, mapping, composition, id> seg(a);
    for(int q=0;q<Q;q++){
    	int t,l,r;
    	long long x;
    	cin>>t;
        if(t==1){
            cin>>l>>r;
            l-=1;
            seg.apply(l,r,F({0ll,1ll,0ll}));
        }
        else if(t==2){
            cin>>l>>r>>x;
            l-=1;
            seg.apply(l,r,F({0ll,0ll,x}));
        }
    	else if(t==3){
            cin>>l>>r;
            l-=1;
	    	long long ans=seg.prod(l,r).s;
	    	cout<<ans<<endl;
        }
    }
    return 0;
}
0