B. Our Tanya is Crying Out Loud - Codeforces Round 466 (Div. 2)
Solutions Codeforces 贪心
Lastmod: 2025-01-03 周五 18:55:25

https://codeforces.com/contest/940/problem/B

题目大意

给定数字 $n, k, a, b (1 \le n, k, a, b \le 2 \times 10^9)$,使用两种操作将 $n$ 变为 $1$,

  1. 花费 $a$ 使 $n$ 减 $1$。
  2. 花费 $b$ 使 $n$ 变为 $n / k$ (仅在 $k | n$ 时可用)。

简要题解

显然贪心即可,对于每次先将 $n$ 减到 $k$ 的某个倍数,然后判断哪个操作更好。

特别注意 $k$ 可以等于 $1$!

复杂度

$T$:$O(\log_k(N))$

$S$:$O(1)$

代码实现

#include <bits/stdc++.h>
using namespace std;

int io_=[](){ ios::sync_with_stdio(false); cin.tie(nullptr); return 0; }();

using LL = long long;
using ULL = unsigned long long;
using LD = long double;
using PII = pair<int, int>;
using VI = vector<int>;
using MII = map<int, int>;

template<typename T> void cmin(T &x,const T &y) { if(y<x) x=y; }
template<typename T> void cmax(T &x,const T &y) { if(x<y) x=y; }
template<typename T> bool ckmin(T &x,const T &y) { 
    return y<x ? (x=y, true) : false; }
template<typename T> bool ckmax(T &x,const T &y) { 
    return x<y ? (x=y, true) : false; }
template<typename T> void cmin(T &x,T &y,const T &z) {// x<=y<=z 
    if(z<x) { y=x; x=z; } else if(z<y) y=z; }
template<typename T> void cmax(T &x,T &y,const T &z) {// x>=y>=z
    if(x<z) { y=x; x=z; } else if(y<z) y=z; }

// mt19937 rnd(chrono::system_clock::now().time_since_epoch().count());
// mt19937_64 rnd_64(chrono::system_clock::now().time_since_epoch().count());

/*
---------1---------2---------3---------4---------5---------6---------7---------
1234567890123456789012345678901234567890123456789012345678901234567890123456789
*/

void solve() {
  int n, k, a, b; cin >> n >> k >> a >> b;

  LL ans = 0;
  while (k != 1 && n >= k) {
    int diff = n % k;
    ans += 1LL * diff * a;
    n -= diff;
    LL ca = 1LL * (n - n / k) * a;
    if (ca < b) {
      ans += ca;
    } else {
      ans += b;
    }
    n /= k;
  }

  ans += 1LL * (n - 1) * a;

  cout << ans << '\n';
}

int main() {
  int t = 1; 
  // cin >> t; 
  while (t--) {
    solve();
  }
  return 0;
}
Prev: C. Phone Numbers - Codeforces Round 466 (Div. 2)
Next: A. Points on the line - Codeforces Round 466 (Div. 2)