2021 icpc昆明 C.Cities(区间dp)

LINK

最近快没动力了,什么题都不会,我不知道我在学什么

题意

给定一个长度$n$的数组$a$,$a_i\in [1,n]$且没有元素出现过多于$15$次

现在每次可以把一段颜色相同的区间染成另一种颜色

问,最少操作多少次使得$a$数组只剩下一种颜色。

---

比较明显的区间$dp$,主要是转移不好想

定义$f[i][j]$表示$[i,j]$染成一种颜色的最小操作次数

$$f[i][j]=min(f[i][j-1],f[i+1][j])+1$$

但是当两端相等时我们可以节省一次操作

$$f[i][j]=min(f[i][k]+f[k+1][j])$$

此时需要满足$a[k]==a[j]$,把左边染成$k$颜色,右边染成$j$颜色

题目保证了$k$只需要不超过$15$次

所以初始先把颜色相同的缩点,然后去转移

无法证明其正确性。

#include <bits/stdc++.h>
using namespace std;
const int maxn = 5009;
int t,n,a[maxn],col[maxn],f[maxn][maxn],top,nxt;
int pre[maxn],las[maxn];
int main()
{
	cin >> t;
	while( t-- )
	{
		cin >> n; top = 0;
		for(int i=1;i<=n;i++)	cin >> a[i];
		for(int i=2;i<=n;i++)
		{
			if( a[i]==a[i-1] )	continue;
			else	col[++top] = a[i-1];
		}
		memset( pre,0,sizeof pre );
		memset( las,0,sizeof las );
		col[++top] = a[n];
		for(int i=1;i<=top;i++)	
			pre[i] = las[col[i]],las[col[i]] = i;
		memset( f,20,sizeof f );
		for(int i=1;i<=top;i++)	f[i][i] = 0;
		for(int l=2;l<=top;l++)
		for(int i=1;i+l-1<=top;i++)
		{
			int j = i+l-1;
			f[i][j] = min( f[i+1][j],f[i][j-1] )+1;
			for(int k=pre[j];k>=i;k=pre[k] )
				f[i][j] = min( f[i][j],f[i][k]+f[k+1][j] );
		}
		cout << f[1][top] << endl;
	}
}