AC日记——Paint Pearls hdu 5009

Posted 2020-09-16 Only U - IU

Paint Pearls

 

思路：

　　离散化+dp+剪枝；

　　dp是个n方的做法；

　　重要就在剪枝；

　　如果一个长度为n的区间，有大于根号n种颜色，还不如一个一个涂；

 

来，上代码：

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

#define maxn 50005

int n,ai[maxn],dp[maxn],ls[maxn],size,num,map[maxn];

inline void in(int &now)
{
    char Cget=getchar();now=0;
    while(Cget>‘9‘||Cget<‘0‘) Cget=getchar();
    while(Cget>=‘0‘&&Cget<=‘9‘)
    {
        now=now*10+Cget-‘0‘;
        Cget=getchar();
    }
}

int main()
{
    while(scanf("%d",&n)==1)
    {
        num=0;
        for(int i=1;i<=n;i++)
        {
            in(ai[i]);
            if(ai[i]==ai[i-1])
            {
                n--,i--;
                continue;
            }
            ls[i]=ai[i];
        }
        sort(ls+1,ls+n+1),size=unique(ls+1,ls+n+1)-ls-1;
        for(int i=1;i<=n;i++) ai[i]=lower_bound(ls+1,ls+size+1,ai[i])-ls,map[ai[i]]=0;
        for(int i=1;i<=n;i++)
        {
            if(!map[ai[i]]) num++;
            map[ai[i]]++,dp[i]=min(num*num,i);
        }
        for(int i=1;i<=n;i++) map[ai[i]]--;
        for(int i=1;i<=n;i++)
        {
            if(dp[i]>=dp[i+1]) continue;
            num=0;
            for(int j=i+1;j<=n;j++)
            {
                if(!map[ai[j]]) num++;
                if(num*num+dp[i]>=dp[n])
                {
                    for(int v=j-1;v>=i+1;v--) map[ai[v]]--;
                    break;
                }
                map[ai[j]]++,dp[j]=min(dp[j],dp[i]+num*num);
            }
        }
        printf("%d\n",dp[n]);
    }
    return 0;
}