結果

問題 No.2287 ++ -- *=a /=a
コンテスト
ユーザー risujiroh
提出日時 2023-04-29 00:11:36
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 3,551 bytes
コンパイル時間 3,124 ms
コンパイル使用メモリ 268,276 KB
実行使用メモリ 13,640 KB
最終ジャッジ日時 2024-11-17 22:54:18
合計ジャッジ時間 11,594 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 26 TLE * 1
権限があれば一括ダウンロードができます

ソースコード

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 d = [&](auto&& vx, auto&& vy) {
    int64_t ret = 0;
    int64_t base = 1;
    for (i : iota(0, max(ssize(vx), ssize(vy)))) {
      ret += ((i < ssize(vy) ? vy[i] : 0) - (i < ssize(vx) ? vx[i] : 0)) * base;
      base *= a;
    }
    return abs(ret);
  };
  show(f);
  int64_t ans = abs(x - y);
  for (k : iota(0, ssize(vx) + 1)) {
    auto vx2 = vx | drop(k);
    for (i : iota(0, ssize(vy) + 1)) {
      auto vy2 = ALL(vector, vy | drop(i));
      ans = min(ans, k + d(vx2, vy2) + 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);
        }
      }
      ans = min(ans, k + d(vx2, vy2) + 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