P4995 跳跳！(贪心)

题目描述

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

这一天，你和朋友小 F 一起出去玩耍的时候，遇到了一堆高矮不同的石头，其中第 �i 块的石头高度为 ℎ�hi​，地面的高度是 ℎ0=0h0​=0。你估计着，从第 �i 块石头跳到第 �j 块石头上耗费的体力值为 (ℎ�−ℎ�)2(hi​−hj​)2，从地面跳到第 �i 块石头耗费的体力值是 (ℎ�)2(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≤�≤�1≤i≤n，有 0<ℎ�≤1040<hi​≤104，且保证 ℎ�hi​ 互不相同。

对于 10%10% 的数据，�≤3n≤3；

对于 20%20% 的数据，�≤10n≤10；

对于 50%50% 的数据，�≤20n≤20；

对于 80%80% 的数据，�≤50n≤50；

对于 100%100% 的数据，�≤300n≤300。

因为要消耗力气最多，所以每次都要挑高度相差最大的两块石头跳，按这个思路模拟。

#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<string>
#include<vector>
#include<math.h>
#include<iomanip>
#include<set>
#include<queue>
#include<stack> 
#include<map>
using namespace std;
long long n, ans = 0;//石头数量，答案
struct MyStruct
{
	long long a;//石头高度
	int s;//记录石头有没有跳过
}pp[1000000];
bool cmp(MyStruct x, MyStruct y)
{
	return x.a > y.a;
}
int main()
{
	cin >> n;
	for (int i = 0; i < n; i++)
	{
		cin >> pp[i].a;
		pp[i].s = 1;//没跳过为1，跳过为0
	}
	sort(pp, pp + n, cmp);//高度从大到小排序，保证第一个跳最高得石头
	ans = ans + pow(pp[0].a, 2);//处理第一块石头
	pp[0].s = 0;
	long long k = 0;//除了从地面第一次跳，用来记录跳得次数
	long long f = 0;//记录当前停留在那一块石头上
	while (k != n - 1)
	{
		long long ma = -1;//因为贪心找差值最大
		long long ff = f;//用以一个新标记记录下一个该跳的石头
		for (int i = 0; i < n; i++)
		{
			if (pp[i].s == 1 && pp[ff].s == 0)//如果石头没跳过
			{
				if (abs(pp[i].a - pp[ff].a) > ma)//更新ma
				{
					ma = abs(pp[i].a - pp[ff].a);
					f = i;
				}
			}
		}
		ans = ans + pow(ma, 2);
		k++;
		pp[f].s = 0;//当前石头跳过，状态为0
	}
	cout << ans;
}