AtCoder Regular Contest 100 D:Equal Cut(前缀和优化暴力)解题报告
题目大意:依旧言简意赅。给你N个数存为Ai,切三刀,分为四组,每一组的和为b·c·d·e,要求这四个数的极差(最大数和最小数的差)最小,输出这个最小值。
题目思路:看一眼数据范围就可以不用想纯暴力了。这个题我一开始也想了很久,但我后来仔细一看,发现是切三刀,而不是分为四组,要求每一组的数在数组里都是连续的,那就变得简单多了。先预处理出前缀和,暴力中间的断点,在循环中找一前一后两个断点,如果移动能使得前后两个区域之间的差变小便移动,否则不移动,用一个变量存储所有循环中间断点时出现的最小值,输出即可。(long long!)
AC代码
//库省略
using namespace std;
const int maxn=200005;
ll sum[maxn],a[maxn];
ll n,b,c,d,e;
ll ans=1e15+7;
bool check(ll x,ll y,int i)
{
if(abs((x+a[i])-(y-a[i]))<abs(x-y))
return 1;
return 0;
}
int main()
{
cin>>n;
ll s=0;
for(int i=0;i<n;i++)
{
cin>>a[i];
s+=a[i];
sum[i]=s;
}
int j=0,k=2;
for(int i=1;i<n-2;i++)
{
b=sum[j];
c=sum[i]-sum[j];
d=sum[k]-sum[i];
e=sum[n-1]-sum[k];
while(check(b,c,j+1) && j+1<i)
{
j++;
b=sum[j];
c=sum[i]-sum[j];
}
while(check(d,e,k+1) && k+1<n-1)
{
k++;
d=sum[k]-sum[i];
e=sum[n-1]-sum[k];
}
ll test[10];
test[0]=b;test[1]=c;
test[2]=d;test[3]=e;
sort(test,test+4);
ans=min(ans,test[3]-test[0]);
}
cout<<ans<<endl;
return 0;
}