P10904 [蓝桥杯 2024 省 C] 挖矿

P10904 [蓝桥杯 2024 省 C] 挖矿

题目描述

小蓝正在数轴上挖矿，数轴上一共有 $n$ 个矿洞，第 $i$ 个矿洞的坐标为 $a_i$。小蓝从 $0$ 出发，每次可以向左或向右移动 $1$ 的距离，当路过一个矿洞时，就会进行挖矿作业，获得 $1$ 单位矿石，但一个矿洞不能被多次挖掘。小蓝想知道在
移动距离不超过 $m$ 的前提下，最多能获得多少单位矿石？

输入格式

输入的第一行包含两个正整数 $n,m$，用一个空格分隔。

第二行包含 $n$ 个整数 $a_1,a_2,\cdots ,a_n$，相邻整数之间使用一个空格分隔。

输出格式

输出一行包含一个整数表示答案。

输入输出样例 #1

输入 #1

5 4
0 -3 -1 1 2

输出 #1

4

说明/提示

【样例说明】

路径：$0\to -1\to 0\to 1\to 2$，可以对 { 0 , − 1 , 1 , 2 } \{0,-1,1,2\} {0,−1,1,2} 四个矿洞挖掘并获得最多 $4$ 块矿石。

【评测用例规模与约定】

对于 $20\%$ 的评测用例，$1\le n\le 10^3$；
对于所有评测用例，$1\le n\le 10^5$，$-10^6\le a_i\le 10^6$，$1\le m\le 2\times 10^6$。

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

const int N = 2e6 + 10;
int n, m;
vector<int> l(N), r(N);
int ans = 0, cnt = 0;

int main() {
    cin >> n >> m;
    for (int i = 1, x; i <= n; i++) {
        cin >> x;
        if (x < 0) {
            l[-x]++;//不管怎样，应统计成正数，同时也方便计算
        } else if (x > 0) {
            r[x]++;
        } else {
            cnt++;//x=0的情况下加一就行，勿遗漏
        }
    }
    for (int i = 1; i <= m; i++) {//前缀和
        l[i] += l[i - 1];
        r[i] += r[i - 1];
    }
    for (int i = 1; i <= m; i++) {
        int t = l[i];
        if (m - i * 2 > 0) {//这里是向左走，如果想要返回采右边的矿就需要乘二
            t += r[m - i * 2];
        }
        ans = max(ans, t);
        t = r[i];
        if (m - i * 2 > 0) {
            t += l[m - i * 2];
        }
        ans = max(ans, t);
    }
    cout << ans + cnt << endl;
    return 0;
}