結果

問題 No.619 CardShuffle
ユーザー 👑 hos.lyrichos.lyric
提出日時 2019-01-21 07:44:57
言語 D
(dmd 2.106.1)
結果
AC  
実行時間 215 ms / 3,000 ms
コード長 5,250 bytes
コンパイル時間 1,732 ms
コンパイル使用メモリ 162,084 KB
実行使用メモリ 19,496 KB
最終ジャッジ日時 2023-09-03 22:46:52
合計ジャッジ時間 7,116 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,380 KB
testcase_01 AC 2 ms
4,380 KB
testcase_02 AC 1 ms
4,376 KB
testcase_03 AC 1 ms
4,388 KB
testcase_04 AC 2 ms
4,376 KB
testcase_05 AC 1 ms
4,380 KB
testcase_06 AC 2 ms
4,384 KB
testcase_07 AC 2 ms
4,376 KB
testcase_08 AC 1 ms
4,376 KB
testcase_09 AC 1 ms
4,384 KB
testcase_10 AC 1 ms
4,376 KB
testcase_11 AC 1 ms
4,384 KB
testcase_12 AC 1 ms
4,380 KB
testcase_13 AC 2 ms
4,376 KB
testcase_14 AC 2 ms
4,380 KB
testcase_15 AC 2 ms
4,380 KB
testcase_16 AC 206 ms
18,056 KB
testcase_17 AC 201 ms
18,196 KB
testcase_18 AC 205 ms
17,564 KB
testcase_19 AC 197 ms
17,520 KB
testcase_20 AC 132 ms
18,672 KB
testcase_21 AC 211 ms
18,364 KB
testcase_22 AC 181 ms
18,036 KB
testcase_23 AC 211 ms
19,132 KB
testcase_24 AC 180 ms
19,496 KB
testcase_25 AC 122 ms
18,468 KB
testcase_26 AC 197 ms
18,532 KB
testcase_27 AC 212 ms
18,200 KB
testcase_28 AC 197 ms
18,124 KB
testcase_29 AC 215 ms
18,536 KB
testcase_30 AC 143 ms
18,152 KB
testcase_31 AC 1 ms
4,380 KB
testcase_32 AC 133 ms
19,212 KB
testcase_33 AC 205 ms
18,608 KB
testcase_34 AC 171 ms
18,540 KB
testcase_35 AC 99 ms
14,592 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std.conv, std.stdio, std.string;
import std.algorithm, std.array, std.bigint, std.container, std.math, std.numeric, std.range, std.regex, std.typecons;
import core.bitop;

class EOFException : Throwable { this() { super("EOF"); } }
string[] tokens;
string readToken() { for (; tokens.empty; ) { if (stdin.eof) { throw new EOFException; } tokens = readln.split; } auto token = tokens.front; tokens.popFront; return token; }
int readInt() { return readToken.to!int; }
long readLong() { return readToken.to!long; }
real readReal() { return readToken.to!real; }

void chmin(T)(ref T t, in T f) { if (t > f) t = f; }
void chmax(T)(ref T t, in T f) { if (t < f) t = f; }

int binarySearch(T)(in T[] as, in bool delegate(T) test) { int low = -1, upp = cast(int)(as.length); for (; low + 1 < upp; ) { int mid = (low + upp) >> 1; (test(as[mid]) ? low : upp) = mid; } return upp; }
int lowerBound(T)(in T[] as, in T val) { return as.binarySearch((T a) => (a < val)); }
int upperBound(T)(in T[] as, in T val) { return as.binarySearch((T a) => (a <= val)); }


enum MO = 10L^^9 + 7;

class SegmentTree(string Op) if (Op == "+" || Op == "*") {
  int n;
  long[] prod;
  this(long[] ini) {
    for (n = 1; n < ini.length; n <<= 1) {}
    prod = new long[n << 1];
    foreach (i, val; ini) {
      prod[n + i] = val;
    }
    foreach_reverse (a; 1 .. n) {
      prod[a] = mixin(`(prod[a << 1] ` ~ Op ~ ` prod[a << 1 | 1]) % MO`);
    }
  }
  void update(int a, long val) {
    prod[a += n] = val;
    for (a >>= 1; a; a >>= 1) {
      prod[a] = mixin(`(prod[a << 1] ` ~ Op ~ ` prod[a << 1 | 1]) % MO`);
    }
  }
  long rangeProd(int a, int b) {
    long ret = (Op == "+") ? 0 : 1;
    for (a += n, b += n; a <= b; a >>= 1, b >>= 1) {
      if ( a & 1) mixin(`(ret ` ~ Op ~ `= prod[a++]) %= MO;`);
      if (~b & 1) mixin(`(ret ` ~ Op ~ `= prod[b--]) %= MO;`);
    }
    return ret;
  }
}

/*
  C:  0 1 2 3
  O: 0 1 2 3 4
*/

int N;
long[] C;
string[] O;
int Q;
string[] T;
int[] X, Y;

void main() {
  try {
    for (; ; ) {
      N = readInt() / 2 + 1;
      C = new long[N];
      O = new string[N + 1];
      C[0] = readLong();
      foreach (i; 1 .. N) {
        O[i] = readToken();
        C[i] = readLong();
      }
      O[0] = O[N] = "+";
      Q = readInt();
      T = new string[Q];
      X = new int[Q];
      Y = new int[Q];
      foreach (q; 0 .. Q) {
        T[q] = readToken();
        X[q] = readInt();
        Y[q] = readInt();
      }
      
      auto segProd = new SegmentTree!"*"(C);
      auto sums = new long[N];
      auto pluses = new RedBlackTree!int();
      int bef = -1;
      foreach (i; 0 .. N + 1) {
        if (O[i] == "+") {
          pluses.insert(i);
          if (bef != -1) {
            sums[bef] = segProd.rangeProd(bef, i - 1);
          }
          bef = i;
        }
      }
      auto segSum = new SegmentTree!"+"(sums);
      debug {
        writeln("sums = ", sums);
      }
      
      void updateC(int i) {
        debug {
          writeln("updateC ", i);
        }
        segProd.update(i, C[i]);
        const lb = pluses.lowerBound(i + 1).back;
        const ub = pluses.upperBound(i).front;
        sums[lb] = segProd.rangeProd(lb, ub - 1);
        segSum.update(lb, sums[lb]);
      }
      void updateOp(int i) {
        debug {
          writeln("updateOp ", i, " ", O[i]);
        }
        const lb = pluses.lowerBound(i).back;
        const ub = pluses.upperBound(i).front;
        switch (O[i]) {
          case "+": {
            // "*" -> "+"
            pluses.insert(i);
            sums[lb] = segProd.rangeProd(lb, i - 1);
            sums[i] = segProd.rangeProd(i, ub - 1);
          } break;
          case "*": {
            // "+" -> "*"
            pluses.removeKey(i);
            sums[lb] = segProd.rangeProd(lb, ub - 1);
            sums[i] = 0;
          } break;
          default: assert(false);
        }
        segSum.update(lb, sums[lb]);
        segSum.update(i, sums[i]);
      }
      foreach (q; 0 .. Q) {
        switch (T[q]) {
          case "!": {
            if (X[q] % 2 == 0) {
              if (O[X[q] / 2] != O[Y[q] / 2]) {
                swap(O[X[q] / 2], O[Y[q] / 2]);
                updateOp(X[q] / 2);
                updateOp(Y[q] / 2);
              }
            } else {
              swap(C[X[q] / 2], C[Y[q] / 2]);
              updateC(X[q] / 2);
              updateC(Y[q] / 2);
            }
          } break;
          case "?": {
            const l = pluses.upperBound(X[q] / 2).front;
            const r = pluses.lowerBound(Y[q] / 2 + 1).back;
            debug {
              writeln("? ", X[q] / 2, " ", Y[q] / 2);
              writeln("  l = ", l, ", r = ", r);
            }
            long ans;
            if (l <= r) {
              ans += segProd.rangeProd(X[q] / 2, l - 1);
              ans += segSum.rangeProd(l, r - 1);
              ans += segProd.rangeProd(r, Y[q] / 2);
            } else {
              ans += segProd.rangeProd(X[q] / 2, Y[q] / 2);
            }
            ans %= MO;
            writeln(ans);
          } break;
          default: assert(false);
        }
        debug {
          writeln("C = ", C);
          writeln("pluses = ", pluses);
        }
      }
    }
  } catch (EOFException e) {
  }
}
0