涂色最麻烦的方案就是每个点涂一次颜色,花费为n,故不管怎么样任何一次涂色都不能有大于sqrt(n)种颜色。
然后就可以写个dp了,dp[i]存储涂完前i个珍珠的花费,记录一个left数组,left[j]表示用j种颜色能涂到的最左边的左边一个。
然后递推dp[i]时就枚举j求一个dp[left[j]]+j*j最小就行了,求dp[i]时的left[j]时,用before记录dp[i-1]的left[j-1],然后就很好推了。
#include<cstdio>
#include<cmath>
#include<iostream>
#include<cstring>
using namespace std;
int Scan() //输入外挂
{
int res=0,ch,flag=0;
if((ch=getchar())=='-')
flag=1;
else if(ch>='0'&&ch<='9')
res=ch-'0';
while((ch=getchar())>='0'&&ch<='9')
res=res*10+ch-'0';
return flag?-res:res;
}
int dp[50010],a[50010],Left[225],before;
int main()
{
int n;
while(scanf("%d",&n)!=EOF)
{
getchar();
for(int i=1;i<=n;i++)a[i]=Scan();
memset(Left,0,sizeof(Left));
memset(dp,0,sizeof(dp));
int t=sqrt(n);a[0]=0;
for(int i=1;i<=n;i++)
{
dp[i]=i;
for(int j=1;j<=t;j++)
{
if(j==1)
{
before=Left[j];
if(a[i]!=a[i-1])Left[j]=i-1;
else break;
}
else
{
if(a[before]!=a[i])
{
int temp=Left[j];
Left[j]=before;
before=temp;
}
else break;
}
}
for(int j=1;j<=t;j++)dp[i]=min(dp[i],dp[Left[j]]+j*j);
}
printf("%d\n",dp[n]);
}
return 0;
}