题目链接:Codeforces Round #538 (Div. 2)_D. Flood Fill
题目描述:一排不同颜色的方块,每次可以把连续的相同颜色方块变成相邻的颜色,要把全部方块变成一种颜色最少要几步?如:
- [5,2,2,1]
- [5,5,5,1]
- [1,1,1,1]
需要三步。
思路:区间dp,用dp[l][r]存从l到r都成一种颜色时至少需要变换的次数,数组a是原方块,数组b是不同连通区域,即把连续相同的颜色区块算成一块。
if(b[l]==b[r]) dp[l][r]=dp[l+1][r-1]+1;
else dp[l][r]=min(dp[l][r-1],dp[l+1][r])+1;
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=5010;
int dp[maxn][maxn];
int a[maxn];
int b[maxn];
int main()
{
std::ios::sync_with_stdio(false);
int n,cnt=0;
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>a[i];
if(a[i]!=a[i-1]) b[++cnt]=a[i];
}
//cout<<"cnt:"<<cnt<<endl;
for(int i=2;i<=cnt;i++)
{
for(int l=1;l+i-1<=cnt;l++)
{
int r=l+i-1;
if(b[l]==b[r]) dp[l][r]=dp[l+1][r-1]+1;
else dp[l][r]=min(dp[l][r-1],dp[l+1][r])+1;
}
}
cout<<dp[1][cnt]<<endl;
}