牛客-小a与“204”（思维）

链接：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∑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中的数

将 0， 2 ，4  计数，根据多少确定不同的情况计算，思维

AC代码：

#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<string>
#include<vector>
#include<queue>
#include<deque>
#include<stack>
#include<cmath>
#include<list>
#include<map>
#include<set>
using namespace std;
typedef long long ll;
const double PI = acos(-1.0);
const double EXP = 1E-8;
const int INF = 0x3f3f3f3f;
int a[100010];
int main()
{
    int n;
    int sum_0=0,sum_2=0,sum_4=0;
    int ans=0;
    cin>>n;
    for(int i=1; i<=n; i++)
    {
        scanf("%d",&a[i]);
        if(a[i]==0)sum_0++;
        if(a[i]==2)sum_2++;
        if(a[i]==4)sum_4++;
    }
    int temp=min(sum_4,sum_0);
    int x=abs(sum_0-sum_4);
    if(sum_0==sum_4)
    {
        if(sum_2==0)
            ans=sum_4*32;
        else
            ans=sum_4*32+4;
    }
    else
    {
        if(sum_4>sum_0)
        {
            if(sum_2>x-1)
                ans=(2*sum_0+1)*16+4*(2*(x-1)+1);
            else
                ans=(2*sum_0+1)*16+8*sum_2;
        }
        else
        {
            if(sum_2>x)
                ans=sum_4*32+4*(2*x+1);
            else
                ans=sum_4*32+8*sum_2;
        }
    }
    cout<<ans<<endl;
    return 0;
}