算法竞赛石子合并

题目链接：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