题目链接:
https://www.luogu.com.cn/problem/P4995
题目:
你是一只小跳蛙,你特别擅长在各种地方跳来跳去。
这一天,你和朋友小 F 一起出去玩耍的时候,遇到了一堆高矮不同的石头,其中第 i 块的石头高度为 ,地面的高度是 =0。你估计着,从第 a 块石头跳到第 b 块石头上耗费的体力值为 ,从地面跳到第 i 块石头耗费的体力值是 。
为了给小 F 展现你超级跳的本领,你决定跳到每个石头上各一次,并最终停在任意一块石头上,并且小跳蛙想耗费尽可能多的体力值。
当然,你只是一只小跳蛙,你只会跳,不知道怎么跳才能让本领更充分地展现。
不过你有救啦!小 F 给你递来了一个写着 AK 的电脑,你可以使用计算机程序帮你解决这个问题,万能的计算机会告诉你怎么跳。
那就请你——会写代码的小跳蛙——写下这个程序,为你 NOIp AK 踏出坚实的一步吧!
输入格式:
输入一行一个正整数 n,表示石头个数。
输入第二行 n 个正整数,表示第 i 块石头的高度 。
输出格式:
输出一行一个正整数,表示你可以耗费的体力值的最大值。
例:
#1:
输入:
2 2 1
输出:
5
#2:
输入:
3 6 3 5
输出:
49
样例解释:
两个样例按照输入给定的顺序依次跳上去就可以得到最优方案之一。
提示:
对于所有的数据保证n300,0<。
思路:
这是一题典型的贪心算法题。
接下来是贪心证明:每一块石头都只能被跳一次并且从平地开始起跳,假设当前处于高度为的石头上,若我们想在下一次跳花费更多的体力,则需要下一次跳到最高或最低的石头上。因为我们从平地起跳且没有比平地高度还要低的石头,所以跳到最高的石头上,即可获得这一跳所花费的最大体力值:在下一跳前,因为当前处于最高的石头上,所以选择最低的石头即可。至此我们得出了贪心证明。
首先我们用整形n来存储石头的数量,使用for循环循环用数组arr来存储石头的高度。
接着对数组arr进行从低到高进行升序排序。
我们定义俩指针l和r分别指向平地(因为高度最低)和最高的石头即可。我们使用flag来表示状态,当flag=1时,表示目前处于高度低的地方,需要往高度高的石头上跳;当flag=0时,表示目前处于高度高的地方,需要往高度低的石头上跳。
接着使用while循环,当l<r时循环,我们根据flag的状态,来判断跳的方向,当flag=1时,计算出该跳所需体力并且加到res上,然后让l加一;当flag=0时,同样计算出该跳所需的体力并且加到res上,然后让r减一。
最后,我们输出res即可。
AC代码:
#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
int n, h[305] = { 0 };
long long res = 0;
bool Cmp(int x, int y) {
return x < y;
}
int main() {
scanf("%d", &n);
for (int i = 1; i <= n; i++) {
scanf("%d", &h[i]);
}
sort(h, h + n + 1, Cmp);
int beg = 0, ed = n, flag = 1;
//flag=1时为低往高跳,flag=0时为高往低跳
while (beg < ed) {
if (flag) {
res += pow(h[ed] - h[beg], 2);
beg++;
flag = 0;
}
else {
res += pow(h[ed] - h[beg], 2);
ed--;
flag = 1;
}
}
printf("%lld", res);
return 0;
}