BZOJ3229 石子合并

Description
　　在一个操场上摆放着一排N堆石子。现要将石子有次序地合并成一堆。规定每次只能选相邻的2堆石子合并成新的一堆，并将新的一堆石子数记为该次合并的得分。
　　试设计一个算法，计算出将N堆石子合并成一堆的最小得分。

Input
　　第一行是一个数N。
　　以下N行每行一个数A，表示石子数目。

Output
　　共一个数，即N堆石子合并成一堆的最小得分。

Sample Input
4
1
1
1
1

Sample Output
8

HINT

对于 100% 的数据，1≤N≤40000

对于 100% 的数据，1≤A≤200

题解： 一般的石子合并在dp+四边形优化后复杂度为O($n^2$)，这题数据范围显然不行。
所以此题正解为GarsiaWachs算法
下面简述一下该算法的流程以及结论：

（为了方便起见，我们规定a[0] = a[n+1] = INF)

①从序列的左端开始找第一个满足 a[k−1] ≤ a[k+1] 的 k，然后合并 a[k−1] 和 a[k]
②从当前合并位置向左寻找第一个满足 a[j] > a[k-1] + a[k] 的 j，然后将合并后的值插在 a[j] 后
③重复上述过程(n-1次)，直到只剩下一堆，总合并代价即为①中每次合并代价之和

算法复杂度为O(nlogn)

该定理保证，如此操作后答案不会改变

ACODE:

//O(nlogn)
#include<cstdio>
#include<algorithm>
typedef long long ll;
using namespace std;
const int MX = 4e4 + 7;
const int INF = 0x3f3f3f3f;
int n,m,ans,a[MX];
int main(int argc, char const *argv[])
{
	cin >> n;
	for(int i = 1;i <= n;++i)
		cin >> a[i];
	//方便起见 将两端的值设为INF
	a[0] = a[n+1] = INF;
	m = n;
	//n-1次合并
	for(int i = 1;i < n;++i){
		int k = 0;
		//从左往右找到第一个满足a[k-1] <= a[k + 1]的k
		for(int j = 1;j <= m;++j)
		{
			if(a[j-1] <= a[j + 1]){
				k = j;
				break;
			}
		}
		//将a[k-1] 和 a[k] 合并
 		a[k-1] += a[k];
 		for(int j = k;j < m;++j)
 			a[j] = a[j + 1];
 		ans += a[--k];
 		//从k向左找第一个a[j] > a[k - 1] + a[k] 的 j,把合并后的值插在a[j]后面
 		while(k > 0 && a[k-1] <= a[k])
 		{
 			swap(a[k-1],a[k]);
 			k--;
 		}
 		a[m] = INF;
 		m--;
	}
	cout << ans << endl;
	return 0;
}

参考自：

1. Regina8023
2. lwq12138