CSDN平台比赛第五期第四题：三而竭

本文链接：https://blog.csdn.net/yrt19/article/details/126774118

CSDN平台比赛第五期第四题：三而竭

前言

机缘巧合之下我得知了CSDN举办的这个比赛。为了复习推研机试，就报名参赛准备练习一下。本来在时间截止时只拿了80分，以为没有名次了，不曾想这次比赛报名人少，我的名次还能拿到奖励，实属幸运。领取奖励需要撰写一篇比赛相关的博客，故作此文。

题面描述

一鼓作气再而衰三而竭。小艺总是喜欢把任务分开做。小艺接到一个任务，任务的总任务量是 $n$ 。第一天小艺能完成 $x$ 份任务。第二天能完成 $\frac x k$ ……第 $t$ 天能完成 $\frac x{k^{t-1}}$ 。小艺想知道自己第一天至少完成多少才能完成最后的任务。

关于数据范围，比赛结束后的题解并未提及，凭记忆应当是 $\le n \le 10^9$ ， $k > 2$ ，且 $n$ 、 $x$ 、 $k$ 均为正整数。

思路分析

其实这个题目描述还是有点不清晰的，按最后的结果来看，实际上第 $t$ 天能完成的任务量应该是 $\lfloor \frac x{k^{t-1}}\rfloor$ 。

思考这个问题首先可以从一个数学结论下手： $\lfloor \frac x{k^{t}}\rfloor=\lfloor \frac {\lfloor\frac x {k^{t-1}}\rfloor}{k}\rfloor$ 。要证明这个结论，可以设 $\lfloor \frac x{k^{t}}\rfloor=a$ ，则有 $ak^t\le x\le (a+1)k^t-1$ ，所以 $ak\le\frac x{k^{t-1}}\le (a+1)k-\frac1{k^{t-1}}$ ，有 $ak\le\lfloor\frac x{k^{t-1}}\rfloor\le (a+1)k-1$ ，那么自然能得出 $\lfloor \frac {\lfloor\frac x {k^{t-1}}\rfloor}{k}\rfloor=a=\lfloor \frac x{k^{t}}\rfloor$ 。所以说我们可以利用这个性质不断展开 $\lfloor \frac x{k^{t}}\rfloor$ ，如果看做一个关于 $t$ 的数列（不妨记作 ${c_t\}$ ， $\{c_t\}=\lfloor \frac x{k^{t}}\rfloor$ ），后一项就能通过前一项导出（ $c_t=\lfloor\frac{c_{t-1}}{k}\rfloor$ ），题目要求的是这个无限数列的和（不妨记作 $S$ ）要不小于给定值。

先给 $x$ 找一个大致区间，首先 $x\le n$ ，否则第一天就直接把所有任务完成了。其次由下取整的性质可以知道 $n\le S\le x+\frac x k+\frac x {k^2}+...=x\times\frac k {k-1}$ ，故 $x\ge\frac{(k-1)n}{k}$ ，这个夹的区间长度是 $\frac n k$ ，由题目的数据范围可知这个区间长度最大也就 $\frac{10^9}{2}$ ，只要我们验证其间每个 $x$ 的值是否符合题目条件所需的算法复杂度不那么高，就完全足够支撑我们直接对这个区间进行遍历搜索（其实这个时候可以二分查找，但是反正遍历也能过，就没必要写二分了）。

要保证单次查询的复杂度足够低，可以用类似dp的思路，记录上一次搜索的状态，在下一次搜索中加以利用。由之前的提到的数学结论，我们有 $c_t=\lfloor\frac{c_{t-1}}{k}\rfloor$ ，也就是只要从某一项开始与上次搜索相同，那么后续的所有项（以及它们的和）也就完全与上一次搜索相同。由于 $k\ge2$ ， $x\le n\le10^9$ ，那么 $c_t$ 最多到 $30$ 项时就开始全为 $0$ 了，我们就可以开个 $30$ 项的数组，第 $t$ 个项记录 $c_t$ 的值与从 $c_t$ 开始所有项的和，那么每次计算就不用重复算出所有 $c_t$ 的值了——因为它们大多数都与上一次搜索相同。通过这个小技巧，如果 $x$ 从下界开始逐步加一进行遍历，那么每隔 $k$ 项才会更新一次 $c_1$ ，每隔 $k^2$ 项才会更新一次 $c_2$ ，每隔 $k^3$ 项才会更新一次 $c_3$ ……，按期望来看每次对 $x$ 的搜索验算复杂度只有 $1+\frac{1}{k}+\frac{1}{k^2}+\frac{1}{k^3}+...=\frac{k}{k-1}\le2$ ，完全能支撑我们的逐个遍历算法。

最终代码

这份代码是能通过比赛中本题的所有测例的：

// 请关闭中文输入法，用英文的字母和标点符号。
// 如果你想运行系统测试用例，请点击【执行代码】按钮，如果你想提交作答结果，请点击【提交】按钮，
// 注意：除答案外，请不要打印其他任何多余的字符，以免影响结果验证
// 本OJ系统是基于 OxCoder 技术开发，网址：www.oxcoder.com
// 模版代码提供基本的输入输出框架，可按个人代码习惯修改

#include <iostream>
#include <string>
#include <sstream>
#include <vector>

struct Info {
    int sum;
    int now;
};

int info_len = 30;
Info* info_array = new Info[info_len];

void renew(int index, int now, int k) {
    if (now == info_array[index].now) return;
    info_array[index].now = now;
    if (index < info_len - 1) {
        renew(index + 1, now / k, k);
        info_array[index].sum = info_array[index + 1].sum + now;
    } else {
        info_array[index].sum = now;
    }
}

int solution(std::vector<int>& vec){
    int n = vec[0], k = vec[1];
    int result = (k - 1) * n / k;
    info_array[0].now = result;
    for (int i = 1; i < info_len; ++i) {
        info_array[i].now = info_array[i - 1].now / k;
    }
    info_array[info_len - 1].sum = info_array[info_len - 1].now;
    for (int i = info_len - 2; i >= 0; --i) {
        info_array[i].sum = info_array[i].now + info_array[i + 1].sum;
    }
    for ( ; info_array[0].sum < n; renew(0, ++result, k));
    return result;
}

int main() {
    std::vector<int> vec;
    std::string line_0, token_0;
    getline(std::cin >> std::ws, line_0);
    std::stringstream tokens_0(line_0);
    while(std::getline(tokens_0, token_0, ' ')){
        vec.push_back(std::stoi(token_0));
    }
    int result = solution(vec);
    std::cout << result << std::endl;
    return 0;
}