結果

問題 No.879 Range Mod 2 Query
コンテスト
ユーザー tnakao0123
提出日時 2019-09-09 13:45:01
言語 C++11（廃止可能性あり）
(gcc 13.3.0 + boost 1.89.0)
結果
AC  
実行時間 165 ms / 3,000 ms
コード長 4,691 bytes
記録
コンパイル時間 1,376 ms
コンパイル使用メモリ 92,420 KB
実行使用メモリ 10,624 KB
最終ジャッジ日時 2024-06-28 11:57:15
合計ジャッジ時間 4,689 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 21
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:177:8: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
  177 |   scanf("%d%d", &n, &qn);
      |   ~~~~~^~~~~~~~~~~~~~~~~
main.cpp:183:10: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
  183 |     scanf("%d", &ai);
      |     ~~~~~^~~~~~~~~~~
main.cpp:190:10: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
  190 |     scanf("%d%d%d", &c, &l, &r);
      |     ~~~~~^~~~~~~~~~~~~~~~~~~~~~
main.cpp:199:12: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
  199 |       scanf("%d", &x);
      |       ~~~~~^~~~~~~~~~

ソースコード

diff #
raw source code 

/* -*- coding: utf-8 -*-
 *
 * 879.cc:  No.879 Range Mod 2 Query  - yukicoder
 */

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<iostream>
#include<string>
#include<vector>
#include<map>
#include<set>
#include<stack>
#include<list>
#include<queue>
#include<deque>
#include<algorithm>
#include<numeric>
#include<utility>
#include<complex>
#include<functional>
 
using namespace std;

/* constant */

const int MAX_N = 100000;
const int MAX_E2 = 1 << 18; // = 262144

/* typedef */

typedef long long ll;

template <typename T, const int MAX_E2>
struct SegTreeSum {
  int n, e2;
  T nodes[MAX_E2], dlxs[MAX_E2], dlys[MAX_E2];
  int ods[MAX_E2], evs[MAX_E2];
  bool dlcs[MAX_E2];
  SegTreeSum() {}

  void init(int _n) {
    n = _n;
    for (e2 = 1; e2 < n; e2 <<= 1);
  }

  T &get(int i) { return nodes[e2 - 1 + i]; }

  void set(int i, T v) {
    int j = e2 - 1 + i;
    nodes[j] = v;
    if (v & 1) ods[j] = 1;
    else evs[j] = 1;
  }

  void setall() {
    for (int j = e2 - 2; j >= 0; j--) {
      int j0 = j * 2 + 1, j1 = j0 + 1;
      nodes[j] = nodes[j0] + nodes[j1];
      ods[j] = ods[j0] + ods[j1];
      evs[j] = evs[j0] + evs[j1];
    }
  }

  void _update(int k, int l) {
    int k0 = k * 2 + 1, k1 = k0 + 1, lh = (l >> 1);
    if (dlcs[k]) {
      if (dlxs[k] & 1)
	swap(ods[k0], evs[k0]), swap(ods[k1], evs[k1]);
      nodes[k0] = ods[k0], nodes[k1] = ods[k1];
      dlxs[k0] += dlys[k0] + dlxs[k];
      dlxs[k1] += dlys[k1] + dlxs[k];
      dlys[k0] = dlys[k1] = 0;
      dlcs[k0] = dlcs[k1] = true;
      dlxs[k] = 0;
      dlcs[k] = false;
    }

    nodes[k0] += dlys[k] * lh;
    nodes[k1] += dlys[k] * lh;
    if (dlys[k] & 1)
      swap(ods[k0], evs[k0]), swap(ods[k1], evs[k1]);
    dlys[k0] += dlys[k], dlys[k1] += dlys[k];
    dlys[k] = 0;
  }

  void updateall(int k, int i0, int i1) {
    if (i0 + 1 < i1) {
      _update(k, i1 - i0);
      int im = (i0 + i1) / 2;
      updateall(k * 2 + 1, i0, im);
      updateall(k * 2 + 2, im, i1);
    }
  }
  void updateall() { updateall(0, 0, e2); }

  void printall() {
    for (int k = 0, bk = 1; bk <= e2; k++, bk <<= 1) {
      for (int i = bk - 1; i < bk * 2 - 1; i++)
	printf("%lld(%d,%d)(%lld,%lld,%d) ",
	       nodes[i], ods[i], evs[i], dlxs[i], dlys[i], dlcs[i]);
      putchar('\n');
    }
  }

  void clear_range(int r0, int r1, int k, int i0, int i1) {
    if (r1 <= i0 || i1 <= r0) return;
    if (r0 <= i0 && i1 <= r1) {
      nodes[k] = ods[k];
      dlxs[k] += dlys[k];
      dlys[k] = 0;
      dlcs[k] = true;
      return;
    }

    _update(k, i1 - i0);

    int im = (i0 + i1) / 2;
    int k0 = k * 2 + 1, k1 = k0 + 1;
    clear_range(r0, r1, k0, i0, im);
    clear_range(r0, r1, k1, im, i1);

    nodes[k] = nodes[k0] + nodes[k1];
    ods[k] = ods[k0] + ods[k1];
    evs[k] = evs[k0] + evs[k1];
  }
  void clear_range(int r0, int r1) { clear_range(r0, r1, 0, 0, e2); }

  void add_range(int r0, int r1, T v, int k, int i0, int i1) {
    if (r1 <= i0 || i1 <= r0) return;
    if (r0 <= i0 && i1 <= r1) {
      nodes[k] += v * (i1 - i0);
      if (v & 1) swap(ods[k], evs[k]);
      dlys[k] += v;
      return;
    }

    _update(k, i1 - i0);

    int im = (i0 + i1) / 2;
    int k0 = k * 2 + 1, k1 = k0 + 1;
    add_range(r0, r1, v, k0, i0, im);
    add_range(r0, r1, v, k1, im, i1);

    nodes[k] = nodes[k0] + nodes[k1];
    ods[k] = ods[k0] + ods[k1];
    evs[k] = evs[k0] + evs[k1];
  }
  void add_range(int r0, int r1, T v) { add_range(r0, r1, v, 0, 0, e2); }

  T sum_range(int r0, int r1, int k, int i0, int i1) {
    if (r1 <= i0 || i1 <= r0) return 0;
    if (r0 <= i0 && i1 <= r1) return nodes[k];

    _update(k, i1 - i0);

    int im = (i0 + i1) / 2;
    T v0 = sum_range(r0, r1, k * 2 + 1, i0, im);
    T v1 = sum_range(r0, r1, k * 2 + 2, im, i1);
    return v0 + v1;
  }
  T sum_range(int r0, int r1) { return sum_range(r0, r1, 0, 0, e2); }
};

/* global variables */

SegTreeSum<ll,MAX_E2> st;

/* subroutines */

/* main */

int main() {
  int n, qn;
  scanf("%d%d", &n, &qn);

  st.init(n);

  for (int i = 0; i < n; i++) {
    int ai;
    scanf("%d", &ai);
    st.set(i, ai);
  }
  st.setall();

  while (qn--) {
    int c, l, r;
    scanf("%d%d%d", &c, &l, &r);
    l--;

    if (c == 1) {
      //printf("1 %d %d\n", l, r);
      st.clear_range(l, r);
    }
    else if (c == 2) {
      int x;
      scanf("%d", &x);
      //printf("2 %d %d %d\n", l, r, x);
      st.add_range(l, r, x);
    }
    else {
      //printf("3 %d %d\n", l, r);
      ll sum = st.sum_range(l, r);
      printf("%lld\n", sum);
    }

    //st.updateall();
    //st.printall();
    //for (int i = 0; i < n; i++) printf("%lld ", st.get(i)); putchar('\n');
  }

  return 0;
}
0