結果

問題 No.95 Alice and Graph
ユーザー te-shte-sh
提出日時 2017-05-17 14:47:15
言語 D
(dmd 2.106.1)
結果
AC  
実行時間 1,785 ms / 5,000 ms
コード長 1,931 bytes
コンパイル時間 939 ms
コンパイル使用メモリ 105,552 KB
実行使用メモリ 21,388 KB
最終ジャッジ日時 2024-06-12 19:18:15
合計ジャッジ時間 7,442 ms
ジャッジサーバーID
(参考情報)
judge1 / judge5
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 566 ms
19,504 KB
testcase_01 AC 84 ms
8,792 KB
testcase_02 AC 80 ms
7,368 KB
testcase_03 AC 2 ms
6,940 KB
testcase_04 AC 866 ms
20,992 KB
testcase_05 AC 683 ms
13,720 KB
testcase_06 AC 851 ms
20,784 KB
testcase_07 AC 83 ms
10,052 KB
testcase_08 AC 3 ms
6,940 KB
testcase_09 AC 1 ms
6,940 KB
testcase_10 AC 1 ms
6,940 KB
testcase_11 AC 4 ms
6,940 KB
testcase_12 AC 3 ms
6,944 KB
testcase_13 AC 1,785 ms
21,388 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

import std.algorithm, std.conv, std.range, std.stdio, std.string;

void main()
{
  auto rd1 = readln.split.to!(size_t[]), n = rd1[0], m = rd1[1], k = rd1[2];
  auto gij = new int[][](n, n);
  foreach (_; 0..m) {
    auto rd2 = readln.split.to!(size_t[]), u = rd2[0] - 1, v = rd2[1] - 1;
    gij[u][v] = gij[v][u] = 1;
  }

  auto wij = gij.warshalFloyd;

  auto calc(size_t[] ai)
  {
    auto l = ai.length;

    auto dp = new int[][](1 << l, l);
    foreach (i; 0..(1 << l)) dp[i][] = int.max;
    dp[0][0] = 0;

    foreach (i; 1..(1 << l)) {
      foreach (j1; 0..l) {
        if (!i.bitTest(j1)) continue;
        if (i.bitComp(j1) == 0) {
          if (wij[0][ai[j1]]) dp[i][j1] = wij[0][ai[j1]];
        } else {
          foreach (j2; 0..l) {
            if (j1 == j2 || !i.bitTest(j2) || !wij[ai[j1]][ai[j2]]) continue;
            dp[i][j1] = min(dp[i][j1], dp[i.bitComp(j1)][j2] + wij[ai[j1]][ai[j2]]);
          }
        }
      }
    }

    return dp[$-1].any!(a => a <= k);
  }

  size_t[] ai;
  foreach_reverse (i; 1..n) {
    if (calc(ai ~ i))
      ai ~= i;
    if (ai.length >= k)
      break;
  }

  writeln(ai.map!(a => 2 ^^ a - 1).sum);
}

import std.traits;

T[][] warshalFloyd(T)(const T[][] a)
  if (isNumeric!T)
{
  import std.algorithm;

  auto n = a.length;

  auto b = new T[][](n);
  foreach (i; 0..n) b[i] = a[i].dup;

  foreach (k; 0..n)
    foreach (i; 0..n)
      foreach (j; 0..n)
        if ((i == k || b[i][k]) && (k == j || b[k][j])) {
          auto s = b[i][k] + b[k][j];
          if (i == j || b[i][j])
            b[i][j] = min(b[i][j], s);
          else
            b[i][j] = s;
        }

  return b;
}

pragma(inline) {
  pure bool bitTest(T)(T n, size_t i) { return (n & (T(1) << i)) != 0; }
  pure T bitSet(T)(T n, size_t i) { return n | (T(1) << i); }
  pure T bitReset(T)(T n, size_t i) { return n & ~(T(1) << i); }
  pure T bitComp(T)(T n, size_t i) { return n ^ (T(1) << i); }
}
0