題干：

小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中的數

解題報告：

? ?直接構造就可以了，，寫的不好會使得代碼丑陋233、、這樣貪心的正確性證明不太好說，，（有種情況，，比如2多出來很多，那么最后會是連續的2 2 2 2 2 2，所以沒有貢獻，會不會把2插入序列中，使得結果優于4 0 4 0 2 2 2 2 2這樣？）

AC代碼：

#include<cstdio> #include<iostream> #include<algorithm> #include<queue> #include<map> #include<vector> #include<set> #include<string> #include<cmath> #include<cstring> #define ll long long #define pb push_back #define pm make_pair #define fi first #define se second using namespace std; const int MAX = 2e5 + 5; int a[MAX],b[MAX]; int main() {int n;cin>>n;for(int i = 1; i<=n; i++) scanf("%d",a+i);sort(a+1,a+n+1);for(int i = 1,l=1,r=n; i<=n; i++) {if(i%2==1) b[i] = a[r],r--;else b[i] = a[l],l++;}ll ans = 0 ;for(int i = 1; i<=n; i++) ans += (b[i] - b[i-1])*(b[i] - b[i-1]);printf("%lld\n",ans);return 0 ;}

?

總結

以上是生活随笔為你收集整理的【牛客 - 317B】小a与204（贪心，构造，水题）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。