D. Dog Walking（枚举，贪心）_dog walking算法-优快云博客

D. Dog Walking

题意

你一开始在 $0$ ，给你一个数组 $a$ 和 $k$ ， $a_i$ 表示你从所在地往左或右走多少，若 $a_i=0$ 你可以指定这个位置是 $[- k, k]$ 中的一个数，问你最后走回 $0$ 的条件下最多走多少个不同的点。

思路

枚举，贪心

是否成立在于我们是否能用有的 $0$ 将最终位置挪回到 $0$ 点，对于第 $i$ 个 $0$ 影响的只有 $i$ 后面的点。

最多走多少个不同的点等价于求通过最优操作后整个行进路线最左端 $L$ 和最右端 $R$ 的距离加 $1$ 。但是由于 $0$ 的位置不确定我们很难直接确定怎么操作是最优的，最终是哪两个点是 $L$ 和 $R$ ，因此考虑暴力枚举任意两个点，将这两个点之间的长度最长化，由于最终答案一定是最优化后的两个点之间的长度，因此我枚举的答案一定能包含我们的最优解。

对于每个 $0$ 我们可以加 $[- k, k]$ ，对于我们最长化这个两点之间区间是不方便的，因此考虑将所有的 $0$ 变成 $- k$ ，这样等价于我们可以给每个 $0$ 加 $[0,2\times k]$ 之间的数了，这样就是单调的了，方便讨论，设 $s_i:是a_i的前缀和即当前在哪个位置$ ， $cnt_i:表前i个数里有多少个1$ ，则对于任意一个点对 $(i, j)$ 分情况来看：

$s_i \le s_j$ ，由于想让长度尽量长，因此希望两点之间的 $0$ 尽量变成大的数，假设最多可以移动 $v=(cnt_j-cnt_{i-1})\times2 k$ ，由于转化以后的问题所有操作只能为正的，因此最终 $s_n<0$ ，所以我们最多移动 $R=-s_n$ ，多了就没法移回来了，最少移动 $L=-s_n-(cnt_j-cnt_{i-1})\times 2k$ ，表示除这个区间外是否能将 $s_n$ 抵消，不行就要这个区间内抵消一部分，则我们的最优长度就是 $s_j-s_{i-1}+1+max(R,min(L,v))$
$s_i\ge s_j$ ，表示在图上就是 $j < - i$ ，如果我们在区间内进行操作就会让 $j$ 去向右迫近 $i$ 因此我们要尽量少的进行操作，因此最优长度就是 $s_j-s_{i-1}+1-min(L,0)$

暴力枚举每一个点对，取最大即可

Code

#include <bits/stdc++.h>
#define x first
#define y second
#define debug(x) cout<<#x<<":"<<x<<endl;
using namespace std;
typedef long double ld;
typedef long long LL;
typedef pair<int, int> PII;
typedef pair<double, double> PDD;
typedef unsigned long long ULL;
const int N = 1e5 + 10, M = 2 * N, INF = 0x3f3f3f3f, mod = 1e9 + 7;
const double eps = 1e-8, pi = acos(-1), inf = 1e20;
int dx[] = {-1, 0, 1, 0}, dy[] = {0, 1, 0, -1};
int h[N], e[M], ne[M], w[M], idx;
void add(int a, int b, int v = 0) {
    e[idx] = b, w[idx] = v, ne[idx] = h[a], h[a] = idx ++;
}
LL n, m, k;
LL a[N];
int cnt[N];
int main() {
    ios::sync_with_stdio(false), cin.tie(0);
    cin >> n >> k;
   
    for (int i = 1; i <= n; i ++) {
        cin >> a[i];
        if (!a[i]) a[i] = -k, cnt[i] = 1;
        a[i] += a[i - 1];
        cnt[i] += cnt[i - 1];
    }

    k *= 2;
    if (a[n] > 0 || -a[n] > (LL)cnt[n] * k) {
        cout << -1 << '\n';
        return 0;
    }
   
    LL res = 0;
    for (int i = 0; i <= n; i ++)
        for (int j = i; j <= n; j ++) {
            LL l = -a[n] - (cnt[n] - (cnt[j] - cnt[i])) * k,
               r = -a[n]; 
            LL v = (cnt[j] - cnt[i]) * k;
            res = max(res, a[j] - a[i] + 1 + min(r, max(l, v)));
            res = max(res, a[i] - a[j] + 1 - max(0ll, l));
        }
    
    cout << res << '\n';
    return 0;
}