結果

問題 No.1508 Avoid being hit
ユーザー te-shte-sh
提出日時 2021-06-08 22:32:46
言語 D
(dmd 2.107.1)
結果
WA  
実行時間 -
コード長 4,880 bytes
コンパイル時間 1,930 ms
コンパイル使用メモリ 212,120 KB
実行使用メモリ 15,936 KB
最終ジャッジ日時 2023-09-04 12:51:34
合計ジャッジ時間 8,234 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 91 ms
11,568 KB
testcase_01 AC 1 ms
4,380 KB
testcase_02 AC 1 ms
4,384 KB
testcase_03 WA -
testcase_04 AC 2 ms
4,384 KB
testcase_05 AC 1 ms
4,384 KB
testcase_06 AC 1 ms
4,380 KB
testcase_07 AC 1 ms
4,380 KB
testcase_08 WA -
testcase_09 WA -
testcase_10 AC 2 ms
4,380 KB
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 AC 1 ms
4,380 KB
testcase_16 AC 34 ms
8,620 KB
testcase_17 AC 19 ms
6,516 KB
testcase_18 AC 21 ms
6,452 KB
testcase_19 AC 27 ms
7,596 KB
testcase_20 AC 51 ms
11,760 KB
testcase_21 AC 33 ms
7,948 KB
testcase_22 AC 40 ms
8,872 KB
testcase_23 AC 46 ms
11,272 KB
testcase_24 WA -
testcase_25 AC 101 ms
11,272 KB
testcase_26 AC 8 ms
4,384 KB
testcase_27 WA -
testcase_28 WA -
testcase_29 AC 101 ms
12,836 KB
testcase_30 AC 55 ms
8,020 KB
testcase_31 AC 88 ms
11,244 KB
testcase_32 AC 95 ms
12,232 KB
testcase_33 WA -
testcase_34 AC 94 ms
11,920 KB
testcase_35 AC 11 ms
5,956 KB
testcase_36 AC 56 ms
7,376 KB
testcase_37 AC 25 ms
5,580 KB
testcase_38 AC 77 ms
11,012 KB
testcase_39 AC 51 ms
7,284 KB
testcase_40 AC 96 ms
10,988 KB
testcase_41 AC 57 ms
7,676 KB
testcase_42 AC 98 ms
11,008 KB
testcase_43 AC 116 ms
15,936 KB
testcase_44 AC 2 ms
4,384 KB
testcase_45 AC 2 ms
4,384 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

// URL: https://yukicoder.me/problems/no/1508

import std;

version(unittest) {} else
void main()
{
  int N, Q; io.getV(N, Q);
  int[] A; io.getA(Q, A); --A[];
  int[] B; io.getA(Q, B); --B[];

  auto s = new Section!int[][](Q+1);
  s[0] ~= section(0, N);

  foreach (i, Ai, Bi; lockstep(A, B)) {
    if (Ai > Bi) swap(Ai, Bi);
    foreach (si; s[i]) {
      auto t = section(max(0, si.start-1), min(N, si.end+1));

      auto avoidHit(Section!int t, int a)
      {
        if (t.include(a)) {
          if (t.length == 1) return section(-1, 0);
          if (a == t.start) {
            t = section(t.start+1, t.end);
          } else if (a == t.end-1) {
            t = section(t.start, t.end-1);
          } else {
            auto ta = section(t.start, a);
            if (!s[i+1].empty && mergeable(s[i+1].back, ta))
              s[i+1][$-1] = merge(s[i+1][$-1], ta);
            else
              s[i+1] ~= ta;
            t = section(a+1, t.end);
          }
        }
        return t;
      }

      t = avoidHit(t, Ai);
      if (t.start == -1) continue;
      t = avoidHit(t, Bi);
      if (t.start == -1) continue;

      if (!s[i+1].empty && mergeable(s[i+1].back, t))
        s[i+1][$-1] = merge(s[i+1][$-1], t);
      else
        s[i+1] ~= t;
    }
  }

  if (s[$-1].empty) {
    io.put("NO");
  } else {
    auto r = new int[](Q+1), i = 0;
    for (; i < N; ++i)
      if (s[$-1].any!(si => si.include(i)))
        break;

    foreach (j; 0..Q) {
      for (auto k = i-1; k <= i+1; ++k)
        if (s[$-j-2].any!(si => si.include(k))) {
          r[j+1] = i = k;
          break;
        }
    }

    io.put("YES");
    foreach_reverse (ri; r) io.put(ri+1);
  }
}

struct Section(T)
{
  T start, end;

  pure nothrow @nogc @safe
  {
    this(T start, T end)
      in { assert(start < end); }
    do
    {
      this.start = start;
      this.end = end;
    }

    auto length() { return end - start; }
  }
}

pure nothrow @nogc @safe
{
  auto section(T)(T start, T end)
    in { assert(start < end); }
  do
  {
    return Section!T(start, end);
  }

  bool include(T)(Section!T a, T b)
  {
    return a.start <= b && b < a.end;
  }
  bool overlaped(T)(Section!T a, Section!T b)
  {
    return a.start < b.end && a.end > b.start;
  }
  bool mergeable(T)(Section!T a, Section!T b)
  {
    return overlaped(a, b) || a.start == b.end || a.end == b.start;
  }
  auto merge(T)(Section!T a, Section!T b)
    in { assert(mergeable(a, b)); }
  do
  {
    return section(min(a.start, b.start), max(a.end, b.end));
  }
}

auto io = IO!()();
struct IO(alias IN = stdin, alias OUT = stdout)
{
  import core.stdc.stdlib : exit;

  void getV(T...)(ref T v)
  {
    foreach (ref w; v) get(w);
  }
  void getA(T)(size_t n, ref T v)
    if (hasAssignableElements!T)
  {
    v = new T(n);
    foreach (ref w; v) get(w);
  }
  void getC(T...)(size_t n, ref T v)
    if (allSatisfy!(hasAssignableElements, T))
  {
    foreach (ref w; v) w = new typeof(w)(n);
    foreach (i; 0..n) foreach (ref w; v) get(w[i]);
  }
  void getM(T)(size_t r, size_t c, ref T v)
    if (hasAssignableElements!T && hasAssignableElements!(ElementType!T))
  {
    v = new T(r);
    foreach (ref w; v) getA(c, w);
  }
  template getS(E...)
  {
    void getS(T)(size_t n, ref T v)
    {
      v = new T(n);
      foreach (ref w; v) foreach (e; E) mixin("get(w."~e~");");
    }
  }

  const struct PutConf
  {
    bool newline = true, flush, exit;
    string floatFormat = "%.10f", delimiter = " ";
  }

  void put(alias conf = "{}", T...)(T v)
  {
    putMain!conf(v);
  }
  void putB(alias conf = "{}", S, T)(bool c, S t, T f)
  {
    if (c) put!conf(t);
    else put!conf(f);
  }
  void putRaw(T...)(T v)
  {
    OUT.write(v);
    OUT.writeln;
  }

  private
  {
    dchar[] buf;
    auto sp = (new dchar[](0)).splitter;

    void nextLine()
    {
      IN.readln(buf);
      sp = buf.splitter;
    }
    void get(T)(ref T v)
    {
      if (sp.empty) nextLine();
      v = sp.front.to!T;
      sp.popFront();
    }

    void putMain(alias conf, T...)(T v)
    {
      mixin("const PutConf c = "~conf~";");
      foreach (i, w; v) {
        putOne!conf(w);
        if (i+1 < v.length) OUT.write(c.delimiter);
      }
      static if (c.newline) OUT.writeln;
      static if (c.flush) OUT.flush();
      static if (c.exit) exit(0);
    }
    void putOne(alias conf, T)(T v)
    {
      mixin("const PutConf c = "~conf~";");
      static if (isInputRange!T && !isSomeString!T) putRange!conf(v);
      else static if (isFloatingPoint!T) OUT.write(format(c.floatFormat, v));
      else static if (hasMember!(T, "fprint")) v.fprint(OUT);
      else OUT.write(v);
    }
    void putRange(alias conf, T)(T v)
    {
      mixin("const PutConf c = "~conf~";");
      auto w = v;
      while (!w.empty) {
        putOne!conf(w.front); w.popFront();
        if (!w.empty) OUT.write(c.delimiter);
      }
    }
  }
}
0