牛客寒假训练营1

 

 

B题：小a与“204”
 

题目描述

小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∑i=1n(ai−ai−1)2的最大值

示例1

输入

复制

2
2 4

输出

复制

20

说明

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

示例2

输入

复制

3
2 0 4

输出

复制

36

说明

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

示例3

输入

复制

5 
2 4 0 2 4

输出

复制

52

备注:

1⩽n⩽1051⩽n⩽105，保证aiai为2/0/42/0/4中的数

 

思路：

一开始我使用全排列进行枚举所有的可能，然后比较每种可能的总和，但是这样会Runtime Error

后来看官方题解的思路：

 

我们使用更一般的思路：

将输入的数据按字典序升序排好序，此后先放置最大的数，然后再放最小的数，再放置第二大的数，再放倒数第二小的数，

如此递推，既可以得到最大的sum

 

代码：

#include<bits/stdc++.h>
using namespace std;

int main()
{
    int n;
    int a[100000];
	
    scanf("%d",&n);
    long long sum=0;
    for(int i=1;i<=n;i++)
    {
	  scanf("%d",&a[i]);
    }
    a[0]=0;
    sort(a+1,a+n+1);    //升序排序 
    int k=a[0];
    int First=1;
    int Last=n;
    while(First<=Last)   //最大的减去最小的，第二大的减去倒数第二小的
    {
       if(First<=Last)
       {
           sum+=pow(a[Last]-k,2);
           k=a[Last];
           Last--;
       }
       if(First<=Last)
       {
           sum+=pow(a[First]-k,2);
           k=a[First];
           First++; 
       } 
    }
    printf("%lld\n",sum);
    return 0;
}