結果

問題 No.2287 ++ -- *=a /=a
ユーザー risujirohrisujiroh
提出日時 2023-04-29 00:20:29
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 224 ms / 2,000 ms
コード長 3,580 bytes
コンパイル時間 4,084 ms
コンパイル使用メモリ 270,144 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-04-29 01:05:34
合計ジャッジ時間 4,901 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 19 ms
5,376 KB
testcase_07 AC 19 ms
5,376 KB
testcase_08 AC 19 ms
5,376 KB
testcase_09 AC 14 ms
5,376 KB
testcase_10 AC 11 ms
5,376 KB
testcase_11 AC 11 ms
5,376 KB
testcase_12 AC 9 ms
5,376 KB
testcase_13 AC 10 ms
5,376 KB
testcase_14 AC 9 ms
5,376 KB
testcase_15 AC 10 ms
5,376 KB
testcase_16 AC 8 ms
5,376 KB
testcase_17 AC 9 ms
5,376 KB
testcase_18 AC 91 ms
5,376 KB
testcase_19 AC 50 ms
5,376 KB
testcase_20 AC 39 ms
5,376 KB
testcase_21 AC 33 ms
5,376 KB
testcase_22 AC 30 ms
5,376 KB
testcase_23 AC 28 ms
5,376 KB
testcase_24 AC 26 ms
5,376 KB
testcase_25 AC 25 ms
5,376 KB
testcase_26 AC 24 ms
5,376 KB
testcase_27 AC 224 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#if __INCLUDE_LEVEL__

void solve() {
  auto [x, y, a] = input<int64_t, 3>();
  auto go = [&](int64_t n) {
    vector<int64_t> ret;
    while (n) {
      ret.push_back(n % a);
      n /= a;
    }
    return ret;
  };
  auto vx = go(x);
  auto vy = go(y);
  show(vx);
  show(vy);
  vector<array<int64_t, 2>> f(size(vy) + 1);
  f[0][1] = 1;
  for (i : iota(1, ssize(f))) {
    f[i][0] = 1 + min(vy[i - 1] + f[i - 1][0], vy[i - 1] + 1 + f[i - 1][1]);
    f[i][1] =
        1 + min(a - vy[i - 1] + f[i - 1][0], a - vy[i - 1] - 1 + f[i - 1][1]);
  }
  auto eval = [&](auto&& v) {
    int64_t ret = 0;
    for (e : v | reverse) {
      ret *= a;
      ret += e;
    }
    return ret;
  };
  show(f);
  int64_t ans = abs(x - y);
  vector<int64_t> memo(ssize(vx) + 1);
  for (k : iota(0, ssize(memo))) {
    memo[k] = eval(vx | drop(k));
  }
  for (i : iota(0, ssize(vy) + 1)) {
    auto vy2 = ALL(vector, vy | drop(i));
    for (ty = eval(vy2); int k : iota(0, ssize(vx) + 1)) {
      ans = min(ans, k + abs(memo[k] - ty) + f[i][0]);
    }
    {
      int j = 0;
      while (j < ssize(vy2) && vy2[j] + 1 == a) {
        vy2[j] = 0;
        ++j;
      }
      if (j < ssize(vy2)) {
        ++vy2[j];
      } else {
        vy2.push_back(1);
      }
    }
    for (ty = eval(vy2); int k : iota(0, ssize(vx) + 1)) {
      ans = min(ans, k + abs(memo[k] - ty) + f[i][1]);
    }
  }
  print(ans);
}

#else  // __INCLUDE_LEVEL__

#include <bits/stdc++.h>

namespace std {

template <class T>
auto operator>>(istream& is, T&& t) -> decltype(tuple_cat(t), is) {
  return apply([&is](auto&... xs) -> istream& { return (is >> ... >> xs); }, t);
}

template <class T>
auto operator<<(ostream& os, T&& t) -> decltype(tuple_cat(t), os) {
  auto f = [&os](auto&... xs) -> ostream& {
    [[maybe_unused]] auto sep = "";
    ((os << exchange(sep, " ") << xs), ...);
    return os;
  };
  return apply(f, t);
}

template <class T>
istream& operator>>(istream& is, vector<T>& v) {
  for (auto&& e : v) {
    is >> e;
  }
  return is;
}

template <class T>
ostream& operator<<(ostream& os, const vector<T>& v) {
  static constexpr auto SEP = is_convertible_v<int, T> ? " " : "\n";
  for (auto sep = ""; const auto& e : v) {
    os << exchange(sep, SEP) << e;
  }
  return os;
}

}  // namespace std

template <class... Ts, class... Args>
auto input(Args&&... args) {
  if constexpr (sizeof...(Ts) == 0) {
    static_assert(sizeof...(Args) == 0);
    return input<int>();
  } else if constexpr (1 < sizeof...(Ts)) {
    static_assert(sizeof...(Args) == 0);
    return input<std::tuple<Ts...>>();
  } else {
    using T = std::tuple_element_t<0, std::tuple<Ts...>>;
    if constexpr (sizeof...(args)) {
      T x(std::forward<Args>(args)...);
      std::cin >> x;
      return x;
    } else {
      T x;
      std::cin >> x;
      return x;
    }
  }
}

template <class T, int N>
std::array<T, N> input() {
  return input<std::array<T, N>>();
}

template <class... Ts>
void print(Ts&&... xs) {
  std::cout << std::tie(xs...) << '\n';
}

using namespace std;

#define show(...) static_cast<void>(0)
#define for(...) for ([[maybe_unused]] auto&& __VA_ARGS__)
#define lambda(...) [&](auto&&... args) { return __VA_ARGS__; }
#define $1 get<0>(tie(args...))
#define $2 get<1>(tie(args...))
#define ALL(f, r, ...) lambda(f(begin($1), end($1), ##__VA_ARGS__))(r)

namespace std::ranges::views {

#include __BASE_FILE__

}  // namespace std::ranges::views

int main() {
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  for (_ : views::iota(0, input())) {
    views::solve();
  }
}

#endif  // __INCLUDE_LEVEL__
0