【算法】DP+数学优化

【题意】把n个1~m的数字分成k段,每段的价值为段内不同数字个数的平方,求最小总价值。n,m,ai<=40000

【题解】

参考自:WerKeyTom_FTD

令f[i]表示把前i个数分成若干段的最小价值。

转移中我们定义,从i开始往前到有j个不同的数的最小位置为b[j]。

f[i]=f[b[j]-1]+j^2。

考虑最坏情况,每个数自成一段,则总价值为n。

所以当段内不同的数个数>√n时,就不可能是最优解了(此时价值>n)。

所以f[i]=f[b[j]-1]+j^2,1<=j<=√n。

快速计算的关键在递推b[j]数组,首先在递推过程中同步计算上一个等数位置last[]和桶c[]。

枚举j:

如果last[i]>=b[j],不会新增数,不改变。

否则,b[j]++直到c[a[b[j]]]=b[j],此时b[j]++得到新的b[j]。

注意若元素个数不满√n个,出现新元素就top++。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int maxn=; int n,N,a[maxn],last[maxn],c[maxn],b[maxn],f[maxn],m,top;
int main(){
scanf("%d%d",&n,&N);
for(int i=;i<=n;i++)scanf("%d",&a[i]);
m=(int)sqrt(n)+;
top=;
for(int i=;i<=n;i++){
last[i]=c[a[i]];
c[a[i]]=i;
for(int j=;j<=min(m,top);j++)if(last[i]<b[j]){
while(c[a[b[j]]]!=b[j])b[j]++;
b[j]++;
}
if(!last[i])b[++top]=;
f[i]=i;
for(int j=;j<=min(m,top);j++)f[i]=min(f[i],f[b[j]-]+j*j);
}
printf("%d",f[n]);
return ;
}
05-12 15:24