题目链接:http://7xjob4.com1.z0.glb.clouddn.com/853ec5b32d48782501bfcf3dd089c8ce
分析:对一个数组,把它分成四等份,使得每份之间的个数相差最小(即均等分配),暴力解决:枚举各个断点。我wa了11次,主要怪在自己没看清题目,数组是无序的。
#include <iostream>
#include <string.h>
#include <cmath>
#include <algorithm>
using namespace std;
int main()
{
const double eps=1e-9;
const int maxn=10005;
int T,N,arr[maxn],tmp[165],a,b,c,cnt=0;
cin>>T;
while(T--){
cnt++;
cin>>N;
double ave=N/4.0,ans=10000000.0;
memset(tmp,0,sizeof(tmp));
for(int i=0;i<N;i++){
cin>>arr[i];
}
sort(arr,arr+N);
for(int i=0;i<N;i++)
tmp[arr[i]]=i+1;
for(int i=0;i<=160;i++){
for(int j=i+1;j<=160;j++){
for(int k=j+1;k<=160;k++){
if(tmp[i]&&tmp[j]&&tmp[k]&&tmp[i]>1&&tmp[k]<N){
double ret=abs(tmp[i]-ave)+abs(tmp[j]-tmp[i]-ave)+abs(tmp[k]-tmp[j]-ave)+abs(N-tmp[k]-ave);
if(ret<ans){
ans=ret;
a=i,b=j,c=k;
}
}
}
}
}
cout<<"Case "<<cnt<<": "<<a<<' '<<b<<' '<<c<<endl;
}
return 0;
}