题目链接:https://ac.nowcoder.com/acm/contest/9667/F
题目描述
牛牛有 n 堆石子, 每堆石子有 a[i] 个, 牛牛每次可以选择相邻的两堆石子,然后拿走少的那一堆,得到的价值是两堆石子个数之和, 直到只剩下一堆石子。
如果拿走了第 i 堆石子, 那么第 i-1 堆和第 i+1 堆 就会相邻。
牛牛想知道该怎么拿,才能使得到的价值最多。
输入描述:
第一行一个整数 n, 1 ≤ n ≤ 2e5
第二行 n 个整数 a[i],0 ≤ a [i] ≤ 1e9
输出描述:
输出得到的最大价值
示例1
输入
5
2 5 3 5 1
输出
31
分析
贪心算法,每次找价值最大的石子堆与相邻的石子堆合并,这样最大价值就是max*(n-1) + sum - max (max代表最大值,sum代表石子堆价值总和)
AC代码
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
ll a[200005];
int main()
{
int n;
cin>>n;
ll mx=0;
unsigned long long sum=0;
for(int i=1;i<=n;i++)
{
cin>>a[i];
sum+=a[i];
if(a[i]>mx)
mx=a[i];
}
if(n==1) printf("%d\n",a[1]);//只有一堆石子,单独判断
else
{
sum+=(n-2)*mx;
cout<<sum<<endl;
}
return 0;
}
注意
两个int类型变量相乘,如果乘积超出int范围,要开long long,在本题中mx和n都在int范围内,但乘积可能会超出int,所以mx开long long