Stage 4: Grand Prix of Korea (2021/10/24 17:00-22:00)

コンテストページ

official.contest.yandex.ru

問題概要

A 問題: Automatic Sprayer 2

実行時間制限: 2 秒 / メモリ制限: 1 GB

 n \times n の各成分が非負整数の行列  A がある。 A から、 n \times n の行列  E を以下のように作った。

  •  E _ {ij} = \displaystyle{\sum _ {x=1}^n \sum_{y=1}^n (|x-i|+|y-j|)A _ {xy}} \ (1 \leq i \leq n, 1 \leq j \leq n)

 E が与えられるので、 A としてありうるものを  1 つ出力せよ。

入力は以下の制約を満たす。

  •  2 \leq n \leq 1000
  •  0 \leq E _ {ij} \leq 10^{16} \ (1 \leq i \leq n, 1 \leq j \leq n)
  • 条件を満たす行列  A で各成分の和が  10^{12} 以下であるものが存在する。

B 問題: Cilantro

実行時間制限: 0.5 秒 / メモリ制限: 1 GB

Y と N のみからなる長さ  n の文字列  S, T が与えられる。

スタック  s を用意し、 i=1,2,\cdots,n の各文字について以下の操作を順に行う。

  •  S の先頭の文字を削除しそれを  s に push する。
  •  s を pop する。ただしこのとき取り出す文字は  T _ i に一致しなければならない。

操作  1 で取り出した文字が  S _ i に対応することがありうるような整数  i をすべて列挙し、それらの和を答えよ。ただし、操作を行うことができない場合は  0 と出力せよ。

入力は以下の制約を満たす。

  •  1 \leq n \leq 5000000
  •  S, T の長さは  n で Y と N のみからなる。

C 問題: Equivalent Pipelines

実行時間制限: 3 秒 / メモリ制限: 1 GB

 n 頂点の辺に重みが付いた木  T _ 1, T _ 2, \cdots, T _ d が与えられる。

 T の異なる頂点  i, j について、 v _ T(i,j) を、頂点  i, j を結ぶパス内の辺の重みの最小値とする。また、 2 つの木  T _ 1, T _ 2 v _ {T _ 1}(i,j) = v _ {T _ 2}(i,j) \ (1 \leq i \gt j \leq n) を満たすとき、木  T _ 1, T _ 2 は同等であるという。

 i \ (1 \leq i \leq d) について、木  T _ i が木  T _ j と同等であるような最小の  j を出力せよ。

入力は以下の制約を満たす。

  •  d \geq 1
  •  n \geq 2
  •  dn \leq 500000
  • 辺の重みは  1 以上  10 ^ 9 以下

D 問題: Flowerbed Redecoration

実行時間制限: 1 秒 / メモリ制限: 1 GB

 n m 列のグリッドがあり、各マスには大文字のアルファベットが書かれている。

 i = 1, 1 + y, 1 + 2y, \cdots, n - d + 1, j = 1, 1+x, 1+2x, \cdots, m-d+1 について、以下の操作を行う。

  • マス  (i,j) を左上とする正方形の各マスの文字を、正方形を時計回りに  90 度回転させた位置に移動させる。

操作を行った後のグリッドの各マスの文字を出力せよ。

入力は以下の制約を満たす。

  •  1 \leq n \times m \leq 10 ^ 6
  •  1 \leq y \leq n
  •  1 \leq x \leq m
  •  1 \leq d \leq \min(n, m)
  •  n \equiv d \ \pmod y
  •  m \equiv d \ \pmod x

E 問題: Goose Coins

実行時間制限: 3 秒 / メモリ制限: 1 GB

 n 種類のコインがあり、 i \ (1 \leq i \leq n) 番目のコインの価値は  c _ i 円、重さは  w _ i である。 i \ (1 \leq i \leq n - 1) に対し、 c _ {i + 1} c _ i の倍数である。各コインは無限に存在する。

ちょうど  k 枚のコインを使ってちょうど  p 円を払うとき、コインの重さの合計の最小値と最大値を出力せよ。ただし、ちょうど  p 円を払うことができないとき、 -1 と出力せよ。

入力は以下の制約を満たす。

  •  1 \leq n \leq 60
  •  1 \leq n \leq 10 ^ 3
  •  1 \leq p \leq 10 ^ {18}
  •  1 \leq c _ i \leq 10 ^ {18}, 1 \leq w _ i \leq 10 ^ {15} \ (1 \leq i \leq n)
  •  i \ (1 \leq i \leq n - 1) に対し、 c _ {i + 1} i の倍数である。

F 問題: Hedgehog Graph

実行時間制限: 5 秒 / メモリ制限: 1 GB

 10 ^ 6 の頂点の有向グラフ  G がある。 G の各頂点の出次数は  1 であり、 G は閉路を  1 つだけ持ちその頂点数は  3 以上である。また、 G の各辺  u \rightarrow v に対し、 v はこの閉路に含まれる。

以下のクエリを  900 回まで投げることができる。

  •  G の頂点  v 1 以上  5 \times 10 ^ {18} 以下の整数 x を出力する。頂点  v から辺に沿って  x 回移動した後の頂点が与えられる。

 G の閉路の頂点数を求めよ。ただし、 n \ (1 \leq n \leq 10) 個のテストケースが与えられるので、すべて答えよ。

G 問題: Lamb's Respite

実行時間制限: 3 秒 / メモリ制限: 1 GB

体力が整数  h で表されるキャラクターがいる。キャラクターは正の整数で表される最大体力  H を持つ。体力が変化し  H より大きくなった場合体力は  H となり、 0 より小さくなった場合は  0 となってキャラクターは死亡する。

体力が変化する行動を  n 回行う。 i \ (1 \leq i \leq n) 回目の行動ではキャラクターの体力が  a _ i だけ変化する。

キャラクターは魔法を使うことができる。魔法を使うと、魔法が持続している間、体力が  \lceil \frac{H}{10} \rceil 以下であるか、体力の変化により体力が  \lceil \frac{H}{10} \rceil 以下になる (死亡する場合を含む) 場合、体力が  \lceil \frac{H}{10} \rceil となり、魔法が切れるまで体力が変化しなくなる。

以下の  2 種類のクエリを  q 個処理せよ。

  • クエリ  1: 整数  l, r, x が与えられる。キャラクターの開始時の体力と最大体力が  x であり、 l 回目の行動の前から  r 回目の行動の後まで魔法を使ったとき、 n 回目の行動の後のキャラクターの体力を出力する。ただし、キャラクターが途中で死亡する場合、 0 を出力する。
  • クエリ  2: 整数  i, x が与えられる。 a _ i x に更新する。

入力は以下の制約を満たす。

  •  1 \leq n, q \leq 300000
  •  |a _ i| \leq 10 ^ 9 \ (1 \leq i \leq n)
  • クエリ  1 について、 1 \leq l \leq r \leq n, 1 \leq x \leq 10 ^ 9
  • クエリ  2 について、 1 \leq i \leq n, -10 ^ 9 \leq x \leq 10 ^ 9
  • クエリ  1 1 つ以上存在する。

H 問題: Or Machine

実行時間制限: 4 秒 / メモリ制限: 1 GB

 n 個の整数  x _ 1, x _ 2, \cdots, x _ n が与えられる。

 l 種類の操作がある。 i \ (1 \leq i \leq n) 番目の操作は以下のような操作である。

  • 整数  a, b が与えられる。 x _ a x _ a x _ b のビットごとの論理和に置き換える。

操作を  1, 2, \cdots, n の順に行い、操作  n を行った後再び操作  1 に戻り繰り返す。合計で  t 回の操作を行ったとき、 x _ 1, x _ 2, \cdots, x _ n の値を出力せよ。

入力は以下の制約を満たす。

  •  1 \leq n, l \leq 2^{18}
  •  1 \leq t \leq 10^{18}
  •  1 \leq a, b \leq n
  •  0 \leq x _ i \lt 2 ^ 8

I 問題: Organizing Colored Sheets

実行時間制限: 3 秒 / メモリ制限: 1 GB

 n \times m のグリッドがあり、各マスは色で塗られているか塗られていないかのどちらかである。

 1 \leq w \leq m, 1 \leq h \leq n を満たす整数  w, h を選び、 w \times h の色紙のみを以下の条件を満たすようにグリッド上に何枚か配置したい。

  • 色紙がグリッドの外部にはみ出ることはない。
  • 色紙を回転してはならない。
  • 各マスが  2 枚以上の色紙で覆われてもよい。
  • 色で塗られていない各マスは  1 枚以上の色紙で覆われている。
  • 色で塗られている各マスは色紙で覆われていない。

条件を満たすようにグリッド上に色紙を配置できるような整数  w, h の組の個数を求めよ。

入力は以下の制約を満たす。

  •  1 \leq n, m \leq 3000
  • 色で塗られていないマスが  1 つ以上存在する。

J 問題: Periodic Ruler

実行時間制限: 2 秒 / メモリ制限: 1 GB

無限に長い数直線の整数の位置にそれぞれ印が付いている。色は  1 以上  100 以下の整数で表され、 i の位置の印の色は  c _ i である。

付けられた印の色は周期  t を持っている。ただし、周期とは、すべでの整数  i に対し  c _ i = c _ {i + t} であるような最小の正の整数  t である。

印の色について  n 個の情報が与えられる。 i \ (1 \leq i \leq n) 番目の情報は整数  x _ i と色  a _ i からなり、 x _ i の位置の印の色が  a _ i であることを示す。

周期  t としてありえない整数を列挙し、その個数とそれらの和を出力せよ。

入力は以下の制約を満たす。

  •  1 \leq n \leq 50
  •  |x _ i| \leq 10 ^ 9 \ (1 \leq i \leq n)
  •  1 \leq a _ i \leq 100 \ (1 \leq i \leq n)
  •  1 \leq i, j \leq n, i \neq j ならば  x _ i \neq x _ j である。

K 問題: Three Competitions

実行時間制限: 5 秒 / メモリ制限: 1 GB

 n 人の人が  3 回の大会に参加した。それぞれの大会で、 n 人の人はそれぞれ異なる順位を得た。

 3 回中  2 回以上の大会で人  a の順位が人  b の順位より小さかったとき、人  a が人  b に勝ったという。また、人の列  p _ 1, p _ 2, \cdots, p _ k \ (k \geq 2) であって、 1 \leq i \leq k - 1 に対して人  p _ i が人  p _ {i + 1} に勝ち、また  p _ 1=a, p _ k=b であるとき、人  a は人  b に間接的に勝ったという。

 q 個のクエリが与えられる。各クエリでは、異なる人  a, b が与えられるので、人  a が人  b に間接的に勝ったかどうか答えよ。

入力は以下の制約を満たす。

  •  2 \leq n \leq 10 ^ 5
  •  1 \leq q \leq 2 \times 10 ^ 5
  •  1 \leq a, b \leq n, a \neq b

L 問題: Utilitarianism 2

実行時間制限: 7 秒 / メモリ制限: 1 GB

 n 人のワクチン生産者、 m 個の病院と  k 人の人がいる。人  i は生産者  a _ i から病院  b _ i c _ i 個のワクチンを届けることができる。異なる人が同じ生産者と病院の組み合わせを持つことはない。また、それぞれのワクチン生産者と病院は  1 人の人しか利用してはならない。

それぞれの人に対し、その人がいない場合、 m 個の病院に届くワクチンの総数が何個減少するか求めよ。

入力は以下の制約を満たす。

  •  1 \leq n, m \leq 2000
  •  1 \leq k \leq \min(8000, n \times n)
  •  1 \leq a _ i \leq n \ (1 \leq i \leq k)
  •  1 \leq b _ i \leq m \ (1 \leq i \leq k)
  •  1 \leq c _ i \leq 10^{12} \ (1 \leq i \leq k)
  •  1 \leq i, j \leq n, i \neq j ならば  (a _ i, b _ i) \neq (a _ j, b _ j) である。

M 問題: Yet Another Range Query Problem

実行時間制限: 3 秒 / メモリ制限: 1 GB

 1 以上  n 以下の整数からなる長さ  n の数列  A が与えられる。

 n \times n の行列  B がある。 1 \leq i \leq j \leq n のとき、 B _ {ij} = \min(A _ i, A _ {i + 1}, \cdots, A _ j) \times \max(A _ i, A _ {i + 1}, \cdots, A _ j) である。そうでないとき、 B _ {ij} = 0 である。

 q 個のクエリが与えられる。各クエリでは整数  l, r, s, e が与えられるので、 \displaystyle{\sum _ {x=l}^r \sum _ {y=s}^e B _ {xy}} \bmod 10 ^ 9 + 7 を求めよ。

入力は以下の制約を満たす。

  •  1 \leq n, q \leq 150000
  •  1 \leq A _ i \leq n \ (1 \leq i \leq n)
  •  1 \leq l \leq r \leq n, 1 \leq s \leq e \leq n

問題文

A 問題: Automatic Sprayer 2

https://official.contest.yandex.ru/opencupXXII/contest/30766/problems/A/ https://official.contest.yandex.ru/testsys/statement-image?imageId=396117fb4db718568cc88ac6ec31f8d149c4ef3835fb0401308c6a3c7c151636

B 問題: Cilantro

https://official.contest.yandex.ru/opencupXXII/contest/30766/problems/B/ https://official.contest.yandex.ru/testsys/statement-image?imageId=627558b9d5945e8a14101d3b57fbeb6d7bc65d6235b2876501e2f642be14f3a8

C 問題: Equivalent Pipelines

https://official.contest.yandex.ru/opencupXXII/contest/30766/problems/C/ https://official.contest.yandex.ru/testsys/statement-image?imageId=7c76c56aa057727fc8c78ce30caec9ec7d4e900061addb39c8cd516cb1547be1

D 問題: Flowerbed Redecoration

https://official.contest.yandex.ru/opencupXXII/contest/30766/problems/D/ https://official.contest.yandex.ru/testsys/statement-image?imageId=af78efd0e23a1fe94d81937b6bdc044ee916bdd8a7e3a2e998db46a5573a1401

E 問題: Goose Coins

https://official.contest.yandex.ru/opencupXXII/contest/30766/problems/E/ https://official.contest.yandex.ru/testsys/statement-image?imageId=4c01eecb10f69f7737cfae064f7a786b187305cfba788d982909f2c92d64e152

F 問題: Hedgehog Graph

https://official.contest.yandex.ru/opencupXXII/contest/30766/problems/F/ https://official.contest.yandex.ru/testsys/statement-image?imageId=e63ba41f3d4abc8591a114a88aed679fe35f5523d1365b63ecad4e37f0e8fa6c

G 問題: Lamb's Respite

https://official.contest.yandex.ru/opencupXXII/contest/30766/problems/G/ https://official.contest.yandex.ru/testsys/statement-image?imageId=10c2477b2bfe8178596a337f9e311319f29eb57e74da8d95b9afeaf9cfb4da11

H 問題: Or Machine

https://official.contest.yandex.ru/opencupXXII/contest/30766/problems/H/ https://official.contest.yandex.ru/testsys/statement-image?imageId=5a3e6a6678681275b0f02fe6dba9ece65e6a46a8bb78bc0bef74fe7333c38aaf

I 問題: Organizing Colored Sheets

https://official.contest.yandex.ru/opencupXXII/contest/30766/problems/I/ https://official.contest.yandex.ru/testsys/statement-image?imageId=f6c254126c1e781aac9e072c5f36ab821a8f34a710f95ef7363a084f604a4dac

J 問題: Periodic Ruler

https://official.contest.yandex.ru/opencupXXII/contest/30766/problems/J/ https://official.contest.yandex.ru/testsys/statement-image?imageId=f022e09541c75a2afedf1b2f928e0b747bef0b734ff7751939cf7d1da64eb8bc

K 問題: Three Competitions

https://official.contest.yandex.ru/opencupXXII/contest/30766/problems/K/ https://official.contest.yandex.ru/testsys/statement-image?imageId=640a301b7b3eaf8ad6c00ca6693b3f20efead7451cbbcaf78ed48e846c4dc526

L 問題: Utilitarianism 2

https://official.contest.yandex.ru/opencupXXII/contest/30766/problems/L/ https://official.contest.yandex.ru/testsys/statement-image?imageId=edbc3ea64109f24361cea53be65f9254fe5bc062ff7131ff6167f37342126a63

M 問題: Yet Another Range Query Problem

https://official.contest.yandex.ru/opencupXXII/contest/30766/problems/M/ https://official.contest.yandex.ru/testsys/statement-image?imageId=36c7c7dc69eb451647eb8f6bfa2f0bc455ab126b5bc49e91de05e39525b1acda

AOJ 0367: Charging System for Network/ネットワークの課金システム

https://judge.u-aizu.ac.jp/onlinejudge/description.jsp?id=0367&lang=ja

解法

ネットワークはマシンを頂点としケーブルの太さを各辺の重みとする木と考えます。

add クエリが扱いにくいので、各頂点  v に対し整数  A _ v を辺  (v, w) の重みが常に  A _ v + A _ w となるようにすると、add クエリが  A _ x d を加えるというクエリになります。 A _ v の初期値は BFS をすることで計算できます。

send クエリは、頂点  s から頂点  t の最短路を  v _ 1, v _ 2, \cdots, v _ M \ (v _ 1 = s, v _ M = t) としたとき、 \displaystyle{\sum _ {i = 1} ^ {M - 1} f(A _ {v _ i} + A _ {v _ {i + 1}})} (ただし  f(n) n K の倍数のとき  0、そうでないとき  n) を求めることに帰着され、これは列  [B _ 1, B _ 2, \cdots, B _ M] の情報を  B _ 1, B _ M, \displaystyle{\sum _ {i = 1} ^ {M - 1} f(B _ i + B _ {i + 1})} の組として持つと、HLD にセグメント木を載せることで処理できます。

計算量は  O(N + Q(\log N) ^ 2) です。

実装

https://judge.u-aizu.ac.jp/onlinejudge/review.jsp?rid=5956782

#include <bits/stdc++.h>
using namespace std;
struct monoid{
  bool e;
  long long L, R, S;
  monoid(): e(true){
  }
  monoid(long long x): e(false), L(x), R(x), S(0){
  }
};
monoid f(monoid A, monoid B, int K){
  if (A.e){
    return B;
  }
  if (B.e){
    return A;
  }
  monoid ans;
  ans.e = false;
  ans.L = A.L;
  ans.R = B.R;
  ans.S = A.S + B.S;
  if ((A.R + B.L) % K != 0){
    ans.S += A.R + B.L;
  }
  return ans;
};
struct segment_tree{
  int N;
  array<vector<monoid>, 2> ST;
  int K;
  segment_tree(){
  }
  segment_tree(vector<long long> &A, int K): K(K){
    int N2 = A.size();
    N = 1;
    while (N < N2){
      N *= 2;
    }
    ST[0] = vector<monoid>(N * 2 - 1);
    ST[1] = vector<monoid>(N * 2 - 1);
    for (int i = 0; i < N2; i++){
      ST[0][N - 1 + i] = monoid(A[i]);
      ST[1][N - 1 + i] = monoid(A[i]);
    }
    for (int i = N - 2; i >= 0; i--){
      ST[0][i] = f(ST[0][i * 2 + 1], ST[0][i * 2 + 2], K);
      ST[1][i] = f(ST[1][i * 2 + 2], ST[1][i * 2 + 1], K);
    }
  }
  void update(int i, int x){
    i += N - 1;
    ST[0][i].L += x;
    ST[0][i].R += x;
    ST[1][i].L += x;
    ST[1][i].R += x;
    while (i > 0){
      i = (i - 1) / 2;
      ST[0][i] = f(ST[0][i * 2 + 1], ST[0][i * 2 + 2], K);
      ST[1][i] = f(ST[1][i * 2 + 2], ST[1][i * 2 + 1], K);
    }
  }
  monoid range_fold(int L, int R, int d, int i, int l, int r){
    if (r <= L || R <= l){
      return monoid();
    } else if (L <= l && r <= R){
      return ST[d][i];
    } else {
      int m = (l + r) / 2;
      if (d == 0){
        return f(range_fold(L, R, d, i * 2 + 1, l, m), range_fold(L, R, d, i * 2 + 2, m, r), K);
      } else {
        return f(range_fold(L, R, d, i * 2 + 2, m, r), range_fold(L, R, d, i * 2 + 1, l, m), K);
      }
    }
  }
  monoid range_fold(int L, int R, int d){
    return range_fold(L, R, d, 0, 0, N);
  }
};
struct heavy_light_decomposition{
  vector<int> p, sz, in, next;
  segment_tree ST;
  int K;
  heavy_light_decomposition(vector<int> &p, vector<vector<int>> &c, vector<long long> &A, int K): p(p), K(K){
    int N = p.size();
    sz = vector<int>(N, 1);
    dfs1(c);
    in = vector<int>(N);
    next = vector<int>(N);
    next[0] = 0;
    int t = 0;
    dfs2(c, t);
    vector<long long> A2(N);
    for (int i = 0; i < N; i++){
      A2[in[i]] = A[i];
    }
    ST = segment_tree(A2, K);
  }
  void dfs1(vector<vector<int>> &c, int v = 0){
    for (int &w : c[v]){
      dfs1(c, w);
      sz[v] += sz[w];
      if (sz[w] > sz[c[v][0]]){
        swap(w, c[v][0]);
      }
    }
  }
  void dfs2(vector<vector<int>> &c, int &t, int v = 0){
    in[v] = t;
    t++;
    for (int w : c[v]){
      if (w == c[v][0]){
        next[w] = next[v];
      } else {
        next[w] = w;
      }
      dfs2(c, t, w);
    }
  }
  void update(int v, int x){
    ST.update(in[v], x);
  }
  int lca(int u, int v){
    while (true){
      if (in[u] > in[v]){
        swap(u, v);
      }
      if (next[u] == next[v]){
        return u;
      }
      v = p[next[v]];
    }
  }
  monoid path_fold(int u, int v){
    int w = lca(u, v);
    monoid L;
    while (next[u] != next[w]){
      L = f(L, ST.range_fold(in[next[u]], in[u] + 1, 1), K);
      u = p[next[u]];
    }
    L = f(L, ST.range_fold(in[w], in[u] + 1, 1), K);
    monoid R;
    while (next[v] != next[w]){
      R = f(ST.range_fold(in[next[v]], in[v] + 1, 0), R, K);
      v = p[next[v]];
    }
    R = f(ST.range_fold(in[w] + 1, in[v] + 1, 0), R, K);
    return f(L, R, K);
  }
};
int main(){
  int N, K;
  cin >> N >> K;
  vector<vector<pair<int, int>>> E(N);
  for (int i = 0; i < N - 1; i++){
    int a, b, c;
    cin >> a >> b >> c;
    E[a].push_back(make_pair(c, b));
    E[b].push_back(make_pair(c, a));
  }
  vector<int> p(N, -1);
  vector<vector<int>> c(N);
  vector<long long> a(N);
  a[0] = 0;
  queue<int> q;
  q.push(0);
  while (!q.empty()){
    int v = q.front();
    q.pop();
    for (auto P : E[v]){
      int w = P.second;
      if (w != p[v]){
        p[w] = v;
        c[v].push_back(w);
        a[w] = P.first - a[v];
        q.push(w);
      }
    }
  }
  heavy_light_decomposition T(p, c, a, K);  
  int Q;
  cin >> Q;
  for (int i = 0; i < Q; i++){
    string S;
    cin >> S;
    if (S == "add"){
      int x, d;
      cin >> x >> d;
      T.update(x, d);
    }
    if (S == "send"){
      int s, t;
      cin >> s >> t;
      cout << T.path_fold(s, t).S << endl;
    }
  }
}

AOJ 2733: Cube Dividing

https://judge.u-aizu.ac.jp/onlinejudge/description.jsp?id=2733

公式解説とは異なる解法で解きました。

問題概要

 A \times B \times C 個の同じ大きさの立方体で構成された幅  A、高さ  B、奥行き  C の立方体がある。 1 \leq i \leq N を満たす各  i について、左から  X _ i 番目・下から  Y _ i 番目・前から  Z _ i 番目の立方体を取り除いた。残っている各立方体を頂点とし隣り合う立方体に対応する頂点の間に辺を張った無向グラフの連結成分の個数を求めよ。

解法

以下、 x, y, z 軸をそれぞれ左から右、下から上、前から後ろに取ります。

座標圧縮をして、見る必要のある  x 座標を絞ります。具体的には、 x 座標が  0 X _ i \ (1 \leq i \leq N) X _ i+1 \ (1 \leq i \leq N, X _ i \lt A - 1) の場所のみ考えればよいです。

 x 座標について、その位置にある取り除いた立方体の  y 座標のリストを作り、同様に座標圧縮をします。さらに、各  y 座標について、その位置にある取り除いた立方体の  z 座標のリストを作り、ソートして取り除いた立方体の間の直方体に番号を振ります。これにより、立方体が取り除かれた後の立体は  O(N) 個の直方体に分割されます。これらの直方体を頂点とし、接している  2 つの直方体に対応する頂点の間に辺を張った無向グラフを作ることを考えます。

 x 座標のそれぞれの隣接する  2 つの  y 座標について、 z 座標が変化すると  2 つの立方体がそれぞれどの直方体に属するか (あるいは、取り除かれていて立体に含まれていないか) がどのように変化するかを調べ、それらが同時に直方体に属している場合、その  2 つの直方体は接しているので、辺を追加します。

次に、それぞれの隣接する  2 つの  x 座標について、座標圧縮で得た考慮する必要のある  y 座標のリストの和集合を求めます。上と同様に、それぞれの  y 座標についての  z 座標の変化による立方体の所属の変化を調べ、辺を追加します。

グラフが完成したら、そのグラフを連結成分に分解し、連結成分の個数が答えとなります。

計算量は  O(N \log N) です。

実装

judge.u-aizu.ac.jp

#include <bits/stdc++.h>
using namespace std;
int main(){
  int A, B, C, N;
  cin >> A >> B >> C >> N;
  vector<int> X(N), Y(N), Z(N);
  for (int i = 0; i < N; i++){
    cin >> X[i] >> Y[i] >> Z[i];
  }
  vector<int> rx;
  rx.push_back(0);
  for (int i = 0; i < N; i++){
    rx.push_back(X[i]);
    if (X[i] < A - 1){
      rx.push_back(X[i] + 1);
    }
  }
  sort(rx.begin(), rx.end());
  rx.erase(unique(rx.begin(), rx.end()), rx.end());
  int A2 = rx.size();
  rx.push_back(A);
  for (int i = 0; i < N; i++){
    X[i] = lower_bound(rx.begin(), rx.end(), X[i]) - rx.begin();
  }
  vector<vector<int>> id(A2);
  for (int i = 0; i < N; i++){
    id[X[i]].push_back(i);
  }
  int V = 0;
  vector<pair<int, int>> E;
  vector<vector<int>> ry(A2);
  vector<int> B2(A2);
  vector<vector<vector<pair<int, int>>>> upd(A2);
  vector<vector<int>> uc(A2);
  for (int i = 0; i < A2; i++){
    ry[i].push_back(0);
    int cnt = id[i].size();
    for (int j = 0; j < cnt; j++){
      ry[i].push_back(Y[id[i][j]]);
      if (Y[id[i][j]] < B - 1){
        ry[i].push_back(Y[id[i][j]] + 1);
      }
    }
    sort(ry[i].begin(), ry[i].end());
    ry[i].erase(unique(ry[i].begin(), ry[i].end()), ry[i].end());
    B2[i] = ry[i].size();
    vector<vector<int>> id2(B2[i]);
    for (int j = 0; j < cnt; j++){
      int Y2 = lower_bound(ry[i].begin(), ry[i].end(), Y[id[i][j]]) - ry[i].begin();
      id2[Y2].push_back(id[i][j]);
    }
    upd[i].resize(B2[i]);
    for (int j = 0; j < B2[i]; j++){
      int cnt2 = id2[j].size();
      for (int k = 0; k < cnt2; k++){
        upd[i][j].push_back(make_pair(Z[id2[j][k]], -1));
      }
      vector<int> rz;
      rz.push_back(-1);
      rz.push_back(C);
      for (int k = 0; k < cnt2; k++){
        rz.push_back(Z[id2[j][k]]);
      }
      sort(rz.begin(), rz.end());
      int cnt3 = rz.size();
      for (int k = 0; k < cnt3 - 1; k++){
        if (rz[k + 1] - rz[k] > 1){
          upd[i][j].push_back(make_pair(rz[k] + 1, V));
          V++;
        }
      }
    }
    uc[i].resize(B2[i]);
    for (int j = 0; j < B2[i]; j++){
      uc[i][j] = upd[i][j].size();
    }
    for (int j = 0; j < B2[i] - 1; j++){
      vector<tuple<int, int, int>> upd2;
      for (int k = 0; k < 2; k++){
        for (int l = 0; l < uc[i][j + k]; l++){
          upd2.push_back(make_tuple(upd[i][j + k][l].first, k, upd[i][j + k][l].second));
        }
      }
      sort(upd2.begin(), upd2.end());
      int cnt2 = upd2.size();
      vector<int> c = {-1, -1};
      for (int k = 0; k < cnt2; k++){
        int t = get<0>(upd2[k]);
        int h = get<1>(upd2[k]);
        int v = get<2>(upd2[k]);
        c[h] = v;
        bool ok = true;
        if (k < cnt2 - 1){
          if (get<0>(upd2[k + 1]) == t){
            ok = false;
          }
        }
        if (ok){
          if (c[0] != -1 && c[1] != -1){
            E.push_back(make_pair(c[0], c[1]));
          }
        }
      }
    }
  }
  for (int i = 0; i < A2 - 1; i++){
    vector<int> ry2;
    for (int j = i; j <= i + 1; j++){
      for (int k = 0; k < B2[j]; k++){
        ry2.push_back(ry[j][k]);
      }
    }
    sort(ry2.begin(), ry2.end());
    ry2.erase(unique(ry2.begin(), ry2.end()), ry2.end());
    int cnt = ry2.size();
    for (int j = 0; j < cnt; j++){
      vector<int> yp(2);
      for (int k = 0; k < 2; k++){
        yp[k] = upper_bound(ry[i + k].begin(), ry[i + k].end(), ry2[j]) - ry[i + k].begin() - 1;
      }
      vector<tuple<int, int, int>> upd2;
      for (int k = 0; k < 2; k++){
        for (int l = 0; l < uc[i + k][yp[k]]; l++){
          upd2.push_back(make_tuple(upd[i + k][yp[k]][l].first, k, upd[i + k][yp[k]][l].second));
        }
      }
      sort(upd2.begin(), upd2.end());
      int cnt2 = upd2.size();
      vector<int> c = {-1, -1};
      for (int k = 0; k < cnt2; k++){
        int t = get<0>(upd2[k]);
        int h = get<1>(upd2[k]);
        int v = get<2>(upd2[k]);
        c[h] = v;
        bool ok = true;
        if (k < cnt2 - 1){
          if (get<0>(upd2[k + 1]) == t){
            ok = false;
          }
        }
        if (ok){
          if (c[0] != -1 && c[1] != -1){
            E.push_back(make_pair(c[0], c[1]));
          }
        }
      }
    }
  }
  int M = E.size();
  vector<vector<int>> E2(V);
  for (int i = 0; i < M; i++){
    int v = E[i].first;
    int w = E[i].second;
    E2[v].push_back(w);
    E2[w].push_back(v);
  }
  int ans = 0;
  vector<bool> used(V, false);
  for (int i = 0; i < V; i++){
    if (!used[i]){
      used[i] = true;
      ans++;
      queue<int> Q;
      Q.push(i);
      while (!Q.empty()){
        int v = Q.front();
        Q.pop();
        for (int w : E2[v]){
          if (!used[w]){
            used[w] = true;
            Q.push(w);
          }
        }
      }
    }
  }
  cout << ans << endl;
}

2021牛客暑期多校训练营5 (2021/07/31 12:00-17:00)

コンテストページ

ac.nowcoder.com

問題概要

A 問題: Away from College

(問題文を読んでいません 原文を読んでください)

B 問題: Boxes

実行時間制限: 1 秒 / メモリ制限: 262144 KB

 n 個の箱があり、それぞれの箱には黒いボールまたは白いボールがそれぞれ  \displaystyle{\frac{1}{2}} の確率で入っている。箱は閉じられていて、中のボールを見ることはできない。

以下の操作 1 または操作 2 を繰り返すことにより、すべての箱に入っている玉の色を当てたい。

  • 操作 1: 整数  i \ (1 \leq i \leq n) を選ぶ。コスト  w_i を支払い、箱  i を開けて入っているボールの色を知る。

  • 操作 2: コスト  C を支払い、まだ開けられていない箱に入っている黒いボールの個数を教えてもらう。

すべての箱に入っている玉の色を当てるのに必要なコストの期待値の最小値を求めよ。

想定解答との絶対誤差または相対誤差が 10 ^ {-6} 以下であれば正解として扱われる。

入力は以下の制約を満たす。 -  1 \leq n \leq 10 ^ 5 -  0 \lt C \leq 10 ^ 9 -  0 \lt w _ i \leq 10 ^ 9

C 問題: Cheating and Stealing

実行時間制限: 2 秒 / メモリ制限: 262144 KB

W と L のみからなる長さ  n の文字列  S が与えられる。 i = 1, 2, \cdots, |S| に対し、 f _ i(S) を以下の操作で定義する。

  1. 文字列  S の接頭辞であって、接頭辞に含まれる W, L の個数をそれぞれ  cnt _ W, cnt _ L とおくと、 \max(cnt _ W, cnt _ L) \geq i かつ  |cnt _ W - cnt _ L| \geq 2 が成立するもののうち最も短いものを取る。
  2. 条件を満たす接頭辞が存在しない場合、操作を終了する。得た得点が  f _ i(S) の値である。
  3. 条件を満たす接頭辞が存在する場合、その接頭辞に含まれる W の個数が L の個数より多いならば 1 点の得点を得る。その後、その接頭辞を削除し、1. に戻る。

 \displaystyle{\sum _ {i=1} ^ {|S|} f _ i(S) \times (|S| + 1) ^ {i-1}} 998244353 で割った余りを求めよ。

入力は以下の制約を満たす。

  •  1 \leq n \leq 10 ^ 6

D 問題: Double Strings

実行時間制限: 2 秒 / メモリ制限: 262144 KB

文字列  A, B が与えられる。 A の部分列  A' B の部分列  B' の組であって、 A' \lt B' を満たすものの個数を  998244353 で割った余りを求めよ。ただし、同じ部分文字列であっても  A B の異なる位置から取ったものは区別する。

入力は以下の制約を満たす。

  •  1 \leq |A|, |B| \leq 5000
  •  A, B は小文字のアルファベットからなる。

E 問題: Eert Esiwtib

実行時間制限: 3 秒 / メモリ制限: 262144 KB

長さ  n の数列  a _ 1, a _ 2, \cdots, a _ n と頂点  1 を根とする  n 頂点の木が与えられる。木の各辺には OR, AND, XOR のうち 1 つの演算子が書かれている。また、木の頂点  i には整数  a _ i が書かれている。

 u, v を木の異なる 2 頂点とし、それらの間のパスの頂点を  p _ 0 = u, p _ 1, p _ 2, \cdots, p _ {s-1}, p _ s = v とおく。また、頂点  p _ {i-1}, p _ i を結ぶ辺に書かれている演算子 opt _ i とおく。このとき、このパスの重みを  (a _ {p _ 0} opt _ 1 (a _ {p _ 1} opt _ 2 ( \cdots (a _ {p _ {s - 1}} opt _ {s - 1} a _ {p _ s})))) と定義する。

 q 個のクエリが与えられる。各クエリでは、整数  d と木の頂点  u が与えられる。ただし、 u は葉ではないことが保証される。頂点  i に書かれた整数が  a_ i + i \times d に変化したとき、 u v の先祖であるような頂点  v すべて (ただし頂点  u 自身は除く) について、 (u,v) パスの重みを求め、それらの OR, AND, XOR をそれぞれ求めよ。

入力は以下の制約を満たす。

  •  2 \leq n \leq 10 ^ 5
  •  1 \leq q \leq 10 ^ 5
  •  1 \leq a _ i \leq 10 ^ {18} \ (1 \leq i \leq n)
  •  0 \leq d \leq 100

F 問題: Finding Points

実行時間制限: 1 秒 / メモリ制限: 262144 KB

凸多角形  A  _ 1A _ 2 \cdots A _ n が与えられる。ただし、点  A  _ 1A _ 2 \cdots A _ n は反時計回りに並んでいる。 A _ i の座標は  (x _ i, y _ i) である。この多角形の内部の点  P に対する  \angle A _ 1PA _2, \angle A _ 2PA _ 3, \cdots, \angle A _ NPA _ 1 の最小値としてありうる値の最大値を求めよ。

想定解答との絶対誤差または相対誤差が 10 ^ {-6} 以下であれば正解として扱われる。

入力は以下の制約を満たす。

  •  4 \leq n \leq 100
  •  |x _ i|, |y _ i| \leq 10000 \ (1 \leq i \leq n)

G 問題: Greater Integer, Better LCM

実行時間制限: 2 秒 / メモリ制限: 262144 KB

正の整数  a, b, c \ (a, b \lt c) が与えられる。ただし  c素因数分解した形で与えられ、 c = \displaystyle{\prod _ {i = 1} ^ n p _ i ^ {q _ i}} である。非負整数  x, y であって  \text{lcm}(a+x,b+y)=c を満たすものに対する  x+y の最小値を求めよ。

入力は以下の制約を満たす。

H 問題: Holding Two

実行時間制限: 1 秒 / メモリ制限: 262144 KB

正の整数  n, m が与えられる。以下の条件を満たす各要素が 0 または 1 である  n \times m の行列を構築せよ。条件を満たす行列が存在しない場合、-1 と出力せよ。

  • 行列の異なる  3 つの要素  A _ {i _ 1 j _ 1}, A _ {i _ 2 j _ 2}, A _ {i _ 3 j _ 3} であって、 A  _ {i _ 1 j _ 1} = A _ {i _ 2 j _ 2}
= A _ {i _ 3 j _ 3}, -1 \leq i _ 1 - i _ 2 = i _ 2 - i _ 3 \leq 1, -1 \leq j _ 1 - j _ 2 = j _ 2 - j _ 3 \leq 1 を満たすものが存在しない。

入力は以下の制約を満たす。

  •  1 \leq n, m \leq 1000

I 問題: Interval Queries

実行時間制限: 4 秒 / メモリ制限: 262144 KB

長さ  n の数列  A _ 1, A _ 2, \cdots, A _ n が与えられる。

集合  S に対し、 f(S) x, x+1, \cdots, x+len-1 がすべて  S の要素であるような  x が存在する最大の  len の値とする。

 q 個のクエリが与えられる。各クエリでは、整数  l, r, k が与えられるので、 \displaystyle{\sum _ {i=0} ^ {k-1} f(\lbrace A _ {l-i}, \cdots, A _ l, A _ {l+1}, \cdots, A _ r, \cdots, A _ {r + i} \rbrace) (n+1) ^ i} 998244353 で割った余りを求めよ。

入力は以下の制約を満たす。

  •  1 \leq n \leq 100000
  •  1 \leq q \leq 100000
  •  1 \leq A _ i \leq N \ (1 \leq i \leq N)
  •  1 \leq l - k + 1 \leq l \leq r \leq r + k - 1 \leq n
  •  \displaystyle{\sum _ {i=0} ^ q k _ i} \leq 10 ^ 7

J 問題: Jewels

実行時間制限: 1 秒 / メモリ制限: 262144 KB

 n 個の宝石があり、宝石  i は時刻  t 秒において  (x _ i, y _ i, z _ i + t \times v _ i) にある。 1 秒につき  1 個の宝石を原点とその宝石の距離の二乗のコストで得ることができる。 n 個の宝石をすべて得るのに必要なコストの最小値を求めよ。

入力は以下の制約を満たす。

  •  1 \leq n \leq 300
  •  0 \leq |x _ i|, |y _ i| \leq 1000
  •  0 \leq z _ i, v _ i \leq 1000

K 問題: King of Range

実行時間制限: 1 秒 / メモリ制限: 262144 KB

長さ  n の数列  a _ 1, a _ 2, \cdots, a _ n が与えられる。

 m 個のクエリが与えられる。各クエリでは整数  k が与えられるので、 1 \leq l \leq r \leq n なる  (l, r) の組であって、 \max(a _ l, a _ {l+1}, \cdots, a _ r) - \min(a _ l, a _ {l+1}, \cdots, a _ r) \gt k を満たすものの個数を求めよ。

入力は以下の制約を満たす。

  •  1 \leq n \leq 10 ^ 5
  •  1 \leq m \leq 100
  •  1 \leq a _ i \leq 10 ^ 9 \ (1 \leq i \leq n)
  •  1 \leq k \leq 10 ^ 9

問題文

A 問題: Away from College

https://ac.nowcoder.com/acm/contest/11256/A f:id:SSRS:20210801202520p:plain f:id:SSRS:20210801202536p:plain f:id:SSRS:20210801202559p:plain f:id:SSRS:20210801202610p:plain

B 問題: Boxes

https://ac.nowcoder.com/acm/contest/11256/B f:id:SSRS:20210801202623p:plain f:id:SSRS:20210801202634p:plain f:id:SSRS:20210801202644p:plain

C 問題: Cheating and Stealing

https://ac.nowcoder.com/acm/contest/11256/C f:id:SSRS:20210801202658p:plain f:id:SSRS:20210801202754p:plain f:id:SSRS:20210801202810p:plain

D 問題: Double Strings

https://ac.nowcoder.com/acm/contest/11256/D f:id:SSRS:20210801202823p:plain f:id:SSRS:20210801202835p:plain

E 問題: Eert Esiwtib

https://ac.nowcoder.com/acm/contest/11256/E f:id:SSRS:20210801202901p:plain f:id:SSRS:20210801202913p:plain f:id:SSRS:20210801202926p:plain f:id:SSRS:20210801202940p:plain

F 問題: Finding Points

https://ac.nowcoder.com/acm/contest/11256/F f:id:SSRS:20210801202954p:plain f:id:SSRS:20210801203007p:plain

G 問題: Greater Integer, Better LCM

https://ac.nowcoder.com/acm/contest/11256/G f:id:SSRS:20210801203025p:plain f:id:SSRS:20210801203039p:plain

H 問題: Holding Two

https://ac.nowcoder.com/acm/contest/11256/H f:id:SSRS:20210801203101p:plain f:id:SSRS:20210801203118p:plain

I 問題: Interval Queries

https://ac.nowcoder.com/acm/contest/11256/I f:id:SSRS:20210801203133p:plain f:id:SSRS:20210801203145p:plain

J 問題: Jewels

https://ac.nowcoder.com/acm/contest/11256/J f:id:SSRS:20210801203159p:plain f:id:SSRS:20210801203217p:plain

K 問題: King of Range

https://ac.nowcoder.com/acm/contest/11256/K f:id:SSRS:20210801203233p:plain f:id:SSRS:20210801203245p:plain

2021牛客暑期多校训练营4 (2021/07/26 12:00-17:00)

コンテストページ

ac.nowcoder.com

問題概要

A 問題: Course

実行時間制限: 3 秒 / メモリ制限: 262144 KB

 n 頂点の根付き木と長さ  n の数列  s _ 1, s _ 2, \cdots, s _ n c _ 1, c _ 2, \cdots, c _ n が与えられる。 Q 個のクエリに答えよ。

 i 番目のクエリでは整数  w _ i が与えられるので、長さ  n の非負整数の列  k _ 1, k _ 2, \cdots, k _ n で、以下の条件を満たすものの個数を  998244353 で割った余りを求めよ。

  •  i = 1, 2, \cdots, n について、以下が成り立つ。
    •  S を頂点  i の部分木の頂点の番号の集合とすると、 \displaystyle{\sum _ {v \in S} (k _ v s _ v) \geq c _ i}
  •  \displaystyle{\sum _ {i = 1} ^ N k _ i s _ i = w}

入力は以下の制約を満たす。

  •  1 \leq n \leq 100
  •  1 \leq Q \leq 10
  •  1 \leq s _ i \leq 5 \ (1 \leq i \leq n)
  •  0 \leq c_  i \leq 150 \ (1 \leq i \leq n)
  •  0 \leq w _ i \leq 10 ^ 8 \ (1 \leq i \leq Q)

B 問題: Sample Game

実行時間制限: 1 秒 / メモリ制限: 262144 KB

 1 以上  n 以下の整数を、 i \ (1 \leq i \leq n) は確率  \displaystyle{\frac{w _ i}{\displaystyle{\sum _ {i = 1} ^ {n} w _ i}}} で生成する乱数生成器がある。

以下の操作を行う。

  1. 乱数生成器で整数を  1 つ生成する。生成した整数を  x とおく。
  2.  x が今までに生成したすべての整数と比べ等しいまたは大きい場合、1. に戻る。そうでない場合、3. に進む。
  3. 生成した整数の個数を  m として、 m ^ 2 の得点を得る。

得点の期待値を既約分数で表し、 \mod 998244353 で出力せよ。ただし、得点の期待値は有理数であることが証明できる。

入力は以下の制約を満たす。

  •  2 \leq n \leq 100
  •  1 \leq w _ i \leq 10 ^ 6 \ (1 \leq i \leq n)

C 問題: LCS

実行時間制限: 1 秒 / メモリ制限: 262144 KB

 2 つの文字列  s _ 1, s _ 2 に対し、 LCS(s _ 1, s _ 2) s _ 1, s _ 2最長共通部分列の長さとする。

整数  a, b, c, n が与えられる。小文字のアルファベットからなる  n 文字の文字列  s _ 1, s _ 2, s _ 3 であって、 LCS(s _ 1, s _ 2) = a, LCS(s _ 2, s _ 3) = b, LCS(s _ 1, s _ 3) = c となるものを一つ構築するか、条件を満たす文字列が存在しないと報告せよ。

入力は以下の制約を満たす。

  •  1 \leq n \leq 1000
  •  0 \leq a, b, c \leq n

D 問題: Rebuild Tree

実行時間制限: 1 秒 / メモリ制限: 262144 KB

 n 頂点の木が与えられる。木の辺の集合を  T とする。また、 n 頂点の完全グラフの辺の集合を  B とする。

整数  k が与えられる。以下の条件をすべて満たす集合  X, Y の組の個数を  998244353 で割った余りを求めよ。

  •  X \subseteq T
  •  Y \subseteq B
  •  |X| = n - 1 - k
  •  |Y| = k
  • 辺の集合が  X \cup Y である  n 頂点のグラフは木である。

入力は以下の制約を満たす。

  •  2 \leq n \leq 5 \times 10 ^ 4
  •  1 \leq k \leq \min(100, n - 1)

E 問題: Tree Xor

実行時間制限: 2 秒 / メモリ制限: 262144 KB

 \oplus をビット単位の排他的論理和とする。

 n 頂点の木が与えられる。木の各辺には整数  c が書いてある。また、長さ  n の数列  l _ 1, l _ 2, \cdots, l _ n r _ 1, r _ 2, \cdots, r _ n が与えられる。長さ  n の非負整数の列  w _ 1, w _ 2, \cdots, w _ n であって、以下の条件をすべて満たすものの個数を求めよ。

  •  l _ i \leq w _ i \leq r _ i (1 \leq i \leq n)
  • 木の各辺  (u, v) に対し、 w _ u \oplus w _ v = c である。

入力は以下の制約を満たす。

  •  1 \leq n \leq 10 ^ 5
  •  0 \leq l _ i \leq r _ i \lt 2 ^ {30}
  •  0 \leq c \lt 2 ^ {30}

F 問題: Just a joke

実行時間制限: 1 秒 / メモリ制限: 262144 KB

 n 頂点  m 辺のグラフ  G が与えられる。Alice と Bob が以下の操作のうち 1 つを繰り返すゲームをする。操作ができなくなった人が負けである。勝者は誰か求めよ。

  •  G の辺を 1 本選び、それを削除する。
  •  G の連結成分で閉路を含まないものを 1 つ選び、削除する。

入力は以下の制約を満たす。

  •  1 \leq n \leq 100
  •  0 \leq m \leq min(200, \displaystyle{n(n-1)}{2})

G 問題: Product

実行時間制限: 1 秒 / メモリ制限: 262144 KB

 \displaystyle{\sum _ {i = 1} ^ n a _ i} = D である長さ  n の非負整数の列  a _ 1, a _ 2, \cdots. a _ n すべてに対する  \displaystyle{\frac{D!}{\prod _ {i = 1} ^ n (a _ i + k)!}} の総和を  \mod 998244353 で求めよ。

入力は以下の制約を満たす。

  •  1 \leq n \leq 50
  •  0 \leq k \leq 50
  •  0 \leq D \leq 10 ^ 8

H 問題: Convolution

実行時間制限: 3 秒 / メモリ制限: 262144 KB

 \oplus をビット単位の排他的論理和とする。

 p _ i i 番目の素数とする。 x y x = \displaystyle{\prod _ i p _ i ^ {a _ i}}, y = \displaystyle{\prod  _ i p _ i ^ {b _ i}}素因数分解されるとき、 x \otimes y = \displaystyle{\prod _ i p _ i ^ {|a _ i - b _ i|}} と定義する。

長さ  n の整数の列  a _ 1, a _ 2, \cdots, a _ n と整数  c が与えられる。 b _ i = \displaystyle{\sum _ {\substack{1 \leq j, k \leq n \\ j \otimes k = i}} a _ j k ^ c} としたとき、 \displaystyle{\bigoplus _ {i = 1, n} (b _ i \mod 998244353)} を求めよ。

入力は以下の制約を満たす。

  •  1 \leq n \leq 10 ^ 6
  •  0 \leq a _ i \lt 998244353 \ (1 \leq i \leq n)
  •  0 \leq c \leq 10 ^ 9

I 問題: Inverse Pair

実行時間制限: 1 秒 / メモリ制限: 262144 KB

長さ  n の数列  t _ 1, t _ 2, \cdots, t _ n の重みを、 1 \leq i \lt j \leq n, t _ i > t _ j となる  (i, j) の個数とする。

長さ  n の順列  a _ 1, a _ 2, \cdots, a _ n が与えられる。 b _ i \in \lbrace 0,1 \rbrace を満たす長さ  n の数列  b _ 1, b _ 2, \cdots, b _ n を選び、 c _ i = a _ i + b _ i (1 \leq i \leq n) としたときの  c の重みの最小値を求めよ。

入力は以下の制約を満たす。

  •  1 \leq n \leq 2 \times 10 ^ 5

J 問題: Average

実行時間制限: 1 秒 / メモリ制限: 262144 KB

 n 要素の数列  a _ 1, a _ 2, \cdots, a _ n m 要素の数列  b _ 1, b _ 2, \cdots, b _ m が与えられる。

 n \times m の行列  W がある。 W W _ {ij} = a _ i + b _ j \ (1 leq i \leq n, 1 \leq j \leq m) を満たす。

 W のうち  x 個以上の連続する行と  y 個以上の連続する列を選ぶ。選ばれた行・列の部分行列の平均値の最大値を求めよ。

入力は以下の制約を満たす。

  •  1 \leq n, m \leq 10 ^ 5
  •  1 \leq x \leq n
  •  1 \leq y \leq m
  •  0 \leq a _ i \leq 10 ^ 5 \ (1 \leq i \leq n)
  •  0 \leq b _ i \leq 10 ^ 5 \ (1 \leq i \leq m)

問題文

A 問題: Course

https://ac.nowcoder.com/acm/contest/11255/A f:id:SSRS:20210727003637p:plain f:id:SSRS:20210727003652p:plain f:id:SSRS:20210727003655p:plain

B 問題: Sample Game

https://ac.nowcoder.com/acm/contest/11255/B f:id:SSRS:20210727003812p:plain f:id:SSRS:20210727003824p:plain

C 問題: LCS

https://ac.nowcoder.com/acm/contest/11255/C f:id:SSRS:20210727003851p:plain f:id:SSRS:20210727003902p:plain

D 問題: Rebuild Tree

https://ac.nowcoder.com/acm/contest/11255/D f:id:SSRS:20210727003916p:plain f:id:SSRS:20210727003928p:plain

E 問題: Tree Xor

https://ac.nowcoder.com/acm/contest/11255/E f:id:SSRS:20210727003944p:plain f:id:SSRS:20210727003959p:plain

F 問題: Just a joke

https://ac.nowcoder.com/acm/contest/11255/F f:id:SSRS:20210727004013p:plain f:id:SSRS:20210727004027p:plain

G 問題: Product

https://ac.nowcoder.com/acm/contest/11255/G f:id:SSRS:20210727004042p:plain f:id:SSRS:20210727004055p:plain

H 問題: Convolution

https://ac.nowcoder.com/acm/contest/11255/H f:id:SSRS:20210727004115p:plain f:id:SSRS:20210727004127p:plain

I 問題: Inverse Pair

https://ac.nowcoder.com/acm/contest/11255/I f:id:SSRS:20210727004141p:plain f:id:SSRS:20210727004154p:plain

J 問題: Average

https://ac.nowcoder.com/acm/contest/11255/J f:id:SSRS:20210727004210p:plain f:id:SSRS:20210727004225p:plain

AOJ 1198: Don't Cross the Circles!/円を横切るな!

https://judge.u-aizu.ac.jp/onlinejudge/description.jsp?id=1198&lang=ja

解法

 P, Q のうち一方のみが含まれる円が 1 つ以上存在する場合、 P Q をその円と交差しないように結ぶことはできないので、答えは明らかに NO となります。

 P, Q がともに 1 個以上の円に含まれていて、 P を含む円の集合と  Q を含む円の集合が一致する場合、3 つ以上の円に共通部分はなく 2 つの円が接することはないという条件から、答えは必ず YES となります。

よって、 P, Q がともにどの円にも含まれていない場合のみ考えればよいです。

 n 個の円の中心を考え、円  i, j が二点で交わるとき、またそのときのみに、円  i の中心とと円  j の中心を線分で結ぶと、2 つの線分は端点以外では共有点を持たず、点  P, Q をどの円とも交差しないように結べることとどの線分とも交差しないように結べることは同値になります。

 n 個の円の中心を頂点、中心を結ぶ線分を辺とした平面グラフ  G を考えると、 G の辺は平面を何個かの領域に分割し、点  P, Q をどの辺とも交差しないように結べることは点  P, Q が同じ領域に属することと同値になります。 G が木である場合は領域が 1 つしかないので答えは必ず YES になります。そうでない場合は、 G には閉路が必ず存在します。閉路に現れる頂点を順に列挙し、それぞれの頂点に対応する円の中心を順に線分で結んだ多角形を考えます。もし点  P, Q のうち一方のみがこの多角形に含まれている場合、多角形の辺と交差せずに点  P, Q を結ぶことはできないので、答えは NO となります。そうでない場合、閉路の辺を一つ取りその辺を取り除き  G’ とすると、辺が平面を分割している領域のうちその辺を境に隣接している二つ ( A, B とします) が併合され新しい領域ができます。ここで、点  P, Q がそれぞれ領域  A, B または  B, A に属していることはありえないので、点  P, Q が同じ領域に属することは、グラフ  G’ の辺が平面を分割する領域について点  P, Q が同じ領域に属することと同値になります。よって、点  P, Q のうち一方のみが多角形の内部にあることで答えが NO であることが確定するかグラフ  G が木になり答えが YES が確定するまで  G から閉路を取り辺を取り除くことを繰り返すことで、点  P, Q をどの円とも交差しないように結べるかどうか判定することができます。

時間計算量は  O(n ^ 2 m) です。

実装

judge.u-aizu.ac.jp

#include <bits/stdc++.h>
using namespace std;
const double EPS = 0.0001;
struct point{
  double x, y;
  point(){
  }
  point(double x, double y): x(x), y(y){
  }
  point operator -(point P){
    return point(x - P.x, y - P.y);
  }
};
double abs(point P){
  return sqrt(P.x * P.x + P.y * P.y);
}
double dist(point P, point Q){
  return abs(Q - P);
}
double cross(point P, point Q){
  return P.x * Q.y - P.y * Q.x;
}
struct line{
  point A, B;
  line(point A, point B): A(A), B(B){
  }
};
point vec(line L){
  return L.B - L.A;
}
bool segment_intersect(line L1, line L2){
  if (cross(L1.A - L2.A, vec(L2)) * cross(L1.B - L2.A, vec(L2)) > 0){
    return false;
  }
  if (cross(L2.A - L1.A, vec(L1)) * cross(L2.B - L1.A, vec(L1)) > 0){
    return false;
  }
  return true;
}
bool point_in_polygon(point P, vector<point> Q){
  int N = Q.size();
  point P2 = point(P.x + 100000, P.y + 123456);
  int cnt = 0;
  for (int i = 0; i < N; i++){
    line L1(Q[i], Q[(i + 1) % N]);
    line L2(P, P2);
    if (segment_intersect(L1, L2)){
      cnt++;
    }
  }
  if (cnt % 2 == 1){
    return true;
  } else {
    return false;
  }
}
struct circle{
  point C;
  double r;
  circle(){
  }
};
bool point_in_circle(point P, circle C){
  return dist(P, C.C) < C.r;
}
bool circle_intersects(circle C1, circle C2){
  double d = dist(C1.C, C2.C);
  return abs(C1.r - C2.r) - EPS < d && d < C1.r + C2.r + EPS;
}
void dfs(vector<vector<int>> &E, vector<int> &r, vector<int> &p, vector<int> &d, int r2, int v){
  r[v] = r2;
  for (int w : E[v]){
    if (r[w] == -1){
      d[w] = d[v] + 1;
      p[w] = v;
      dfs(E, r, p, d, r2, w);
    }
  }
}
vector<int> detect_cycle(vector<vector<int>> E){
  int N = E.size();
  vector<int> r(N, -1);
  vector<int> p(N, -1);
  vector<int> d(N, 0);
  for (int i = 0; i < N; i++){
    if (p[i] == -1){
      dfs(E, r, p, d, i, i);
    }
  }
  for (int i = 0; i < N; i++){
    for (int j : E[i]){
      if (r[i] == r[j] && d[j] - d[i] > 1){
        vector<int> c;
        for (int v = j; v != i; v = p[v]){
          c.push_back(v);
        }
        c.push_back(i);
        return c;
      }
    }
  }
  return vector<int>();
}
int main(){
  while (true){
    int n, m;
    cin >> n >> m;
    if (n == 0 && m == 0){
      break;
    }
    vector<circle> C(n);
    for (int i = 0; i < n; i++){
      cin >> C[i].C.x >> C[i].C.y >> C[i].r;
    }
    for (int i = 0; i < m; i++){
      point P, Q;
      cin >> P.x >> P.y >> Q.x >> Q.y;
      set<int> st1, st2;
      for (int j = 0; j < n; j++){
        if (point_in_circle(P, C[j])){
          st1.insert(j);
        }
        if (point_in_circle(Q, C[j])){
          st2.insert(j);
        }
      }
      if (!st1.empty() || !st2.empty()){
        if (st1 == st2){
          cout << "YES";
        } else {
          cout << "NO";
        }
      } else {
        vector<vector<int>> E(n);
        for (int j = 0; j < n; j++){
          for (int k = j + 1; k < n; k++){
            if (circle_intersects(C[j], C[k])){
              E[j].push_back(k);
              E[k].push_back(j);
            }
          }
        }
        bool ok = true;
        while (true){
          vector<int> c = detect_cycle(E);
          if (c.empty()){
            break;
          }
          int cnt = c.size();
          vector<point> p(cnt);
          for (int j = 0; j < cnt; j++){
            p[j] = C[c[j]].C;
          }
          if (point_in_polygon(P, p) != point_in_polygon(Q, p)){
            ok = false;
          }
          for (int j = 0; j < E[c[0]].size(); j++){
            if (E[c[0]][j] == c[1]){
              E[c[0]].erase(E[c[0]].begin() + j);
              break;
            }
          }
          for (int j = 0; j < E[c[1]].size(); j++){
            if (E[c[1]][j] == c[0]){
              E[c[1]].erase(E[c[1]].begin() + j);
              break;
            }
          }
        }
        if (ok){
          cout << "YES";
        } else {
          cout << "NO";
        }
      }
      if (i < m - 1){
        cout << ' ';
      }
    }
    cout << endl;
  }
}

APIO2021 参加記

APIO2021 に参加しました。

競技

Hexagonal Territory を読む。六角グリッドでやばそう。とりあえず Territory (難易度 11) を思い出すがよくわからない。9 点の部分点は数学で簡単に取れそうなので取る。Rainforest Jumps を読む。4 点が自明なので取る。8 点もワーシャルフロイド法をするだけなので、取る。Road Closures を読む。5 点が自明なので取る。7 点も簡単な DP なので取る。

簡単そうな Rainforest Jumps に戻る。小課題 3 の 13 点は距離の配列が得られるので長方形領域の最小値を求めることに帰着。簡単な解法が思いつかなかったのでセグ木を書く。セグ木を 2000 本持たせる  O(N ^ 2 + QN \log N) を投げる。log が付いているので若干不安だが、通った。

小課題 4 を見る。 Q \leq 5 なのでそれぞれの樹から飛び移れる樹を stack を使い  O(N) で取得し、BFS でクエリに答える。コードを投げるが、なかなかジャッジされない。ジャッジキューが詰まっているとの連絡が来ているので、とりあえず別の問題に移る。

Hexagonal Territory の小課題 3 は BFS で解けるとわかっていて実装を後回しにしていたので、実装をして投げる。提出欄を見ると、なんと今までに投げたすべての提出が WJ 状態に戻っていて、リジャッジが行われているとの連絡が来る。このあたりで開始から 1 時間経過。

Rainforest Jumps に戻り、小課題 5 を解く。貪欲に高いほうの樹に飛び移る戦略がだいたい正当なので、ダブリングを実装して投げる。Rainforest Jumps の小課題 3・4 はどうせ次数を持つ木 DP なので、適当に  O(N ^ 2) を実装し投げる。なかなかジャッジが返ってこないので clar を投げる。ジャッジ詰まりの影響でコンテスト時間が 30 分延長になるとの連絡が入る。

残りの問題を考察するが、よくわからない。コンテスト開始から 2 時間経過。ジャッジ詰まりのため各問題に 5 分に一回の提出制限がかけられるとの連絡があり、不安になる。

しばらくして、ジャッジが返り始めるので眺めると、投げた提出が何個か落ちているので確認。Rainforest Jumps の小課題 5 が落ちているので適当に修正し投げる。Hexagonal Territory の愚直 BFS も落ちたので確認すると点数の計算方法を誤っていたので、計算を修正すると通る。Road Closures の木 DP も落ちているので確認すると誤解をしているので直す。コンテストがさらに 30 分延長されて 6 時間になり、提出制限が元に戻る。

Hexagonal Territory と Road Closures は通ったが、Rainforest Jumps が通らないので、ワーシャルフロイド法の結果と比較するランダムチェッカーを書くと、貪欲をすると超えてしまう状態の後に二回以上移動する必要があることがあることを見落としていたことに気付き、右方向への移動だけでのダブリングを別に持ち、いろいろデバッグをすると通る。小課題 6 は二分探索で簡単に小課題 5 に帰着するので、実装。満点も取れそうなのでいろいろ考察し、何回か WA を出しながらなんとか通す。コンテスト開始から約 4 時間が経った。

Hexagonal Territory は自明な小課題以外の得点を取れていないので、面積を求めればよい。ピックの定理のようなものが使えるのではないかと気付き、実装をする。Road Closures の小課題 5 も考察する。マージテクをしたり、木の次数  d の頂点の個数が  \displaystyle{O(\frac{N}{d})} 個であることの利用を考えるが、解法にはたどり着けず終わった。結果は 47+100+38=183。木はどちらかというと得意な分野なのにあまり点数が取れなくて少し悔しい。

終了後

人々の得点を見る。KoD さんと blackyuki さんが Road Closures で満点を取り 200↑ の得点を得ていた。すごい。

結果が発表される。得点は日本 3 位で銀メダルらしい。