Description
有N个人,来自K个家族.他们排成一行准备照相,但是由于天生的排外性,每个人都希望和本家族的人站在一起,中间不要加入别的家族的人.问最少从队列中去掉多少个就可以达到这个目的.
Input
第一行给出N,K。N在[1,100],K在[1,5]
第二行给出N个数,每个数为1到K中的某个数。
Output
最少从队列中去掉多少个就可以达到这个目的
Sample Input
10 3 2 1 2 2 1 1 3 1 3 3
Sample Output
2 【说明】 去掉第二个人和第七个人就可以了。
分析
这道题需要利用一下状态压缩
状态设计成三维dp(i, j, k)表示目前已经处理了i个人,一个二进制状压j表示目前为止存在的家族,目前队列的最后一个人来自k家族。
状态转移比较复杂
对于第i个人,我们有两种选择
把他去掉
这就相当于这个人不存在,所以直接将上一个人的状态拿过来就好
保留他
如果当前状态中存在这个人的家族
dp[i][j][p] = max(dp[i][j][p], dp[i - 1][j][p] + 1);
否则,枚举上一个状态的队列最后一人
for (int k = 0; k <= kk; k++) {
dp[i][j][p] = max(dp[i][j][p], dp[i - 1][j ^ q][k] + 1);
}
代码
#include<bits/stdc++.h>
using namespace std;
int n, kk;
int dp[110][90][6];
int ans;
inline int read() {
int x = 0, f = 1; char ch = getchar();
while (ch < '0' || ch > '9') {if (ch == '-') f = -1; ch = getchar();}
while (ch >= '0' && ch <= '9') {x = (x << 1) + (x << 3) + (ch ^ 48); ch = getchar();}
return x * f;
}
int main() {
n = read(), kk = read();
for (int i = 1; i <= n; i++) {
int p = read();
int q = (1 << (p - 1));
for (int j = 0; j < (1 << kk); j++) {
for (int k = 0; k <= kk; k++) {
dp[i][j][k] = dp[i - 1][j][k];
}
}
for (int j = 0; j < (1 << kk); j++) {
if (j & q) {
dp[i][j][p] = max(dp[i][j][p], dp[i - 1][j][p] + 1);
for (int k = 0; k <= kk; k++) {
dp[i][j][p] = max(dp[i][j][p], dp[i - 1][j ^ q][k] + 1);
}
}
}
}
for (int i = 0; i <= (1 << kk); i++) {
for (int k = 0; k <= kk; k++) {
ans = max(ans, dp[n][i][k]);
}
}
printf("%d", n - ans);
return 0;
}