题目网址:http://hihocoder.com/problemset/problem/1044
未得到状态压缩的状态转移方程
p1 , p2 ..... pm-1 是 第 i 次的前 m 次 状态的 01 状态
目标就是压缩这些01字符串
压缩后:
这个方程其实不太好理解 , 看代码的 s0 状态 还有 s1状态会更好理解。
代码:
#include <cstdio>
#include <iostream>
#include <cstring>
using namespace std;
#define maxn 10
int dp[1003][1<<maxn] ;
int ok[1 << maxn] ;
int w[1003];
void init(int m , int q)
{
for(int i = 0;i < (1 << m) ;i++)
{
int cnt = 0 , val = i ;
while(val)
{
cnt += (val % 2) , val >>= 1;
}
ok[i] = (cnt <= q);
}
}
int main()
{
int n , m , q ;
while(~scanf("%d%d%d",&n,&m,&q))
{
for(int i = 1;i <= n;i++)
scanf("%d",w + i) ;
init( m , q) ;
int ch = (1 << m) - 1 ; // 用于取出低m位字符串
memset(dp , 0 , sizeof(dp));
for(int i = 1;i <= n ;i++)
{
for(int j = 0; j < (1 << m);j++)
{
int s0 = (j << 1) & ch ;
int s1 = ((j << 1) | 1) & ch ;
dp[i][s0] = max(dp[i-1][j] , dp[i][s0]) ; //注意这里要不断去最大值
// 状态有可能会交错
if(ok[s1])
{
dp[i][s1] = max(dp[i-1][j] + w[i] , dp[i][s1]) ;
}
}
}
int ans = 0 ;
for(int j = 0;j < (1 << m);j++)
if(dp[n][j] > ans) ans = dp[n][j] ;
printf("%d\n",ans);
}
}