CF1922F Replace on Segment 题解

最新推荐文章于 2024-10-15 19:28:18 发布

爱编程的pyy

最新推荐文章于 2024-10-15 19:28:18 发布

阅读量771

点赞数 34

分类专栏：题解文章标签：算法 c++ 动态规划

本文链接：https://blog.csdn.net/pyy_official/article/details/135901758

版权

题解专栏收录该内容

7 篇文章 0 订阅

订阅专栏

题目传送门洛谷题目传送门

思路

考虑区间 DP。

设 $f_{i,j,k}$ 表示把区间 $[i, j]$ 变成 $k$ 的最小花费，很明显有转移 $f_{i,j,k} = \displaystyle \min_{l=i}^{j-1}(f_{i,l,k}+f_{l+1,j,k})$ 。

这就完了？一道紫题？怎么可能！

发现这样转移会算漏，因为可以把区间 $[i, j]$ 变成不含 $k$ 的，再花一步变成全部都是 $k$ 。

考虑设 $g_{i,j,k}$ 表示把区间 $[i, j]$ 变成不含 $k$ 的最小花费。除了普通的区间 DP，我们还可以把区间 $[i, j]$ 先变成不含 $k^{'}$ ，再全部变成 $k^{'}$ ，这样也能达到要求。

所以容易写出转移： $g_{i,j,k}=\displaystyle \min_{l=i}^{j-1}(g_{i,l,k}+g_{l+1,j,k})$ ，然后记 $mn$ 为 $\displaystyle \min_{k=1}^{x} g_{i,j,k}$ ，则有 $g_{i,j,k} = \min\{g_{i,j,k},mn+1\}$ 。

时间复杂度 $O(n^3x)$ ，看似不能通过，但是区间 DP 会带一个大约 $\frac{1}{4}$ 的常数。实测跑得飞快。

代码

#include <bits/stdc++.h>
using namespace std;

template<typename T> inline void read(T &x)
{
	x = 0;
	T f = 1;char ch = getchar();
	while(ch<'0'||ch>'9')
	{
		if(ch=='-')
		{
			f = -1,ch = getchar();
			break;
		}
		ch = getchar();
	}
	while(ch>='0'&&ch<='9')
		x = (x<<3)+(x<<1)+ch-48,ch = getchar();
	x*=f;
}
template<typename T> inline T read()
{
	T x;read(x);return x;
}
template<typename T> void write(T x)
{
    if(x<0) x = -x,putchar('-');
    if(x>9) write(x/10);
    putchar(x%10+48);
}
template<typename T> inline void writen(T x)
{
    write(x);
	putchar(10);
}
const int N = 1e2+5;
int n,m,a[N],f[N][N][N],g[N][N][N];
void solve()
{
    read(n),read(m);
    for(int i = 1;i<=n;i++) 
        for(int j = i;j<=n;j++)
            for(int k = 1;k<=m;k++)
                f[i][j][k] = g[i][j][k] = 2e9;
    for(int i = 1;i<=n;i++)
        for(int j = 1;j<=m;j++)
            g[i][i][j] = 0,f[i][i][j] = 1;
    for(int i = 1;i<=n;i++)
        read(a[i]),f[i][i][a[i]] = 0,g[i][i][a[i]] = 1;
    for(int len = 2;len<=n;len++)
        for(int i = 1,j = len;j<=n;i++,j++)
        {
            int mn = 2e9;
            for(int k = 1;k<=m;k++)
            {
                for(int l = i;l<j;l++)
                    g[i][j][k] = min(g[i][l][k]+g[l+1][j][k],g[i][j][k]);
                mn = min(mn,g[i][j][k]);
            }
            for(int k = 1;k<=m;k++) g[i][j][k] = min(g[i][j][k],mn+1),f[i][j][k] = g[i][j][k]+1;
            for(int k = 1;k<=m;k++)
                for(int l = i;l<j;l++)
                    f[i][j][k] = min(f[i][j][k],f[i][l][k]+f[l+1][j][k]);
        }
    int ans = 2e9;
    for(int i = 1;i<=m;i++)
        ans = min(ans,f[1][n][i]);
    writen(ans);
}
signed main()
{
//	freopen(".in","r",stdin);
//	freopen(".out","w",stdout);
    int T;
    read(T);
    while(T--) solve();
	return 0;
}