AcWing[P265]营业额统计

最新推荐文章于 2024-07-25 11:27:31 发布

Albatross_LC

最新推荐文章于 2024-07-25 11:27:31 发布

阅读量21

点赞数

分类专栏： AcWing 文章标签：算法

本文链接：https://blog.csdn.net/weixin_44021857/article/details/132392678

版权

AcWing 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

首先看题面，给出一个数 $n$ 及 $n$ 个数 $a_i(1≤i≤n)$ ,求

$\sum_{i=1}^{n} min_{(1≤j＜i)}\left | a_i - a_j \right | $

可以直接暴力，每次从 $1$ 到 $i$ 遍历，遍历 $n$ 遍，时间复杂度为 $O(n^2)$

但这里 $n$ 的数据范围为 $1 \leq n \leq 32767$ , $32767^2 = 1073676289$ ，显然会超时

所以我们就要用一种能查询 $a_j - a_i$ 的最小值最方便的数据结构，这显然是平衡树~~（当然是看算法标签看出来的）~~

手写平衡树"比较"麻烦，但C++中的STL数据模板库中有为我们封装好的平衡树（用 $v ec t or$ 存）

设当前数组中输入到的数为 $x$ ， $v ec t or$ 数组命名为a则a.insert(upper_bound(a.begin(),a.end(),x),x);表示为将 $x$ 存入 $a$ 平衡树；*upper_bound(a.begin(),a.end(),x)表示为在 $a$ 平衡树中小于 $x$ 的最大值；*--lower_bound(a.begin(),a.end(),x)表示为在 $a$ 平衡树中大于 $x$ 的最小值，因此, $min_{(1≤j＜i)}\left | a_i - a_j \right |$ 就可以表示为min(abs((*upper_bound(a.begin(),a.end(),x)) - x), abs(x - (*--lower_bound(a.begin(),a.end(),x))))

这样，每次输入 $x$ ，并将 $x$ 装入 $a$ 平衡树，并将 $s u m$ 每次加上min(abs((*upper_bound(a.begin(),a.end(),x)) - x), abs(x - (*--lower_bound(a.begin(),a.end(),x))))， $s u m$ 即为所求

~~但WA了8次的经验告诉我，这是错的~~

以输入样例为例，刚才的方法输出答案为 $15$ ，但标准答案为 $12$ ，这是因为 $5$ 出现了两次

注意，问题出来了，*upper_bound(a.begin(),a.end(),x)表示为在 $a$ 平衡树中小于 $x$ 的最大值；*--lower_bound(a.begin(),a.end(),x)表示为在 $a$ 平衡树中大于 $x$ 的最小值，但是，当在 $a$ 中有数字和 $x$ 数值相同呢？

题目中要求的是$\sum_{i=1}^{n} min_{(1≤j＜i)}\left | a_i - a_j \right | $，故当$ a_i == a_j $时，$ min$为 $0$ ，此时 $min$ 为最小，但刚才的步骤把它略过了，所以可以再定义两个数组，将输入过的数字记上，若当前输入数字已被操作过的话，则不对其进行操作

注意：这里一定要设两个数组，因为数据范围中$ |a_i | ≤ 10^6 $，所以$ a_i$可能为负，两个数组一个记录正数，一个记录负数（当然也可以合并在一起）

但，样例还是没过

qwq

我们把 $n$ 次中每次的 $s u m$ 输出出来就可以发现：出问题了！第二个 $s u m$ 为 $6$ ！

再把 $min$ 中的两个数据输出出来，为 $0$ $5$

So，很容易就可以发现出，当 $i == 2$ 时没有比 $1$ 更小的数，故返回为 $0$ ，此时 $min (1, 4) == 1$ ，但 $min$ 左边的数据是不存在的，就需要存一个最大最小值，若 $x$ 为最大（小）值，则返回 $min$ 中左（右）边的数据，最后输出 $s u m$ 即可，这样我们就可以A了这道题

So~~（这道题坑点真多）~~

这里是代码↓↓↓：

#include <bits/stdc++.h>
using namespace std;
const int N = 1000010;
vector<int> a;
int n, x, sum, mi = INT_MAX, mx = INT_MIN;
bool z[N], f[N];
signed main() {
    scanf("%d", &n);
    memset(z, 0, sizeof z);
    memset(f, 0, sizeof f);
    for (int i = 1; i <= n; i ++ ) {
        scanf("%d", &x);
        mi = min(mi, x), mx = max(mx, x);
        a.insert(upper_bound(a.begin(),a.end(),x),x);
        int zx = abs((*upper_bound(a.begin(),a.end(),x)) - x);
        int zd = abs(x - (*--lower_bound(a.begin(),a.end(),x)));
        if ((x >= 0 && z[x] == 0) || (x < 0 && f[abs(x)] == 0)){
            if (x == mi) sum += zx;
            else if (x == mx) sum += zd;
            else sum += min(zx, zd);
        }
        if (x >= 0) z[x] = true;
        else f[abs(x)] = true;
    }
    printf("%d\n", sum);
}