贪心——洛谷P4995 跳跳！

洛谷P4995 跳跳！ - 贪心算法求解体力最大值

最新推荐文章于 2024-10-15 12:30:27 发布

原创最新推荐文章于 2024-10-15 12:30:27 发布 · 446 阅读

0 ·

CC 4.0 BY-SA版权

这是一篇关于使用贪心算法解决洛谷P4995题目的博客。题目描述了一只小跳蛙在不同高度的石头上跳跃，每次跳跃耗费的体力值与两块石头高度差的平方成正比。小跳蛙的目标是耗费尽可能多的体力值。通过输入石头的数量和高度，程序需要输出可以达到的最大体力值。解决方案是按石头高度排序，选取高度差最大的相邻石头进行跳跃。

洛谷P4996 跳跳！
$\color{blue}题目描述$
你是一只小跳蛙，你特别擅长在各种地方跳来跳去。

这一天，你和朋友小 F 一起出去玩耍的时候，遇到了一堆高矮不同的石头，其中第 $i$ 块的石头高度为 $h_i$ 你估计着，从第 $i$ 块石头跳到第 $j$ 块石头上耗费的体力值为 $h_i - h_j) ^2$ ,从地面跳到第 $i$ 块石头耗费的体力值是 $h_i) ^ 2$ 。

为了给小 F 展现你超级跳的本领，你决定跳到每个石头上各一次，并最终停在任意一块石头上，并且小跳蛙想耗费尽可能多的体力值。

当然，你只是一只小跳蛙，你只会跳，不知道怎么跳才能让本领更充分地展现。

不过你有救啦！小 F 给你递来了一个写着 AK 的电脑，你可以使用计算机程序帮你解决这个问题，万能的计算机会告诉你怎么跳。

那就请你——会写代码的小跳蛙——写下这个程序，为你 NOIp AK 踏出坚实的一步吧！

$\color{blue}输入格式$
输入一行一个正整数 $n$ ，表示石头个数。

输入第二行 $n$ 个正整数，表示第 $i$ 块石头的高度 $h_i$ 。

$\color{blue}输出格式$
输出一行一个正整数，表示你可以耗费的体力值的最大值。

$\color{blue}输入输出样例$
输入 #1

2
2 1

输出 #1

输入 #2

3
6 3 5

输出 #2

题目要你花费的体力最大而且全部都跳过，根据题目给的体力公式 $h_i-h_j)^2$ ，要这个值大，很容易想到只要 $h_i$ 和 $h_j$ 的差大就行了。那我把体力存下来，排个序，然后一大一小这样去取就行了。第一个跳的石子一定要是最大值。

#include <algorithm>
#include <cstdio>
#include <iostream>
using namespace std;
using ll = long long;
const double eps = 1e-7;
#define endl '\n'
#define pb(a) push_back(a)
#define ALL(x) x.begin(), x.end()
#define SIZE(x) int(x.size())
#define sb(x) ((x) * (x))
int h[310], hh[310];
int main() {
    int n;
    scanf("%d", &n);
    for (int i = 1; i <= n; ++i) scanf("%d", &h[i]), hh[i] = h[i];
    sort(h + 1, h + n + 1);
    sort(hh + 1, hh + n + 1, greater<int>());
    ll ans = 0;
    int p1 = 1, p2 = 0;
    int cnt = 0;
    while (hh[p1] != h[p2]) {
        if (cnt % 2 == 0) {
            ans += 1ll * sb((hh[p1] - h[p2]));
            p1++;
        } else {
            p2++;
            ans += 1ll * sb((h[p2] - hh[p1 - 1]));
        }
        cnt++;
    }
    printf("%lld\n", ans);
    return 0;
}