USACO 2023 FEB GOLD - Equal Sum Subarrays G

题目

FJ 给了 Bessie 一个长度为 $N(2\le N\le 500,-10^{15}\le a_i\le 10^{15})$ 的数组，且数组所有的 $\frac{N(N+1)}{2}$​ 个区间和互不相同。对于每个 $i\in [1,N]$，帮助 Bessie 把 $a_i$​ 改成一个值，使得数组有两个区间和相等，且改值与 $a_i$​ 的差值最小。

思路

老师拿来打模拟赛，赛时想了一个 $O(n^3\cdot log{n}^2)$ 的做法。

就是我们先将 $\frac{N(N+1)}{2}$ 个区间和都求出来，然后排序。

对于每次需要修改的 $a_i$，找出修改 $a_i$ 后会一起修改的区间和，打上标记。

因为区间和互不相同，那么每个区间和的值只会出现一次，所以我们可以用 map 来维护标记。（$log{n}^2$ 的来源）

因为打了标记的区间和永远不会相等，所以我们只需要考虑没打标记和打了标记之间的差值。

又因为已经排了序，则差值最小的肯定出现在排序后相邻两个标记不同的区间和上，则遍历一遍即可。

结果 TLE 了……

于是我们来考虑如何优化。

发现，map 自带 $log{n}^2$ 的时间复杂度，考虑如何消除 $log{n}^2$。（$n^3$ 能过）

我们可以记录每个区间和的左端点和右端点，则维护标记就可以用数组解决了。（具体实现可以参考代码）

然后就过了这道题！

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
int n, m, flag[505][505];
LL a[505], sum[500005], ans;
struct node { LL s; int i, j; } p[500005];
bool cmp(node a, node b) { return a.s < b.s; }
int main() {
    scanf("%d", &n);
    for (int i = 1; i <= n; i++)
        scanf("%lld", &a[i]), sum[i] = sum[i - 1] + a[i];
    for (int i = 1; i <= n; i++)
        for (int j = i; j <= n; j++)
            p[++m].s = sum[j] - sum[i - 1], p[m].i = i, p[m].j = j;
    sort(p + 1, p + 1 + m, cmp);
    for (int k = 1; k <= n; k++) {
        ans = 1e18;
        for (int i = 1; i <= k; i++)
            for (int j = k; j <= n; j++)
                flag[i][j] = 1;
        for (int i = 2; i <= m; i++)
            if (flag[p[i].i][p[i].j] != flag[p[i - 1].i][p[i - 1].j])
                ans = min(ans, abs(p[i].s - p[i - 1].s));
        for (int i = 1; i <= k; i++)
            for (int j = k; j <= n; j++)
                flag[i][j] = 0;
        printf("%lld\n", ans);
    }
    return 0;
}