hihocoder#1044狀態壓縮dp+滾動數組

//20ms  0
#include<iostream>
#include <string.h>
#include <stdio.h>
using namespace std;
#define  MAXN 1500
int N,M,Q;
int dp[2][MAXN];//前一次的字母以及i的前面m個字母狀態
int num[2][MAXN];//記錄前m個字母中1的個數
int w[MAXN];

int main()
{
	scanf("%d %d %d",&N,&M,&Q);
	for (int i=0;i<N;++i)
	 scanf("%d",&w[i]);
	int ans = 0;
	memset(dp,-1,sizeof(dp));
	for (int s=(1<<M)-1;s>=0;s--)
	{
		int nt = 0;
		int sums = 0;
		for (int i=0;i<M;++i)
		{
			if (s&(1<<i))
			{
				nt++;
				sums+=w[i];
			}
		}
		if (nt<=Q&&sums>dp[0][s]){
		 dp[0][s] = sums;
		 num[0][s] = nt;
		}
	}
	//上一個M個數的狀態
	for (int i=M;i<N;++i)
	{
		for (int s=(1<<M)-1;s>=0;s--)
		{
			if (dp[0][s]>=0)
			{
				int cnt;
				if (s&(1<<0))//第一位是否選了
				 cnt = num[0][s]-1;
				else
				 cnt = num[0][s];
				int state = s>>1;//右移一位
				//若不選該位置
				dp[1][s>>1] =max(dp[1][s>>1],dp[0][s]); 
				if (cnt<Q)
				 dp[1][state|(1<<(M-1))] = max(dp[1][state|(1<<(M-1))],dp[0][s]+w[i]);
			}
		}
		memcpy(dp[0],dp[1],sizeof(dp[0]));
		memset(dp[1],-1,sizeof(dp[1]));
	}

	for (int s=(1<<M)-1;s>=0;s--)
	{
		if (dp[0][s]>ans)
		 ans = dp[0][s];
	}
	printf("%d\n",ans);
	return 0;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章