結果

問題 No.1170 Never Want to Walk
コンテスト
ユーザー allegrogiken
提出日時 2020-08-15 00:38:53
言語 D
(dmd 2.109.1)
結果
RE  
実行時間 -
コード長 2,619 bytes
コンパイル時間 838 ms
コンパイル使用メモリ 114,344 KB
実行使用メモリ 11,316 KB
最終ジャッジ日時 2024-06-22 08:04:19
合計ジャッジ時間 2,780 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 25 RE * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

void main() {
  problem();
}

void problem() {
  auto N = scan!int;
  auto A = scan!long;
  auto B = scan!long;
  auto S = scan!long(N);

  const LEFT_LIMIT = 0;
  const RIGHT_LIMIT = S[$ - 1];

  int[] Z = new int[RIGHT_LIMIT + 1];
  int[long] ZP;
  auto ans = new long[N];

  void solve() {
    foreach(i; 0..N-1) {
      Z[S[i] .. S[i + 1] ] = i + 1;
      ZP[S[i]] = i;
    }
    ZP.deb;
    Z[$-1] = N;

    foreach(stationIndex; 0..N) {
      if (ans[stationIndex] > 0) continue;

      bool[int] visits = [stationIndex: true];
      bool[int] newVisits = visits;

      while(true) {
        bool[int] nextVisits;
        
        foreach(v; newVisits.keys) {
          visits[v] = true;

          if (S[v] + A <= RIGHT_LIMIT) {
            auto right_s = S[v] + A > RIGHT_LIMIT ? RIGHT_LIMIT : S[v] + A;
            auto right_e = S[v] + B > RIGHT_LIMIT ? RIGHT_LIMIT : S[v] + B;
              if (right_s in ZP && !(ZP[right_s] in visits)) nextVisits[ZP[right_s]] = true;
              foreach(n; Z[right_s] .. Z[right_e]) {
                if (!(n in visits)) nextVisits[n] = true;
              }
          }
          
          if (S[v] - A >= LEFT_LIMIT) {
            auto left_s = S[v] - B < LEFT_LIMIT ? LEFT_LIMIT : S[v] - B;
            auto left_e = S[v] - A < LEFT_LIMIT ? LEFT_LIMIT : S[v] - A;
              if (left_s in ZP && !(ZP[left_s] in visits)) {
                nextVisits[ZP[left_s]] = true;
              }
              foreach(n; Z[left_s] .. Z[left_e]) {
                if (!(n in visits)) nextVisits[n] = true;
              }
          }
        }

        if (nextVisits.length == 0) break;
        newVisits = nextVisits;
        [[newVisits]].deb;
      }

      visits.deb;
      const reach = visits.length;
      foreach(v; visits.keys) {
        ans[v] = reach;
      }
    }

    foreach(a; ans) writeln(a);
  }

  solve();
}

// ----------------------------------------------

import std.stdio, std.conv, std.array, std.string, std.algorithm, std.container, std.range, core.stdc.stdlib, std.math, std.typecons, std.numeric;
T[][] combinations(T)(T[] s, in int m) {   if (!m) return [[]];   if (s.empty) return [];   return s[1 .. $].combinations(m - 1).map!(x => s[0] ~ x).array ~ s[1 .. $].combinations(m); }
string scan(){ static string[] ss; while(!ss.length) ss = readln.chomp.split; string res = ss[0]; ss.popFront; return res; }
T scan(T)(){ return scan.to!T; }
T[] scan(T)(int n){ return n.iota.map!(i => scan!T()).array; }
void deb(T ...)(T t){ debug writeln(t); }
alias Point = Tuple!(long, "x", long, "y");

// -----------------------------------------------
0