AtCoder Beginner Contest 336 A-E 题解

比赛链接：https://atcoder.jp/contests/abc336
比赛时间：2024 年 1 月 14 日 20:00-21:40

A题：Long Loong

标签：模拟
题意：给定一个$n$，输出$L$、$n$个$o$和$ng$。
题解：按题意模拟即可。
代码：

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

int main() {
    cout << "L";
    int n;
    cin >> n;
    for (int i = 1; i <= n; i++)
        cout <<  "o";
    cout << "ng";
    return 0;
}

B题：CTZ

标签：模拟
题意：给定一个十进制数$n$，求该数转换成$2$进制之后末尾连续$0$的个数。（$1<=n<=10^9$）
题解：模拟，二进制转换，累计一下末尾$0$个数，直到不是$0$，跳出循环。
代码：

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

int main() {
    int n, c = 0;
    cin >> n;
    while (n) {
        if (n % 2 == 0) c++;
        else break;
        n /= 2;
    }
    cout << c;
    return 0;
}

C题：Even Digits

标签：思维、数学
题意：给出定义十进制数$x$的所有数位都是偶数（即$0、2、4、6、8$）的时候称为 “好整数”，给定一个$n$，求第$n$小的 “好整数”。（$1<=n<=10^{12}$）
前几个 “好整数” 分别为$0、2、4、6、8、20、22、24、26、\mathrm{28...}$
题解：每个数位都是最多$5$种可能，我们可以先求出超过$n$需要的位数，比如第$8$小的，一位最多只有$5$种，两位能到$25$种，所以至少得两位。
接着我们再求目前当前这位对应的到底是$0、2、4、6、8$中的哪一个。
比如第$8$小的，我们求第一位上的数字的时候，
$1?5<8$
$2?5>8$
所以我们能确定，第一位上的数字是$2$。对于后面的第二位来说，我们需要把刚刚第一位带来的整个部分的个数给减掉，$8?5=3$。
$1?1<3$
$1?2<3$
$1?3>=3$
所以第二位上的数字是$4$。
代码：

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

int main() {
    long long n, k = 1, c = 0;
    cin >> n;
    while (k <= n) {
        k *= 5;
        c++;
    }
    for (int i = 1; i <= c; i++) {
        k /= 5;
        for (int j = 1; j <= 5; j++) {
            if (j * k >= n) {
                n -= (j-1) * k;
                cout << (j - 1) * 2;
                break;
            }
        }
    }
    return 0;
}

D题：Pyramid

标签：动态规划、前缀和
题意：金字塔型序列：$1、\mathrm{2...}k?1、k、k?\mathrm{1...2}、1$。给定一个长度为$n$的序列$a_i$，可以进行重复性的两种操作：

1. 将序列中某个数的大小减一
2. 删除第一个或最后一个数

求能够形成的金字塔型序列的最大长度。（$1<=n<=2?10^5，1<=a_i<=10^9$）
题解：比较常见的套路，洛谷也有类似的题：P1091 合唱队形
从左往右维护一个$pre[i]$：以$a_i$作为结尾的最长左金字塔序列的长度
从右往左维护一个$suf[i]$：以$a_i$作为结尾的最长右金字塔序列的长度
我们以$pre[i]$为例，分别来观察一下
例子$1$：$1、2、3$
例子$2$：$1、2、2$
例子$3$：$3、1、2$
按照题目中能把数变小和删除前后数字的操作，能推出当前的$pre[i]$要从前面的$pre[i?1]$和当前$a_i$较小的那个推过来：
$pre[i]=min(pre[i?1]+1,a[i])$
$suf$同理，最终枚举每个数作为金字塔尖，左边金字塔序列长度和右边金子塔序列长度中取小的那个能够形成的金字塔序列长度，然后维护一个最大值。
代码：

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

const int N = 2e5 + 10;
typedef long long ll;
ll a[N], pre[N], suf[N];

int main() {
    ll n, ans = 0;
    cin >> n;
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
        pre[i] = min(pre[i - 1] + 1, a[i]);
    }
    for (int i = n; i >= 1; i--) {
        suf[i] = min(suf[i + 1] + 1, a[i]);
    }
    for (int i = 1; i <= n; i++) {
        ans = max(ans, min(pre[i], suf[i]));
    }
    cout << ans;
    return 0;
}

E题：Digit Sum Divisible

标签：数位$dp$
题意：给定一个$n$，求小于等于$n$的数中有多少个能被自己的位数之和整除。（$1<=n<=10^{14}$）
**题解：**数位$dp$模版题，$dp[pos][sum][mod]$表示当第$pos$位各位数字之和为$sum$除原数的余数是$mod$且有没超出边界时候的个数。
代码：

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

typedef long long ll;
ll n, m, cnt = 0;
// dp[i][j][k]: 第i位 当前数字和为j 取模结果为k的个数
ll a[20], dp[20][10*20][10*20];

ll dfs(ll pos, ll sum, ll mod, ll flag) {
    if (sum > m) return 0;
    if (pos == 0) return sum == m && mod == 0;
    if (!flag && dp[pos][sum][mod] != -1) return dp[pos][sum][mod];
    ll x = flag ? a[pos] : 9;
    ll ans = 0;
    for (ll i = 0; i <= x; i++) {
        ans += dfs(pos - 1, sum + i, (mod * 10 + i) % m, flag && (i == a[pos]));
    }
    if (!flag) dp[pos][sum][mod] = ans;
    return ans;
}

int main() {
    cin >> n;
    while (n) {
        a[++cnt] = n % 10;
        n /= 10;
    }
    ll res = 0;
    for (int i = 1; i <= 9 * cnt; i++) {
        m = i;
        memset(dp, -1, sizeof(dp));
        res += dfs(cnt, 0, 0, 1);
    }
    cout << res;
    return 0;
}