#117716 (C++11) No.3 ビットすごろく

• 問題: No.3 ビットすごろく
• ユーザー: wikeaka
• 提出日時: 2016-09-17 16:32:08
• 言語: C++11 (gcc 11.4.0)
• 結果: AC
• 実行時間: 425 ms / 5,000 ms
• コード長: 1,520 bytes
• コンパイル時間: 733 ms
• コンパイル使用メモリ: 71,652 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-07-01 08:06:07
• 合計ジャッジ時間: 6,579 ms
• ジャッジサーバーID （参考情報）: judge1 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	1 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	25 ms; 5,376 KB
testcase_04	AC	4 ms; 5,376 KB
testcase_05	AC	120 ms; 5,376 KB
testcase_06	AC	31 ms; 5,376 KB
testcase_07	AC	11 ms; 5,376 KB
testcase_08	AC	68 ms; 5,376 KB
testcase_09	AC	181 ms; 5,376 KB
testcase_10	AC	256 ms; 5,376 KB
testcase_11	AC	151 ms; 5,376 KB
testcase_12	AC	115 ms; 5,376 KB
testcase_13	AC	19 ms; 5,376 KB
testcase_14	AC	270 ms; 5,376 KB
testcase_15	AC	414 ms; 5,376 KB
testcase_16	AC	318 ms; 5,376 KB
testcase_17	AC	361 ms; 5,376 KB
testcase_18	AC	13 ms; 5,376 KB
testcase_19	AC	378 ms; 5,376 KB
testcase_20	AC	3 ms; 5,376 KB
testcase_21	AC	2 ms; 5,376 KB
testcase_22	AC	277 ms; 5,376 KB
testcase_23	AC	425 ms; 5,376 KB
testcase_24	AC	383 ms; 5,376 KB
testcase_25	AC	371 ms; 5,376 KB
testcase_26	AC	2 ms; 5,376 KB
testcase_27	AC	20 ms; 5,376 KB
testcase_28	AC	309 ms; 5,376 KB
testcase_29	AC	157 ms; 5,376 KB
testcase_30	AC	1 ms; 5,376 KB
testcase_31	AC	2 ms; 5,376 KB
testcase_32	AC	133 ms; 5,376 KB

ソースコード

#include <iostream>
#include <algorithm>
#include <vector>
#include <bitset>
#include <limits>
#include <numeric>

using namespace std;


using Vertex = int;
using Edge = pair<int, int>;
using Graph = pair<vector<Vertex>, vector<Edge>>;

int step(const int x) {
  const int BIT_MAX = 14;
  bitset<BIT_MAX> bs(x);
  return bs.count();
}

vector<Edge> makeEdges(const int N) {
  vector<Edge> e;

  for (int i=0; i<N; ++i) {
    int s = step(i+1);
    if (0 <= i-s && i-s < N) e.push_back(Edge(i, i-s));
    if (0 <= i+s && i+s < N) e.push_back(Edge(i, i+s));
  }
  return e;
}

int dijkstra(const vector<Edge> e, const int start, const int goal) {
  vector<float> costs(goal - start + 1, numeric_limits<float>::infinity());
  costs[start] = 0;
  vector<int> t(goal - start + 1);
  iota(t.begin(), t.end(), 0);

  int min_v = start;
  while (min_v != goal) {
    for (auto x : e) {
      if (x.first != min_v) continue;
      if (find(t.begin(), t.end(), x.second) == t.end()) continue;
      if (costs[x.second] > costs[x.first] + 1) costs[x.second] = costs[x.first] + 1;
    }

    t.erase(remove(t.begin(), t.end(), min_v));

    min_v = t[0];
    for (auto x : t) {
      if (costs[min_v] > costs[x]) min_v = x;
    }
  }
  if (costs[goal] == numeric_limits<float>::infinity()) return -1;
  return costs[goal];
}

int main() {
  int N;
  cin >> N;

  const vector<Edge> e = makeEdges(N);
  int answer = dijkstra(e, 0, N-1);
  if (answer < 0)
    cout << -1;
  else
    cout << answer + 1;
  cout << endl;
  return 0;
}