http://codeforces.com/contest/711/problem/C
题意很简单:
有m种颜色
给n棵树,每棵树最后有一个颜色x,或者无颜色,用0表示。
要求染成k个子区间,这些子区间是这样划分的, 连续一段颜色一样的树算做一个子区间,不同颜色的数算做不同子区间,样例说的很清楚了
给n,m,k;要求最后染成k个子区间
接下来给一个n*m的矩阵,表示第n棵树染色颜色m的花费。
nmk不超过100
dp[i][j][k]表示 前i棵树染了j个子区间 ,最后一个树是k颜色
转移很好转移
枚举第i棵树的颜色,如果已经有颜色了,则 如果前一棵树颜色和i一样,则
dp[i][j][aa[i]]=min(dp[i][j][aa[i]],dp[i-1][j][x]);
否则
dp[i][j][aa[i]]=min(dp[i][j][aa[i]],dp[i-1][j-1][x]);
如果无颜色,则枚举 给第i棵树涂的颜色,同样按上面的思路去转移。
最坏复杂度n*m*k*k
sb题 m和k写反了找半天错误。。GGGG还是太菜了
#include <cstdio>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
#include <queue>
#include <map>
#include <set>
#include <vector>
#include <iostream>
using namespace std;
const double pi=acos(-1.0);
double eps=0.000001;
typedef long long ll;
ll mp[105][105];
ll aa[105];
ll dp[105][105][105];
int main()
{
ll n,m,k;
cin>>n>>m>>k;
for (int i=1; i<=n; i++)
scanf("%lld",&aa[i]);
for (ll i=1; i<=n; i++)
for (ll j=1; j<=m; j++)
scanf("%lld",&mp[i][j]);
for (ll i=0; i<=100; i++)
for (ll j=0; j<=100; j++)
for (ll x=0; x<=100; x++)
dp[i][j][x]=1e16;
if (aa[1])
dp[1][1][aa[1]]=0;
else
{
for (int i=1; i<=m; i++)
dp[1][1][i]=mp[1][i];
}
for (ll i=2; i<=n; i++)
{
for (ll j=1; j<=i; j++)
{
if (aa[i])
{
for (ll x=1; x<=m; x++)
{
if (x==aa[i])
dp[i][j][aa[i]]=min(dp[i][j][aa[i]],dp[i-1][j][x]);
else
dp[i][j][aa[i]]=min(dp[i][j][aa[i]],dp[i-1][j-1][x]);
}
continue;
}
for (ll x=1; x<=m; x++) //itself
{
for (ll y=1; y<=m; y++)
{
if (x==y)
dp[i][j][x]=min(dp[i][j][x],dp[i-1][j][y]+mp[i][x]);
else
dp[i][j][x]=min(dp[i][j][x],dp[i-1][j-1][y]+mp[i][x]);
}
}
}
}
ll ans=1e16;
for (int i=1; i<=m; i++)
ans=min(ans,dp[n][k][i]);
if (ans==1e16)
printf("-1\n");
else
printf("%lld\n",ans);
return 0;
}