HDU 4960：Another OCD Patient

HDU 4960：Another OCD Patient

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4960

题目大意：有$n$个数$v_i$，现要求将相邻的一些数合并使得合并后为回文数列，将$k$个数合并需要代价$a[k]$，一个数只能被合并一次，问最小代价。

分治+记忆化

代码如下：

#include <iostream>
#include <cstdio>
#include <cstring>
#define N 5005
using namespace std;
typedef long long ll;
const ll inf=1000000000000000;
ll n,v[N],a[N],pre[N],ans,l,r,dp[N];
ll solve(ll l,ll r){
    if(l<1||r>n)return inf;
    if(dp[l]!=-1)return dp[l];
    ll lt=l--,rt=r++;
    dp[lt]=a[lt]+a[n-rt+1];
    while(1<=l&&r<=n){
        ll L=pre[l],R=pre[n]-pre[r-1];
        if(L==R){
            dp[lt]=min(dp[lt],a[lt-l]+a[r-rt]+solve(l,r));
            l--;
        }else if(L>R){
            l--;
        }else r++;
    }
    return dp[lt];
}
int main(void){
    while(~scanf("%lld",&n)){
        if(n==0)break;
        memset(dp,-1,sizeof(dp));
        for(int i=1;i<=n;++i){
            scanf("%lld",&v[i]);
            pre[i]=pre[i-1]+v[i];
        }
        for(int i=1;i<=n;++i)
            scanf("%lld",&a[i]);
        ans=a[n],l=1,r=n;
        while(l<r){
            ll L=pre[l],R=pre[n]-pre[r-1];
            if(L==R){
                ans=min(ans,solve(l,r)+a[r-l-1]);
                l++;
            }else if(L<R){
                l++;
            }else r--;
        }
        printf("%lld\n",ans);
    }
}