結果

問題 No.230 Splarraay スプラレェーイ
ユーザー te-shte-sh
提出日時 2017-05-29 11:02:48
言語 D
(dmd 2.106.1)
結果
MLE  
実行時間 -
コード長 2,523 bytes
コンパイル時間 993 ms
コンパイル使用メモリ 101,908 KB
実行使用メモリ 812,296 KB
最終ジャッジ日時 2024-06-12 19:29:22
合計ジャッジ時間 5,195 ms
ジャッジサーバーID
(参考情報)
judge2 / judge5
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
6,816 KB
testcase_01 AC 1 ms
6,944 KB
testcase_02 AC 1 ms
6,944 KB
testcase_03 AC 1 ms
6,940 KB
testcase_04 AC 1 ms
6,940 KB
testcase_05 AC 15 ms
11,728 KB
testcase_06 AC 552 ms
174,648 KB
testcase_07 AC 10 ms
8,964 KB
testcase_08 AC 17 ms
6,944 KB
testcase_09 MLE -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
権限があれば一括ダウンロードができます

ソースコード

diff #

import std.algorithm, std.conv, std.range, std.stdio, std.string;

alias SegmentTreeLazy!(int, 0, "a + b") SegTree;

void main()
{
  auto n = readln.chomp.to!size_t;
  auto q = readln.chomp.to!size_t;

  auto sa = SegTree(n), sb = SegTree(n);
  auto pa = 0L, pb = 0L;

  foreach (_; 0..q) {
    auto rd = readln.split, x = rd[0], l = rd[1].to!size_t, r = rd[2].to!size_t + 1;
    switch (x) {
    case "0":
      auto ca = sa[l..r], cb = sb[l..r];
      if (ca > cb) pa += ca;
      if (ca < cb) pb += cb;
      break;
    case "1":
      sa[l..r] = 1;
      sb[l..r] = 0;
      break;
    case "2":
      sa[l..r] = 0;
      sb[l..r] = 1;
      break;
    default:
      assert(0);
    }
  }

  pa += sa[0..$];
  pb += sb[0..$];

  writeln(pa, " ", pb);
}

struct SegmentTreeLazy(T, T unit, alias pred = "a + b")
{
  import core.bitop, std.conv, std.functional, std.range;
  alias predFun = binaryFun!pred;
  alias void delegate(ref T, size_t) Ope;

  const size_t n, an;
  T[] buf;
  Ope[][] opes;

  this(size_t n)
  {
    this.n = n;
    an = (1 << ((n - 1).bsr + 1));
    buf = new int[](an * 2);
    opes = new Ope[][](an * 2);
    static if (T.init != unit) {
      buf[] = unit;
    }
  }

  void propagate(size_t k, size_t nl, size_t nr)
  {
    if (opes[k].empty) return;

    size_t nm = (nl + nr) / 2;
    setLazy(opes[k], k*2,   nl, nm);
    setLazy(opes[k], k*2+1, nm, nr);

    opes[k] = [];
  }

  void setLazy(Ope[] ope, size_t k, size_t nl, size_t nr)
  {
    foreach (op; ope) op(buf[k], nr - nl);
    opes[k] ~= ope;
  }

  void addOpe(Ope op, size_t l, size_t r, size_t k, size_t nl, size_t nr)
  {
    if (nr <= l || r <= nl) return;

    if (l <= nl && nr <= r) {
      setLazy([op], k, nl, nr);
      return;
    }
    
    propagate(k, nl, nr);

    auto nm = (nl + nr) / 2;
    addOpe(op, l, r, k*2,   nl, nm);
    addOpe(op, l, r, k*2+1, nm, nr);

    buf[k] = predFun(buf[k*2], buf[k*2+1]);
  }

  void opSliceAssign(T val, size_t l, size_t r)
  {
    Ope op = (ref buf, sz) { buf = val * sz.to!T; };
    addOpe(op, l, r, 1, 0, an);
  }

  int summary(size_t l, size_t r, size_t k, size_t nl, size_t nr)
  {
    if (nr <= l || r <= nl) return 0;

    if (l <= nl && nr <= r) return buf[k];

    propagate(k, nl, nr);

    auto nm = (nl + nr) / 2;
    auto vl = summary(l, r, k*2,   nl, nm);
    auto vr = summary(l, r, k*2+1, nm, nr);

    return predFun(vl, vr);
  }

  int opSlice(size_t l, size_t r)
  {
    return summary(l, r, 1, 0, an);
  }

  pure size_t opDollar() const { return n; }
}
0