洛谷 [USACO23FEB] Equal Sum Subarrays G
题目大意
给你一个长度为 n n n的数组 a a a,且数组 a a a的所有区间和互不相同。对于每一个 i ∈ [ 1 , n ] i\in [1,n] i∈[1,n],请将 a i a_i ai改成一个值,使得数组 a a a中有两个区间的和相等,且要使修改值和原来的 a i a_i ai的差最小,并输出这个差值。
2 ≤ n ≤ 500 , − 1 0 15 ≤ a i ≤ 1 0 15 2\leq n\leq 500,-10^{15}\leq a_i\leq 10^{15} 2≤n≤500,−1015≤ai≤1015
题解
首先,将所有区间放在一个结构体 w w w中。结构体维护区间的三个值:区间的左端点、区间的右端点、区间和。
将 w w w按区间和从小到大排序。对于每一个 i i i,将包含 i i i的区间标记一下。我们知道,如果要使 a i a_i ai修改后有两个区间的区间和相同,那么这两个区间一定是一个包含 i i i、另一个不包含 i i i。那么对于 i i i的答案就是打标记的区间和没打标记的区间的差的最小值。
因为这些区间是已经排好序了,所以打标记的区间和没打标记的区间在 w w w上一定是相邻的。那么,我们可以枚举一遍 w w w,对于每两个相邻的区间,如果其中一个有标记且另一个没有,就用这两个区间的区间和的差来更新答案即可。
总时间复杂度为 O ( n 3 ) O(n^3) O(n3)。
code
#include<bits/stdc++.h>
using namespace std;
int n,w1=0,z[500005];
long long ans,a[505],sum[505];
struct node{
int i,j;
long long sum;
}w[500005];
bool cmp(node ax,node bx){
return ax.sum<bx.sum;
}
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++){
w[++w1]=(node){i,j,sum[j]-sum[i-1]};
}
}
sort(w+1,w+w1+1,cmp);
for(int i=1;i<=n;i++){
ans=1e18;
for(int j=1;j<=w1;j++){
if(w[j].i<=i&&w[j].j>=i) z[j]=1;
}
for(int j=1;j<=w1-1;j++){
if(z[j]^z[j-1]) ans=min(ans,abs(w[j].sum-w[j-1].sum));
}
for(int j=1;j<=w1;j++){
if(w[j].i<=i&&w[j].j>=i) z[j]=0;
}
printf("%lld\n",ans);
}
return 0;
}