bzoj1584 打扫卫生 dp

链接：http://www.lydsy.com/JudgeOnline/problem.php?id=1584

题意：找到某种分割序列方法，使得每一段中所含数的种类平方之和最小。

考试时一时脑残连暴力$dp$都没写出来……

首先暴力dp应该都写得出来……$f[i]=min(f[j]+(cnt[j~i])^2)$

正解有个比较智障的优化……首先可以想到答案不会差过$n^2$（最差就是每一个一段），因此，我们只需要记录每段中有$1,2,3……sqrt(n)$个不同元素的情况，找到这些段开始的位置的前一个位置，记作$pos[j]$，那么，$f[i]=min(f[pos[j]]+j*j)$。

下面重点问题就变为$i$改变时如何修改$pos$数组。为方便我们再记录每种数字出现的上个位置$pre[j]$和每一段中有的数字种类$cnt[j]$。

首先，如果$pre[a[i]]<=pos[j]$，则$cnt[j]++$。

然后对于每一个$cnt[j]>j$的情况，暴力右移左端点，如果这时$pre[a[pos[j]]]==pos[j]$，$cnt[j]--$。

问题得解。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
const int maxn=40005;
int a[maxn],pos[maxn],cnt[maxn],pre[maxn],n,m,f[maxn];
int haha()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    memset(f,0x3f,sizeof(f));f[0]=0;int num=(int)sqrt(n);
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=num;j++)
            if(pre[a[i]]<=pos[j])cnt[j]++;
        pre[a[i]]=i;
        for(int j=1;j<=num;j++)
            while(cnt[j]>j)
            {
                pos[j]++;
                if(pre[a[pos[j]]]==pos[j])cnt[j]--;
            }
        for(int j=1;j<=num;j++)f[i]=min(f[i],f[pos[j]]+j*j);
    }
    printf("%d\n",f[n]);
}
int sb=haha();
int main(){;}