結果

問題 No.879 Range Mod 2 Query
ユーザー kyort0nkyort0n
提出日時 2019-09-06 23:18:06
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 5,939 bytes
コンパイル時間 2,128 ms
コンパイル使用メモリ 178,700 KB
実行使用メモリ 23,804 KB
最終ジャッジ日時 2023-09-07 03:46:45
合計ジャッジ時間 7,028 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 13 ms
23,496 KB
testcase_01 WA -
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<ll, ll> l_l;
typedef pair<int, int> i_i;
template<class T>
inline bool chmax(T &a, T b) {
    if(a < b) {
        a = b;
        return true;
    }
    return false;
}

template<class T>
inline bool chmin(T &a, T b) {
    if(a > b) {
        a = b;
        return true;
    }
    return false;
}

const ll INF = 1e9;
//const ll mod = 1000000007;

std::ostream &operator<<(std::ostream &dest, __int128_t value) {
  std::ostream::sentry s(dest);
  if (s) {
    __uint128_t tmp = value < 0 ? -value : value;
    char buffer[128];
    char *d = std::end(buffer);
    do {
      --d;
      *d = "0123456789"[tmp % 10];
      tmp /= 10;
    } while (tmp != 0);
    if (value < 0) {
      --d;
      *d = '-';
    }
    int len = std::end(buffer) - d;
    if (dest.rdbuf()->sputn(d, len) != len) {
      dest.setstate(std::ios_base::badbit);
    }
  }
  return dest;
}

__int128 parse(string &s) {
  __int128 ret = 0;
  for (int i = 0; i < s.length(); i++)
    if ('0' <= s[i] && s[i] <= '9')
      ret = 10 * ret + s[i] - '0';
  return ret;
}

struct LazySegmentTree {
private:
    int n;
    vector<__int128_t> node, oddnum, evennum, Mod, Add;
 
public:
    LazySegmentTree() {
        int sz = 100500;
        n = 1; while(n < sz) n *= 2;
        node.resize(2*n-1);
        oddnum.resize(2*n-1, 0);
        evennum.resize(2*n-1, 1);
        Add.resize(2*n-1, 0);
        Mod.resize(2*n-1, 0);
        for(int i=0; i<sz; i++) {
            node[i+n-1] = 0;
            evennum[i+n-1] = 1;
            oddnum[i+n-1] = 0;
        }
        for(int i=n-2; i>=0; i--) {
            node[i] = (node[i*2+1] + node[i*2+2]);
            oddnum[i] = (oddnum[i*2+1] + oddnum[i*2+2]);
            evennum[i] = (evennum[i*2+1] + evennum[i*2+2]);
        }
    }
 
    void lazyEvaluate(int k, ll l, ll r) {
        //cerr << "evaluating: " << k << " " << l << " " << r << endl;
        if(Mod[k] != 0) {
            if(Mod[k] == 1) {
                node[k] = oddnum[k];
            } else if(Mod[k] == 2) {
                node[k] = evennum[k];
                swap(oddnum[k], evennum[k]);
            } else exit(0);
            if(r - l > 1) {
                for(int j = k*2+1; j <= k*2+2; j++) {
                    if(Add[j] % 2 == 1) {
                        if(Mod[j] == 0 || Mod[j] == 1) {
                            Mod[j] = 3 - Mod[k];
                        } else if(Mod[j] == 2) {
                            Mod[j] = Mod[k];
                        }
                    } else {
                        if(Mod[j] == 0 || Mod[j] == 1) {
                            Mod[j] = Mod[k];
                        } else if(Mod[j] == 2) {
                            Mod[j] = 3 - Mod[k];
                        }
                    }
                    Add[j] = 0;
                }
            }
            Mod[k] = 0;
        }
        if(Add[k]) {
            node[k] += Add[k] * (r - l);
            if(Add[k] % 2 == 1) {
                swap(evennum[k], oddnum[k]);
            }
            if(r - l > 1) {
                Add[k*2+1] += Add[k];
                Add[k*2+1] += Add[k];
            }
            Add[k] = 0;
        }
    }
 
    void setmod(int a, int b, int k=0, int l=0, int r=-1) {
        if(r < 0) r = n;
        lazyEvaluate(k, l, r);
        if(b <= l || r <= a) return;
        if(a <= l && r <= b) {
            Mod[k] = 1;
            lazyEvaluate(k, l, r);
        }
        else {
            setmod(a, b, 2*k+1, l, (l+r)/2);
            setmod(a, b, 2*k+2, (l+r)/2, r);
            node[k] = (node[2*k+1] + node[2*k+2]);
            oddnum[k] = oddnum[2*k+1] + oddnum[2*k+2];
            evennum[k] = evennum[2*k+1] + evennum[2*k+2];
        }
    }
 
    void add(int a, int b, ll x, int k=0, int l=0, int r=-1) {
        if(r < 0) r = n;
        lazyEvaluate(k, l, r);
        if(b <= l || r <= a) return;
        if(a <= l && r <= b) {
            Add[k] += x;
            lazyEvaluate(k, l, r);
        }
        else {
            add(a, b, x, 2*k+1, l, (l+r)/2);
            add(a, b, x, 2*k+2, (l+r)/2, r);
            node[k] = (node[2*k+1] + node[2*k+2]);
            oddnum[k] = oddnum[2*k+1] + oddnum[2*k+2];
            evennum[k] = evennum[2*k+1] + evennum[2*k+2];
        }
    }

    ll getsum(int a, int b, int k=0, int l=0, int r=-1) {
        if(r < 0) r = n;
        lazyEvaluate(k, l, r);
        if(b <= l || r <= a) return 0;
        if(a <= l && r <= b) return node[k];
        ll vl = getsum(a, b, 2*k+1, l, (l+r)/2);
        ll vr = getsum(a, b, 2*k+2, (l+r)/2, r);
        return (vl + vr);
    }

    void print(int b) {
        for(int i = 0; i <= b + n - 1; i++) {
            cerr << i << " " << node[i] << " " << Add[i] << " " << oddnum[i] << " " << evennum[i] << " " << Mod[i] << endl;
        }
    }
};


int main() {
    //cout.precision(10);
    cin.tie(0);
    ios::sync_with_stdio(false);
    int N, Q;
    cin >> N >> Q;
    LazySegmentTree seg;
    for(int i = 1; i <= N; i++) {
        int a;
        cin >> a;
        seg.add(i, i + 1, a);
        //cerr << "array: " << i << " " << a << endl;
    }
    for(int q = 1; q <= Q; q++) {
        /*
        cerr << "--------------" << endl;
        seg.print(N);
        cerr << "--------------" << endl;
        */
        int ope;
        cin >> ope;
        //cerr << q << " " << ope << endl;
        if(ope == 1) {
            int l, r;
            cin >> l >> r;
            //cerr << "one: " << l << " " << r << endl;
            seg.setmod(l, r + 1);
            continue;
        } else if(ope == 2) {
            int l, r, x;
            cin >> l >> r >> x;
            seg.add(l, r + 1, x);
            continue;
        } else if(ope == 3) {
            //cerr << "A" << endl;
            int l, r;
            cin >> l >> r;
            cout << seg.getsum(l, r + 1) << endl;
        }
    }
    //seg.print(N);
    return 0;
}
0