Codeforces(E. Array Shrinking)区间DP

比较裸的一道区间DP
比赛的时候D都没推出来,这道题连看都没看。。。

直接套区间dp的模板就行了
我跑了两遍,一个跑d[][],一个跑f[][]
d[i][j]表示 i ~ j 能合并成的最大值,如果d[i][j]不为0还表示i~j能合并成一个数
f[i][j]表示 i ~ j 的最小长度

代码:

#include <bits/stdc++.h>
#define ll long long
using namespace std;
int a[1000],d[600][600],f[600][600],inf=99999999;
void init()
{
    for(int i=0;i<550;i++)
    {
        for(int j=0;j<=550;j++)
        {
            if(i==j)
            {f[i][j]=1;}
            else
            {f[i][j]=inf;}
        }
    }
}
int main()
{
    int n;
    cin>>n;
    for(int i=1;i<=n;i++)
    {cin>>a[i];d[i][i]=a[i];}
    for(int len=2;len<=n;len++)
    {
        for(int i=1;i<=n;i++)
        {
            int j=i+len-1;
            if(j>n){break;}
            for(int k=i;k<j;k++)
            {
                if(d[i][k]==d[k+1][j]&&d[i][k]!=0)
                {
                    d[i][j]=max(d[i][j],d[i][k]+1);
                }
            }
        }
    }
    if(d[1][n]!=0)
    {cout<<"1"<<endl;return 0;}
    init();
    for(int len=2;len<=n;len++)
    {
        for(int i=1;i<=n;i++)
        {
            int j=i+len-1;
            if(j>n){break;}
            for(int k=i;k<j;k++)
            {
                int l,r;
                if(d[i][k]!=0)
                {l=1;}
                else
                {l=k-i+1;}
                if(d[k+1][j]!=0)
                {r=1;}
                else
                {r=j-(k+1)+1;}
                if(len==2)
                {
                    if(d[i][k]==d[k+1][j])
                    {f[i][j]=1;}
                    else
                    {f[i][j]=2;}
                    continue;
                }
                f[i][j]=min(f[i][j],l+r);
                f[i][j]=min(f[i][j],f[i][k]+f[k+1][j]);
                f[i][j]=min(f[i][j],l+f[k+1][j]);
                f[i][j]=min(f[i][j],f[i][k]+r);
            }
        }
    }
    cout<<f[1][n]<<endl;
    return 0;
}

发布了36 篇原创文章 · 获赞 4 · 访问量 1383

猜你喜欢

转载自blog.csdn.net/qq_43781431/article/details/104827517