2019牛客寒假算法基础集训营1 小a与‘204’ 贪心 模拟

版权声明：抱最大的希望，为最大的努力，做最坏的打算。 https://blog.csdn.net/qq_37748451/article/details/86601382

链接：https://ac.nowcoder.com/acm/contest/317/B
来源：牛客网
 

题目描述

小a非常喜欢204204这个数字，因为′a′+′k′=204′a′+′k′=204。
现在他有一个长度为nn的序列，其中只含有2,0,42,0,4这三种数字
设aiai为序列中第ii个数，你需要重新排列这个数列，使得∑ni=1(ai−ai−1)2∑i=1n(ai−ai−1)2最大(公式的含义是：每个数与前一个数差的平方的和)
注意：我们默认a0=0a0=0

输入描述:

第一行一个整数nn
接下来一行nn个整数，第ii个数表示aiai

输出描述:

输出一个整数，表示∑ni=1(ai−ai−1)2的最大值

示例1

输入

复制

2
2 4

输出

复制

20

说明

样例1解释：按(4,2)排列是最优的，此时sum=(4−0)2+(2−4)2=20

示例2

输入

复制

3
2 0 4

输出

复制

36

说明

样例2解释：按(4,0,2)排列是最优的，此时sum=(4−0)2+(0−4)2+(2−0)2=36

示例3

输入

复制

5 
2 4 0 2 4

输出

复制

52

备注:

1⩽n⩽105，保证ai为2/0/4中的数

题意：

题目复制过来的可能有点问题

就是让你重新排列序列使得给定的式子最大

思路：
贪心构造  第一个位置放最大的，第二个位置放最小的。。。。。

代码：
 

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<cmath>
#include<set>
#include<queue>
#include<stack>
#include<map>
using namespace std;
typedef long long ll;
const int N=1e5+5;
int a[N];
int b[N];
int b0,b2,b4;
int main()
{
   int n;
   cin>>n;
   ll ans=0;
   b[0]=a[0]=0;
   for(int i=1;i<=n;i++)
        cin>>a[i];
   sort(a+1,a+n+1);
   int flag=1;
   int e=n;
   int f=1;
   for(int i=1;i<=n;i++)
   {
      if(flag==1)
        {
        b[i]=a[e];
        e--;
        flag=0;
        }
      else
      {
        b[i]=a[f];
        f++;
        flag=1;
      }
      //cout<<b[i]<<" ";
   }
   //cout<<endl;
   for(int i=1;i<=n;i++)
   {
      ans+=(b[i]-b[i-1])*(b[i]-b[i-1]);
   }
   cout<<ans<<endl;
   return 0;
}