Pangu and Stones HihoCoder - 1636 区间dp

题目链接：https://vjudge.net/problem/HihoCoder-1636
题意：合并石子，每次合并的堆数有限制。

#include <bits/stdc++.h>
using namespace std;
const int inf=0x3f3f3f3f;
int sum[105],dp[105][105][105],a[105];
int main()
{

    int n,l,r;
    while(~scanf("%d%d%d",&n,&l,&r))
    {
        sum[0]=0;
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&a[i]);
            sum[i]=sum[i-1]+a[i];
        }
        memset(dp,inf,sizeof(dp));
        for(int i=0; i<=n; i++)
        {
            dp[i][i][1]=0;//dp i j k 表示从i到j包含k堆石子花费时间的最小值
        }
        for(int len=2; len<=n; len++) //区间长度
        {
            for(int i=1; i<=n; i++) //左端点
            {
                int j=len+i-1;//右端点
                if(j>n)
                    break;
                for(int k=2; k<=min(len,r); k++) //合成堆数
                {
                    for(int p=i; p<j; p++) //枚举划分
                    {
                        dp[i][j][k]=min(dp[i][j][k],dp[i][p][k-1]+dp[p+1][j][1]);
                    }
                    if(k>=l)//符合条件
                    {
                        dp[i][j][1]=min(dp[i][j][1],dp[i][j][k]+sum[j]-sum[i-1]);
                    }
                }
            }
        }
        if(dp[1][n][1]==inf)
            printf("0\n");
        else
            printf("%d\n",dp[1][n][1]);
    }
    return 0;
}