题目描述
你是一只小跳蛙,你特别擅长在各种地方跳来跳去。
这一天,你和朋友小 F 一起出去玩耍的时候,遇到了一堆高矮不同的石头,其中第 i 块的石头高度为 hi,地面的高度是 ℎ0=0h0=0。你估计着,从第 i 块石头跳到第 j 块石头上耗费的体力值为 (hi−hj)^2,从地面跳到第 i 块石头耗费的体力值是 (hi)^2。
为了给小 F 展现你超级跳的本领,你决定跳到每个石头上各一次,并最终停在任意一块石头上,并且小跳蛙想耗费尽可能多的体力值。
当然,你只是一只小跳蛙,你只会跳,不知道怎么跳才能让本领更充分地展现。
不过你有救啦!小 F 给你递来了一个写着 AK 的电脑,你可以使用计算机程序帮你解决这个问题,万能的计算机会告诉你怎么跳。
那就请你——会写代码的小跳蛙——写下这个程序,为你 NOIp AK 踏出坚实的一步吧!
输入格式
输入一行一个正整数 n,表示石头个数。
输入第二行 n 个正整数,表示第 i 块石头的高度 hi。
输出格式
输出一行一个正整数,表示你可以耗费的体力值的最大值。
输入输出样例
输入 #1复制
2 2 1
输出 #1复制
5
输入 #2复制
3 6 3 5
输出 #2复制
49
说明/提示
样例解释
两个样例按照输入给定的顺序依次跳上去就可以得到最优方案之一。
数据范围
对于 1≤i≤n,有 0<hi≤10^4,且保证 hi 互不相同。
对于 10%的数据,n≤3;
对于 20% 的数据,n≤10;
对于 50%的数据,n≤20;
对于 80% 的数据,n≤50;
对于 100%的数据,n≤300。
#include<bits/stdc++.h>
using namespace std;
const int N = 310;
int a[N];
long long res;
int main(){
int n;
cin >> n;
for (int i = 0; i < n; i++)
cin >> a[i];
sort(a, a + n);
int i = 0, j = n - 1;
res += a[n - 1] * a[n - 1];
for(; i < j;){
res += (a[j] - a[i]) * (a[j] - a[i]);
j--;
res += (a[j] - a[i]) * (a[j] - a[i]);
i++;
}
cout << res << endl;
return 0;
}
本题思路:贪心
每次从可选择的数中的最大值跳到最小值。
例,n = 4 : 8 , 7 , 9, 1;
将数组从小到大排序
第一步:先跳到9, 9 ^2; res += a[n - 1] * a[n - 1];
第二步:从9跳到1, 9 ^2 + (9 - 1)^2; res += (a[3] - a[0]) * (a[3] - a[0]); j--;
第三步:从1跳到8, 9 ^2 + (9 - 1)^2 + (8 - 1 ) ^2; res += (a[2] - a[0]) * (a[2] - a[0]); i++;
第三步:从8跳到7, 9 ^2 + (9 - 1)^2 + (8 - 1 ) ^2 + (8 - 7)^2;