P4995 跳跳

题目描述

你是一只小跳蛙,你特别擅长在各种地方跳来跳去。

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

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

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

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

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


输入格式

输入一行一个正整数 n,表示石头个数。
输入第二行 n 个正整数,表示第 iii 块石头的高度 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<iostream>
#include<algorithm>
using namespace std;

const int MAXN = 300 + 5;
int rock[MAXN];                    // 定义 石头的高度 

int main()
{
    int n;                        // n 个石头 
    long long power = 0;        // 注意!!! 一定要 long long 记录体力值,因为后面是平方和,数字会很大 
    cin >> n;
    
    int l = 0, r = n;         
    for(int i = 1; i <= n; i++)
    {
        cin >> rock[i];
    }
    
    sort(rock + 1, rock + 1 + n);        // 将石头高度 由低到高 排序 
    
    rock[0] = 0;                        // 地面高度为 0  
    bool _Switch = true;                // 设置一个 开关,方便后面移动 指向,并初始化为 true 
    while(l <= r)
    {
        if(_Switch)                        //如果开关打开(true),求体力值,前指向 向后移动,并将 开关 关闭 (false) 
        {
            power = power + (rock[r] - rock[l]) * (rock[r] - rock[l]);
            ++l;
            _Switch = false;
        }
        else                            //如果开关关闭,求体力值,后指向 向前移动,并将 开关 打开 
        {
            power = power + (rock[r] - rock[l]) * (rock[r] - rock[l]);
            --r;
            _Switch = true;
        }
        
    }                                
    
    cout << power << endl;
    
    return 0;    
    
    // 计算体力值的方式为 (Hi - Hj)^2,最贪心的方式为:Hi和 Hj 相差越大,需要的体力值越高 
    // 将石头高度按 从低到高 排序,然后从 地面 跳到 最后一个石头,再跳到 第一个石头,再跳到 倒数第二个石头 
    // 如此反复(即从两边开始向中心跳),即可得到最高的体力值     
    
}

说明/提示

样例解释

原文地址:https://www.cnblogs.com/go-alltheway/p/13905329.html