POJ 1738 石子合并2 GarsiaWachs算法

石子合并(GarsiaWachs算法)

只能用该算法过！！！

详解看代码

//#pragma comment(linker, "/STACK:167772160")//手动扩栈~~~~hdu 用c++交
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<queue>
#include<stack>
#include<cmath>
#include<set>
#include<algorithm>
#include<vector>
// #include<malloc.h>
using namespace std;
#define clc(a,b) memset(a,b,sizeof(a))
#define LL long long
const int Inf = 0x3f3f3f3f;
const double eps = 1e-5;
const double pi = acos(-1);
// inline int r(){
//     int x=0,f=1;char ch=getchar();
//     while(ch>'9'||ch<'0'){if(ch=='-') f=-1;ch=getchar();}
//     while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
//     return x*f;
// }

int stone[50010];
int n,t,ans;

void combine(int k){
    int tem=stone[k]+stone[k-1];//合并k和k-1堆
    ans+=tem;
    for(int i=k;i<t-1;i++) stone[i]=stone[i+1];//k以后的往前移位
    t--;
    int j;
    for(j=k-1;j>0&&stone[j-1]<tem;j--) stone[j]=stone[j-1];//k-1以后的往后移位
    stone[j]=tem;//在j处插入tem
    while(j>2&&stone[j-2]<=stone[j]){//在新得到的序列里递归处理
        int d=t-j;
        combine(j-1);
        j=t-d;//回溯
    }
}
int main(){
    while(~scanf("%d",&n)){
      if(n==0) break;
      for(int i=0;i<n;i++) scanf("%d",&stone[i]);
      t=1;
      ans=0;
      for(int i=1;i<n;i++){
          stone[t++]=stone[i];
        while(t>3&&stone[t-3]<stone[t-1]) combine(t-2);//从1开始遍历到结尾
      }
      while(t>1) combine(t-1);//合并完后如果不为一堆再合并一次
      printf("%d
",ans);
    }
    return 0;
}