「NOIP2018模拟赛」 最优序列 - 状压Dp

题目描述

给出一个长度为n(n<=1000) 的正整数序列，求一个子序列，使得原序列中任意长度为 m的子串中被选出的元素不超过k(k<=m<=10)个，并且选出的元素之和最大。

分析

看到范围，应该是能想到状压Dp的，然而在考场上只打出了一个10分的暴力。当然，可以用费用流做，不过网络流应该不是NOIP的考点吧。

定义状态为前i个数，以i为结尾往前m个数的状态为j所得的最大值。容易知道对于第p个数，只有选与不选两种情况，以此作为转移的要素，可以先将这个状态的最高位抹去，在左移1位，使之形成i-1的状态，进行更新。对于每个状态的j是固定的，所以可以先预处理出合法的状态，即状态中1的个数不超过k的状态。本题还可用滚动数组来优化空间，不过对于n<=1000的数据也没必要了。

代码

#include <iostream>
#include <cstring>
#include <cstdio>
#define lowbit(x) ((x)&-(x))
using namespace std;
int n,m,k,mx;
int a[1005];
bool st[1025];
int f[2][1025];
int main() {
	scanf("%d%d%d",&n,&m,&k);
	for (int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	mx=1<<m;
	for (int i=0;i<mx;i++) {//预处理合法状态 
		int ret=0;
		for (int j=i;j;j-=lowbit(j)) {//lowbit为二进制下最低位的1 
			ret++;
		}
		if (ret<=k) st[i]=1;
	}
	f[0][0]=0;
	int now=0;
	for (int i=1;i<=n;i++) {
		now^=1;
		memset(f[now],-0x3f,sizeof f[now]);
		for (int j=0;j<mx;j++) {
			if (st[j]&&f[now^1][j]>=0) {
				int s0=(j&((1<<m-1)-1))<<1,s1=(j&((1<<m-1)-1))<<1|1;
				//将j的最高位抹去,左移1位,s0为不取的状态,s1为取的状态 
				if (st[s0])
				f[now][s0]=max(f[now][s0],f[now^1][j]);//不取第i个 
				if (st[s1])
				f[now][s1]=max(f[now][s1],f[now^1][j]+a[i]);//取第i个 
			}
		}
	}
	int ans=-0x3f3f3f3f;
	for (int i=0;i<mx;i++)
		if (st[i]) ans=max(ans,f[now][i]);//找答案 
	printf("%d",ans);
	return 0;
}