AtCoder Beginner Contest 366 A~F

A.Election 2（思维）

题意：

$AtCoder$ 市正在举行市长选举。候选人是高桥和青木。
有 $N$ 张有效选票投给两位候选人中的任何一位，目前正在进行计票。这里， $N$ 是奇数。
目前的计票结果是：高桥 $T$ 票，青木 $A$ 票。
请判断此时选举结果是否已经确定。

分析：

我们计算票数少的加上剩下的所有票能不能超过票数多的即可。

代码：

#include <bits/stdc++.h>

using namespace std;

int main() {
    int a, b, c;
    cin >> a >> b >> c;
    if (min(b, c) + a - b - c > max(b, c))
        cout << "No" << endl;
    else
        cout << "Yes" << endl;
    return 0;
}

B. Vertical Writing（模拟）

题意：

给你一个横向书写的文本。将其转换为竖写，用 * 填充空格。

给你 $N$ 个由小写英文字母组成的字符串 $S_1,S_2,\dots ,S_N$ 。设 $M$ 为这些字符串的最大长度。

打印 $M$ 个满足以下条件的字符串 $T_1,T_2,\dots ,T_M$ ：

• 每个 $T_i$ 由小写英文字母和 * 组成。
• 每个 $T_i$ 不以 * 结尾。
• 每个 $1\le i\le N$ 都满足以下条件：
  • 对于每个 $1\le j\le |S_i|$ ， $T_j$ 的 $(N-i+1)$ 个字符存在， $T_1,T_2,\dots ,T_{|S_i|}$ 的 $(N-i+1)$ 个字符按此顺序连接等于 $S_i$ 。
  • 对于每一个 $|S_i|+1\le j\le M$ ， $T_j$ 的 $(N-i+1)$ 个字符要么不存在，要么是 *。

这里， $|S_i|$ 表示字符串 $S_i$ 的长度。

分析：

我们将$n$个字符串顺时针旋转$90$度，然后对于每一行，从右往左，一旦碰到字符，后续再碰到 时，替换成*即可。

代码：

#include <bits/stdc++.h>

using namespace std;
char tmp[105][105];

int main() {
    int n, m = -1;
    cin >> n;
    for (int i = 0; i < n; i++) {
        string s;
        int l;
        cin >> s;
        l = s.length();
        m = max(m, l);
        for (int j = 0; j < l; j++)
            tmp[i][j] = s[j];
    }
    for (int i = 0; i < m; i++) {
        string s = "";
        for (int j = n - 1; j >= 0; j--) {
            if (tmp[j][i])
                s += tmp[j][i];
            else
                s += '*';
        }
        while (s[s.length() - 1] == '*')
            s = s.substr(0, s.length() - 1);
        cout << s << endl;
    }
    return 0;
}

C.Balls and Bag Query（数据结构）

题意：

你有一个空袋子。给你 $Q$ 个查询，必须按顺序处理。

有三种查询。

• 1 x ：将一个写有整数 $x$ 的球放入袋子中。
• 2 x : 从袋子中取出一个写有整数 $x$ 的球并丢弃。当给出这个查询时，可以保证袋子中有一个写着整数 $x$ 的球。
• 3 : 打印袋中写有不同整数的球的个数。

分析：

用$map$维护一下各个数字球的个数，当$map[x]=0$时移除该元素，询问就是$map.size()$。

代码：

#include <bits/stdc++.h>

using namespace std;

int main() {
    int q;
    cin >> q;
    map<int, int> tmp;
    while (q--) {
        int op;
        cin >> op;
        if (op == 1) {
            int x;
            cin >> x;
            tmp[x]++;
        } else if (op == 2) {
            int x;
            cin >> x;
            tmp[x]--;
            if (tmp[x] == 0)
                tmp.erase(x);
        } else {
            cout << tmp.size() << endl;
        }
    }
    return 0;
}

D.Cuboid Sum Query （思维）

题意：

给你一个正整数 $N$ 和一个整数 $A_{x,y,z}$ ，每个整数 $(x,y,z)$ 的三元组都是 $1\le x,y,z\le N$ 。

你将得到以下格式的 $Q$ 个查询，必须按顺序处理。

对于 $i$ 查询 $(1\le i\le Q)$ ，我们会给你一个由整数 $(L{x}_i,R{x}_i,L{y}_i,R{y}_i,L{z}_i,R{z}_i)$ 组成的元组 $1\le L{x}_i\le R{x}_i\le N$ 、 $1\le L{y}_i\le R{y}_i\le N$ 和 $1\le L{z}_i\le R{z}_i\le N$ 。查找：

$\sum _{x=L{x}_i}^{R{x}_i}\sum _{y=L{y}_i}^{R{y}_i}\sum _{z=L{z}_i}^{R{z}_i}{A}_{x,y,z}$ .

分析：

我们令$S_{i,j,k}=\sum _{x=1}^{i-1}\sum _{y=1}^{j-1}\sum _{z=1}^{k-1}{A}_{x,y,z}$ 为 $1\le i,j,k\le N+1$ 。（如果 $i$ ， $j$ 或 $k$ 为 $0$ ，我们定义为 $S_{i,k,k}=0$ ）。

那么 $\sum _{x=L{x}_i}^{R{x}_i}\sum _{y=L{y}_i}^{R{y}_i}\sum _{z=L{z}_i}^{R{z}_i}{A}_{x,y,z}=S_{R{x}_i+1,R{y}_i+1,R{z}_i+1}-S_{L{x}_i,R{x}_i+1,R{z}_i+1}-S_{R{x}_i+1,L{y}_i,R{z}_i+1}-S_{R{x}_i+1,R{y}_i+1,L{z}_i}+S_{L{x}_i,L{y}_i,R{z}_i+1}+S_{L{x}_i,R{y}_i+1,L{z}_i}+S_{R{x}_i+1,L{y}_i,L{z}_i}-S_{L{x}_i,L{y}_i,L{z}_i}$ 成立。

所以，如果我们能快速计算 $S$ ，那么每个查询都能在恒定的时间内得到回答。

事实上，根据 $S_{i,j,k}=S_{i-1,j,k}+S_{i,j-1,k}+S_{i,j,k-1}-S_{i-1,j-1,k}-S_{i-1,j,k-1}-S_{i,j-1,k-1}+S_{i-1,j-1,k-1}+A_{i-1,j-1,k-1}$ 的性质， $S$ 可以快速计算。

代码：

#include <bits/stdc++.h>

using namespace std;
typedef long long LL;
LL a[105][105][105];
LL sum[105][105][105];

int main() {
    int n;
    cin >> n;
    for (int i = 0; i < n; ++i) {
        for (int j = 0; j < n; ++j) {
            for (int k = 0; k < n; ++k) {
                cin >> a[i][j][k];
            }
        }
    }
    for (int i = 0; i < n; ++i) {
        for (int j = 0; j < n; ++j) {
            for (int k = 0; k < n; ++k) {
                sum[i + 1][j + 1][k + 1] =
                        sum[i][j + 1][k + 1] + sum[i + 1][j][k + 1] +
                        sum[i + 1][j + 1][k] - sum[i][j][k + 1] - sum[i][j + 1][k] -
                        sum[i + 1][j][k] + sum[i][j][k] + a[i][j][k];
            }
        }
    }
    int q;
    cin >> q;
    for (int i = 0; i < q; ++i) {
        int lx, rx, ly, ry, lz, rz;
        cin >> lx >> rx >> ly >> ry >> lz >> rz;
        lx--, ly--, lz--;
        LL ans = sum[rx][ry][rz] - sum[lx][ry][rz] - sum[rx][ly][rz] -
                 sum[rx][ry][lz] + sum[lx][ly][rz] + sum[lx][ry][lz] +
                 sum[rx][ly][lz] - sum[lx][ly][lz];
        cout << ans << endl;
    }
    return 0;
}

E.Manhattan Multifocal Ellipse （双指针）

题意：

给你一个二维平面上的 $N$ 点 $(x_1,y_1),(x_2,y_2),\dots ,(x_N,y_N)$ 和一个非负整数 $D$ 。

求 $\sum _{i=1}^N(|x-x_i|+|y-y_i|)\le D$ 的整数对 $(x,y)$ 的个数。

分析：

观察$x_i,y_i$的范围，可知$x,y$的取值范围是$[-2\times 10^6,2\times 10^6]$ 。将$x,y$分开来计算，枚举所有的$x$，计算$ (|x-x \times i|) $的和（可以通过正反方向预处理前缀和来计算），对 $y$ 同理。然后对于所有的 $x,y$，从两端开始双指针枚举和$\le D$的情况。

代码：

#include <bits/stdc++.h>

using namespace std;
typedef long long LL;
const int N = 3e5 + 10;
LL x[N], y[N], a[N], b[N], t[1000005], ans;

int main() {
    int n, d;
    cin >> n >> d;
    for (int i = 1; i <= n; i++) {
        cin >> x[i] >> y[i];
    }
    sort(x + 1, x + n + 1);
    sort(y + 1, y + n + 1);
    x[n + 1] = LONG_LONG_MAX;
    y[n + 1] = LONG_LONG_MAX;
    for (int i = 1; i <= n; i++) {
        a[i] = a[i - 1] + x[i];
        b[i] = b[i - 1] + y[i];
    }
    for (int i = -2e6; i <= 2e6; i++) {
        LL tmp = upper_bound(x + 1, x + n + 2, i) - x;
        LL id = i * (tmp - 1) - a[tmp - 1] + a[n] - a[tmp - 1] - i * (n - tmp + 1);
        if (id <= d) {
            t[id]++;
        }
    }
    for (int i = 1; i <= d; i++) {
        t[i] += t[i - 1];
    }
    for (int i = -2e6; i <= 2e6; i++) {
        LL tmp = upper_bound(y + 1, y + n + 2, i) - y;
        LL id = i * (tmp - 1) - b[tmp - 1] + b[n] - b[tmp - 1] - i * (n - tmp + 1);
        if (id <= d) {
            ans += t[d - id];
        }
    }
    cout << ans << endl;
    return 0;
}

F.Maximum Composition （dp）

题意：

给你 $N$ 个线性函数 $f_1,f_2,\dots ,f_N$ ，其中 $f_i(x)=A_{i}x+B_i$ .

求由 $K$ 组成的序列 $p=(p_1,p_2,\dots ,p_K)$ 中 $f_{p_1}(f_{p_2}(\dots f_{p_K}(1)\dots ))$ 的最大可能值。介于 $1$ 与 $N$ （含）之间的个不同整数的最大可能值 $f_{p_1}(f_{p_2}(\dots f_{p_K}(1)\dots ))$ 。

分析：

我们首先思考当用于序列 $p$ 的 $K$ 整数固定不变时， $p$ 具有什么样的性质？
可以发现：
$f_i(f_j(x))$ 和 $f_j(f_i(x))$ 的排序与 $x$ 无关，完全取决于 $\frac{A_i-1}{B_i}$ 和 $\frac{A_j-1}{B_j}$ 的排序。如果是 $\frac{A_i-1}{B_i}$$>$ $\frac{A_j-1}{B_j}$ ，那么就是 $f_i(f_j(x))>f_j(f_i(x))$ 。

根据这一性质，在给定 $K$ 个整数用于 $p$ 时，当 $p$ 满足 $\frac{A_{p_1}-1}{B_{p_1}}\ge \frac{A_{p_2}-1}{B_{p_2}}\ge \dots \ge \frac{A_{p_K}-1}{B_{p_K}}$ 时，值 $f_{p_1}(f_{p_2}(\dots f_{p_K}(1)\dots ))$ 取最大值。

为简单起见，假设对 $(A,B)$ 进行排序，使 $\frac{A_1-1}{B_1}\ge \frac{A_2-1}{B_2}\ge \dots \frac{A_N-1}{B_N}$ .

那么问题可以改写如下：

求介于 $1$ 和 $N$ 之间的整数序列 $p$ 的最大值 $f_{p_1}(f_{p_2}(\dots f_{p_K}(1)\dots ))$ ，使得 $p_1<p_2<\dots <p_N$。
我们可以用$dp$来解决这问题。
当决定 $p_K,p_{K-1},\dots p_{K-i}$ 时，求 $f_{p_{K-i}}(\dots f_{p_K}(1)\dots )$ 的最大值 $dp_{i,j}=( $从而求得 $p_{K-i}\ge j)$ .
我们定义$d{p}_{i,j}=\text{max}(d{p}_{i,j+1},A_j\times d{p}_{i-1,j+1}+B_j)$ ，所求答案为 $d{p}_{K,1}$ 。

代码：

#include <bits/stdc++.h>

using namespace std;

int main() {
    int n, k;
    cin >> n >> k;
    vector<int> a(n), b(n);
    for (int i = 0; i < n; ++i) {
        cin >> a[i] >> b[i];
    }
    vector<int> num(n);
    for (int i = 0; i < n; ++i)
        num[i] = i;
    sort(num.begin(), num.end(), [&](int i, int j) { return b[i] * (a[j] - 1) > b[j] * (a[i] - 1); });
    vector<long long> dp(k + 1, -1e9);
    dp[0] = 1;
    for (auto i: num) {
        vector<long long> tmp = dp;
        for (int j = 0; j < k; ++j)
            if (dp[j] != -1e9) {
                tmp[j + 1] = max(tmp[j + 1], dp[j] * a[i] + b[i]);
            }
        dp = move(tmp);
    }
    cout << dp[k] << endl;
    return 0;
}

赛后交流

在比赛结束后，会在交流群中给出比赛题解，同学们可以在赛后查看题解进行补题。

群号： 704572101，赛后大家可以一起交流做题思路，分享做题技巧，欢迎大家的加入。