結果

問題 No.1030 だんしんぐぱーりない
ユーザー 👑 hos.lyrichos.lyric
提出日時 2020-04-17 22:12:03
言語 D
(dmd 2.106.1)
結果
AC  
実行時間 272 ms / 2,000 ms
コード長 5,274 bytes
コンパイル時間 813 ms
コンパイル使用メモリ 127,820 KB
実行使用メモリ 31,752 KB
最終ジャッジ日時 2024-06-22 06:40:48
合計ジャッジ時間 10,806 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
6,816 KB
testcase_01 AC 1 ms
6,816 KB
testcase_02 AC 1 ms
6,940 KB
testcase_03 AC 2 ms
6,944 KB
testcase_04 AC 1 ms
6,940 KB
testcase_05 AC 207 ms
29,304 KB
testcase_06 AC 170 ms
30,132 KB
testcase_07 AC 116 ms
18,456 KB
testcase_08 AC 121 ms
13,620 KB
testcase_09 AC 168 ms
27,312 KB
testcase_10 AC 85 ms
7,172 KB
testcase_11 AC 193 ms
13,368 KB
testcase_12 AC 195 ms
23,616 KB
testcase_13 AC 151 ms
16,820 KB
testcase_14 AC 211 ms
26,980 KB
testcase_15 AC 108 ms
11,604 KB
testcase_16 AC 192 ms
22,828 KB
testcase_17 AC 177 ms
19,972 KB
testcase_18 AC 227 ms
20,592 KB
testcase_19 AC 141 ms
7,164 KB
testcase_20 AC 159 ms
20,360 KB
testcase_21 AC 141 ms
21,476 KB
testcase_22 AC 158 ms
20,476 KB
testcase_23 AC 209 ms
10,952 KB
testcase_24 AC 151 ms
11,444 KB
testcase_25 AC 165 ms
23,520 KB
testcase_26 AC 125 ms
13,752 KB
testcase_27 AC 133 ms
7,476 KB
testcase_28 AC 202 ms
23,372 KB
testcase_29 AC 140 ms
20,780 KB
testcase_30 AC 142 ms
20,676 KB
testcase_31 AC 174 ms
11,200 KB
testcase_32 AC 177 ms
31,752 KB
testcase_33 AC 185 ms
18,440 KB
testcase_34 AC 88 ms
7,240 KB
testcase_35 AC 268 ms
29,616 KB
testcase_36 AC 272 ms
29,380 KB
testcase_37 AC 265 ms
30,224 KB
testcase_38 AC 267 ms
29,524 KB
testcase_39 AC 263 ms
30,628 KB
testcase_40 AC 1 ms
6,940 KB
testcase_41 AC 1 ms
6,940 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std.conv, std.functional, std.range, std.stdio, std.string;
import std.algorithm, std.array, std.bigint, std.bitmanip, std.complex, std.container, std.math, std.mathspecial, std.numeric, 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; }

bool chmin(T)(ref T t, in T f) { if (t > f) { t = f; return true; } else { return false; } }
bool chmax(T)(ref T t, in T f) { if (t < f) { t = f; return true; } else { return false; } }

int binarySearch(alias pred, T)(in T[] as) { int lo = -1, hi = cast(int)(as.length); for (; lo + 1 < hi; ) { const mid = (lo + hi) >> 1; (unaryFun!pred(as[mid]) ? hi : lo) = mid; } return hi; }
int lowerBound(T)(in T[] as, T val) { return as.binarySearch!(a => (a >= val)); }
int upperBound(T)(in T[] as, T val) { return as.binarySearch!(a => (a > val)); }

// T: monoid
// op: T * T -> T
// query(a, b): returns t_a ... t_{b-1}
class SegmentTree(T, alias op) {
  import std.functional : binaryFun;
  alias opFun = binaryFun!op;
  const(T) idT;

  int n;
  T[] ts;
  this(int n_, const(T) idT) {
    this.idT = idT;
    for (n = 1; n < n_; n <<= 1) {}
    ts = new T[n << 1];
    ts[] = idT;
  }
  this(inout(T)[] ts_, const(T) idT) {
    this.idT = idT;
    const n_ = cast(int)(ts_.length);
    for (n = 1; n < n_; n <<= 1) {}
    ts = new T[n << 1];
    ts[0 .. n] = idT;
    ts[n .. n + n_] = ts_[];
    ts[n + n_ .. n << 1] = idT;
    build();
  }
  ref T at(int a) {
    return ts[n + a];
  }
  void build() {
    foreach_reverse (a; 1 .. n) ts[a] = opFun(ts[a << 1], ts[a << 1 | 1]);
  }
  void set(int a, const(T) val) {
    ts[a += n] = val;
    for (; a >>= 1; ) ts[a] = opFun(ts[a << 1], ts[a << 1 | 1]);
  }
  void mulL(int a, const(T) val) {
    set(a, opFun(val, ts[a + n]));
  }
  void mulR(int a, const(T) val) {
    set(a, opFun(ts[a + n], val));
  }
  // T query(int a, int b) const {
  T query(int a, int b) {
    T prodL = idT, prodR = idT;
    for (a += n, b += n; a < b; a >>= 1, b >>= 1) {
      if (a & 1) prodL = opFun(prodL, ts[a++]);
      if (b & 1) prodR = opFun(ts[--b], prodR);
    }
    return opFun(prodL, prodR);
  }
}

int N, K, Q;
int[] C;
int[] A;
int[] U, V;

int[][] childrens;
int E;
int[][] par;
int[] dep;
int zeit;
int[] dis, fin;
int[] scores;

void dfs(int u, int p, int score) {
  par[0][u] = p;
  dep[u] = (p == -1) ? 0 : (dep[p] + 1);
  dis[u] = zeit++;
  scores[u] = score;
  foreach (v; childrens[u]) {
    dfs(v, u, max(score, C[v]));
  }
  fin[u] = zeit++;
}

int lca(int u, int v) {
  foreach_reverse (e; 0 .. E) {
    if (dep[u] - (1 << e) >= dep[v]) {
      u = par[e][u];
    }
    if (dep[v] - (1 << e) >= dep[u]) {
      v = par[e][v];
    }
  }
  foreach_reverse (e; 0 .. E) {
    if (par[e][u] != par[e][v]) {
      u = par[e][u];
      v = par[e][v];
    }
  }
  if (u != v) {
    u = par[0][u];
    v = par[0][v];
  }
  return u;
}

void main() {
  try {
    for (; ; ) {
      N = readInt();
      K = readInt();
      Q = readInt();
      C = new int[N];
      foreach (u; 0 .. N) {
        C[u] = readInt();
      }
      A = new int[K];
      foreach (k; 0 .. K) {
        A[k] = readInt() - 1;
      }
      U = new int[N - 1];
      V = new int[N - 1];
      foreach (i; 0 .. N - 1) {
        U[i] = readInt() - 1;
        V[i] = readInt() - 1;
      }
      
      childrens = new int[][N];
      foreach (i; 0 .. N - 1) {
        childrens[V[i]] ~= U[i];
      }
      E = bsr(N) + 1;
      par = new int[][](E, N);
      dep = new int[N];
      zeit = 0;
      dis = new int[N];
      fin = new int[N];
      scores = new int[N];
      dfs(0, -1, C[0]);
      par[0][0] = 0;
      foreach (e; 0 .. E - 1) {
        foreach (u; 0 .. N) {
          par[e + 1][u] = par[e][par[e][u]];
        }
      }
      
      alias Entry = Tuple!(int, "val", int, "key");
      auto segDis = new SegmentTree!(Entry, min)(K, Entry(2 * N, -1));
      auto segFin = new SegmentTree!(Entry, max)(K, Entry(-1, -1));
      auto as = A.dup;
      foreach (k; 0 .. K) {
        segDis.at(k) = Entry(dis[as[k]], as[k]);
        segFin.at(k) = Entry(fin[as[k]], as[k]);
      }
      segDis.build;
      segFin.build;
      foreach (q; 0 .. Q) {
        const typ = readInt();
        switch (typ) {
          case 1: {
            const k = readInt() - 1;
            const u = readInt() - 1;
            as[k] = u;
            segDis.set(k, Entry(dis[as[k]], as[k]));
            segFin.set(k, Entry(fin[as[k]], as[k]));
          } break;
          case 2: {
            const l = readInt() - 1;
            const r = readInt();
            const resDis = segDis.query(l, r);
            const resFin = segFin.query(l, r);
            debug {
              writeln("res = ", resDis, " ", resFin);
            }
            const ans = scores[lca(resDis.key, resFin.key)];
            writeln(ans);
          } break;
          default: assert(false);
        }
      }
    }
  } catch (EOFException e) {
  }
}
0