（纪中）1299. 洗盘子【DP】

---

解题思路

我们可以设吃掉第i盘菜的最少时间为$f_i$
首先我们要明确，这个最少时间的总数一定不超过$n$（一个一个吃都最大为n），设$j$为一次吃$j$盘菜，那么我们显然可知j最大为$sqrt(n)$

那么我们构思一个O(nsqrt(n))的算法：
用$pi,j$表示从第i盘菜往前数，得到$j$个菜种的最远离第$i$盘菜的菜的位置，然后由于每次都需要更新这个p值（因为加了第i盘菜嘛）所以我们把$pi,j$简化为$pj$
得方程：$fi=minf[p[j]-1]+j^2$

那么我们要怎么维护p？我们设一个$lastk$表示第k种编号的菜系最迟出现在什么地方，然后分两种情况：

• $last[dishi]$<$pj$（表示$pj$~ $i$ 里没 $dishi$ 这盘菜），$pj=pj-1$（因为将要多一个菜）
• $last[dishi]$>=$pj$（表示$pj$~ $i$ 里有 $dishi$ 这盘菜），$pj=pj$

---

代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<iomanip>
#include<cstring>
#include<cmath>
#include<map>
#include<queue>
#define ll long long
#define ldb long double
using namespace std;

int n,m,k,l,a,f[40010],p[40010],last[40010];

int main() {
	freopen("cleanup.in","r",stdin);
	freopen("cleanup.out","w",stdout);
	memset(last,-1,sizeof(last));
	scanf("%d%d",&n,&m);
	scanf("%d",&a);
	l=sqrt(n);
	f[1]=1;
	p[1]=1;
	last[a]=1;
	for(int i=2; i<=n; i++) {
		scanf("%d",&a);
		p[0]=i;
		for(int j=l;j>0;j--)
			if(last[a]<p[j]) 
				p[j]=p[j-1];
		f[i]=i;
		for(int j=1;j<=l;j++)
			if(p[j]-1>0)
				f[i]=min(f[i],f[p[j]-1]+j*j);
		last[a]=i;
	}
	printf("%d",f[n]);
}