BZOJ 3229: [Sdoi2008]石子合并 GarsiaWachs算法

3229: [Sdoi2008]石子合并

Time Limit: 3 Sec  Memory Limit: 128 MB
Submit: 844  Solved: 343
[Submit][Status][Discuss]

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

高级石子合并

石子合并问题的专门算法GarsiaWachs算法： 
先从序列中找第一个st【k】使得st【k-1】<=st【k+1】然后合并st【k-1】与st【k】； 
再从序列中从k往前找第一个st【j】使得st【j】>st【k-1】+st【k】然后将这个合并后的放在j位置后； 
如此往复直到只剩一堆； 
此题暴力枚举即可，时间复杂度O（n^2）可以用平衡树去维护那个序列，实现O（nlogn）

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<complex>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<map>
#include<set>
using namespace std;

typedef long long ll;

inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
	return x*f;
}
void print(ll x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print(x/10);putchar(x%10+'0');}

const int N=40010;

int t=1,a[N],n;

ll ans=0;

void combine(int k)
{
	register int i,tmp=a[k-1]+a[k];
	ans+=tmp;
	for(i=k;i<t-1;++i)a[i]=a[i+1];t--;
	for(i=k-1;i&&a[i-1]<tmp;--i)a[i]=a[i-1];
	a[i]=tmp;
	while(i>2&&a[i]>=a[i-2])
	{
		int d=t-i;
		combine(i-1);
		i=t-d;
	}
}

int main()
{
	n=read();
	register int i;
	for(i=0;i<n;++i)a[i]=read();
	for(i=1;i<n;++i){a[t++]=a[i];while(t>=3&&a[t-3]<=a[t-1])combine(t-2);}
	while(t>1)combine(t-1);
	print(ans);puts("");
	return 0;
}
/*
4
1
1
1
1

8
*/