結果

問題 No.880 Yet Another Segment Tree Problem
ユーザー risujirohrisujiroh
提出日時 2020-09-11 17:02:51
言語 C++17
(gcc 13.2.0 + boost 1.83.0)
結果
AC  
実行時間 259 ms / 5,000 ms
コード長 3,369 bytes
コンパイル時間 7,463 ms
コンパイル使用メモリ 443,616 KB
実行使用メモリ 7,340 KB
最終ジャッジ日時 2023-10-24 02:49:35
合計ジャッジ時間 13,441 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge14
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,372 KB
testcase_01 AC 3 ms
4,372 KB
testcase_02 AC 3 ms
4,372 KB
testcase_03 AC 3 ms
4,372 KB
testcase_04 AC 3 ms
4,372 KB
testcase_05 AC 3 ms
4,372 KB
testcase_06 AC 3 ms
4,372 KB
testcase_07 AC 3 ms
4,372 KB
testcase_08 AC 3 ms
4,372 KB
testcase_09 AC 3 ms
4,372 KB
testcase_10 AC 3 ms
4,372 KB
testcase_11 AC 237 ms
7,340 KB
testcase_12 AC 232 ms
7,340 KB
testcase_13 AC 164 ms
7,340 KB
testcase_14 AC 223 ms
7,340 KB
testcase_15 AC 240 ms
7,340 KB
testcase_16 AC 252 ms
7,340 KB
testcase_17 AC 203 ms
7,340 KB
testcase_18 AC 190 ms
7,340 KB
testcase_19 AC 95 ms
7,340 KB
testcase_20 AC 98 ms
7,340 KB
testcase_21 AC 98 ms
7,340 KB
testcase_22 AC 98 ms
7,340 KB
testcase_23 AC 99 ms
7,340 KB
testcase_24 AC 87 ms
7,340 KB
testcase_25 AC 90 ms
7,340 KB
testcase_26 AC 89 ms
7,340 KB
testcase_27 AC 89 ms
7,340 KB
testcase_28 AC 91 ms
7,340 KB
testcase_29 AC 218 ms
7,340 KB
testcase_30 AC 238 ms
7,340 KB
testcase_31 AC 259 ms
7,340 KB
testcase_32 AC 89 ms
7,340 KB
testcase_33 AC 74 ms
7,340 KB
testcase_34 AC 78 ms
7,340 KB
testcase_35 AC 75 ms
7,340 KB
testcase_36 AC 76 ms
7,340 KB
testcase_37 AC 75 ms
7,340 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma GCC optimize("Ofast")
#pragma GCC target("avx2,bmi,bmi2,lzcnt")

#include <bits/extc++.h>
#include <x86intrin.h>

int64_t binary_gcd(int64_t a, int64_t b) {
  a = abs(a), b = abs(b);
  if (a == 0) return b;
  int k = _tzcnt_u64(a | b);
  a >>= _tzcnt_u64(a);
  while (b) {
    b >>= _tzcnt_u64(b);
    if (a > b) std::swap(a, b);
    b -= a;
  }
  return a << k;
}

struct beats {
  static constexpr int inf = 0x3f3f3f3f;

  struct node {
    int mx, lcm = 1;
    int64_t sum;

    friend node operator*(const node& a, const node& b) {
      int lcm = inf;
      if (std::max(a.lcm, b.lcm) < inf)
        lcm = std::min<int64_t>(a.lcm / binary_gcd(a.lcm, b.lcm) * b.lcm, inf);
      return {std::max(a.mx, b.mx), lcm, a.sum + b.sum};
    }
  };

  int sz;
  std::vector<node> tree;

  beats(int n) {
    for (sz = 1; sz < n;) sz *= 2;
    tree.resize(2 * sz);
    for (int i = 0; i < n; ++i) {
      int a;
      std::cin >> a;
      tree[sz + i] = {a, a, a};
    }
    for (int i = sz; i-- > 1;) pull(i);
  }

  void pull(int i) { tree[i] = tree[2 * i] * tree[2 * i + 1]; }
  void push(int i, int j, int64_t len) {
    if (tree[i].mx * len == tree[i].sum)
      tree[j] = {tree[i].mx, tree[i].lcm, tree[i].sum >> 1};
  }
  void assign(int l, int r, int x, int i, int a, int b) {
    if (r <= a or b <= l) return;
    if (l <= a and b <= r) {
      tree[i].mx = tree[i].lcm = x;
      tree[i].sum = tree[i].mx * int64_t(b - a);
      return;
    }
    push(i, 2 * i, b - a), push(i, 2 * i + 1, b - a);
    int m = (a + b) >> 1;
    assign(l, r, x, 2 * i, a, m), assign(l, r, x, 2 * i + 1, m, b);
    pull(i);
  }
  void assign(int l, int r, int x) { assign(l, r, x, 1, 0, sz); }
  void gcd(int l, int r, int x, int i, int a, int b) {
    if (r <= a or b <= l or x % tree[i].lcm == 0) return;
    if (l <= a and b <= r and tree[i].mx * int64_t(b - a) == tree[i].sum) {
      tree[i].mx = tree[i].lcm = binary_gcd(tree[i].mx, x);
      tree[i].sum = tree[i].mx * int64_t(b - a);
      return;
    }
    push(i, 2 * i, b - a), push(i, 2 * i + 1, b - a);
    int m = (a + b) >> 1;
    gcd(l, r, x, 2 * i, a, m), gcd(l, r, x, 2 * i + 1, m, b);
    pull(i);
  }
  void gcd(int l, int r, int x) { gcd(l, r, x, 1, 0, sz); }
  int max(int l, int r, int i, int a, int b) {
    if (r <= a or b <= l) return 0;
    if (l <= a and b <= r) return tree[i].mx;
    push(i, 2 * i, b - a), push(i, 2 * i + 1, b - a);
    int m = (a + b) >> 1;
    return std::max(max(l, r, 2 * i, a, m), max(l, r, 2 * i + 1, m, b));
  }
  int max(int l, int r) { return max(l, r, 1, 0, sz); }
  int64_t sum(int l, int r, int i, int a, int b) {
    if (r <= a or b <= l) return 0;
    if (l <= a and b <= r) return tree[i].sum;
    push(i, 2 * i, b - a), push(i, 2 * i + 1, b - a);
    int m = (a + b) >> 1;
    return sum(l, r, 2 * i, a, m) + sum(l, r, 2 * i + 1, m, b);
  }
  int64_t sum(int l, int r) { return sum(l, r, 1, 0, sz); }
};

int main() {
  using namespace std;
  cin.tie(nullptr)->sync_with_stdio(false);
  int n, q;
  cin >> n >> q;
  beats b(n);
  while (q--) {
    int t, l, r;
    cin >> t >> l >> r;
    --l;
    if (t == 1) {
      int x;
      cin >> x;
      b.assign(l, r, x);
    } else if (t == 2) {
      int x;
      cin >> x;
      b.gcd(l, r, x);
    } else if (t == 3) {
      cout << b.max(l, r) << '\n';
    } else {
      cout << b.sum(l, r) << '\n';
    }
  }
}
0