洛谷 P5569 [SDOI2008]石子合并 GarsiaWachs算法

石子合并终极通用版

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