小a与“204”------数列、排序

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

小a非常喜欢$204$这个数字，因为${}^{}{a}^{\prime }{+}^{\prime }{k}^{\prime }=204$。
现在他有一个长度为$n$的序列，其中只含有$2,0,4$这三种数字
设$a_i$为序列中第$i$个数，你需要重新排列这个数列，使得${\sum }_{i=1}^n(a_i-a_{i-1}{)}^2$最大(公式的含义是：每个数与前一个数差的平方的和)
注意：我们默认$a_0=0$

输入描述:

第一行一个整数$n$
接下来一行$n$个整数，第$i$个数表示$a_i$

输出描述:

输出一个整数，表示${\sum }_{i=1}^n(a_i-a_{i-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⩽{10}^5$，保证$a_i$为$2/0/4$中的数

注：依次把第一大和第二大的数存入数组，依次类推，在按公式计算。

#include<iostream>
#include<algorithm>
using namespace std;
bool cmp(int a ,int b)
{
  return a>b;
}
int a[1000000],b[1000000];
int main()
{
  int n,m=0;
  scanf("%d",&n);
  for(int i=0;i<n;i++)
  {
    cin>>a[i];
  }
  sort(a,a+n,cmp);
  int j=0;
  b[0]=0;
  for(int i=0;i<n;i++)
  {
    if(i%2==0)
      {
        b[i+1]=a[j];
        j++;
      }
    else
      b[i+1]=a[n-j];
  }
  for(int i=1;i<=n;i++)
  {
    //cout<<b[i]<<" ";
    m=m+(b[i]-b[i-1])*(b[i]-b[i-1]);
  }
  printf("%d\n", m);

}

 

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

小a非常喜欢$204$这个数字，因为${}^{}{a}^{\prime }{+}^{\prime }{k}^{\prime }=204$。
现在他有一个长度为$n$的序列，其中只含有$2,0,4$这三种数字
设$a_i$为序列中第$i$个数，你需要重新排列这个数列，使得${\sum }_{i=1}^n(a_i-a_{i-1}{)}^2$最大(公式的含义是：每个数与前一个数差的平方的和)
注意：我们默认$a_0=0$

输入描述:

第一行一个整数$n$
接下来一行$n$个整数，第$i$个数表示$a_i$

输出描述:

输出一个整数，表示${\sum }_{i=1}^n(a_i-a_{i-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⩽{10}^5$，保证$a_i$为$2/0/4$中的数