描述
在一个大型的电影院中,有一排共 n 个座位,每个座位的舒适度不同,用一个数组 comfort[i] 表示第 i 个座位的舒适度。电影院希望安排 m 个观众入场,使得所有观众的舒适度总和最大。但是,由于安全规定,相邻的座位不能同时安排两个观众。
输入
第一行包含两个整数 n 和 m ,分别表示座位的数量和需要安排的观众数量。
第二行包含 n 个正整数(≤ 10000)),表示每个座位的舒适度。
其中 n ≤ 4000,m ≤ ceil(n/2.0)。ceil(n/2.0)对n/2.0上取整
输出
输出一个整数,表示最大可能的舒适度总和。
样例输入
5 3 1 5 3 4 2
样例输出
6
解题分析
这是一道很典型的动态规划题目,我们可以规定一个二维的dp数组,dp[i][j]表示到第i个位置,安排了j个人后座位的最大可能舒适度总和。那么,考虑到“取或不取”的原则,dp[i][j]的第i个座位,我们可以选择安排人坐,那么我们只要在前i-1个位置安排j-1个人即可,如果说我们不安排人坐,那么我们就要在前i-2人位置安排j-1个人(此处i-1个位置已经不能坐人了),所以我们很容易地得到递推公式dp[i][j]=max{dp[i-1][j],dp[i-1][j-1]+confort[i]}。考虑到dp数组的更新情况,我们从小到大循环更新。接下来我们考虑一下一些临界的初始条件,首先dp[i][0]的舒适度都为0,这个没有悬念,因为没有人入座,也就没有舒适度一说;然后就是dp[i][1],这个的话我们只要从第1个座位到第i个座位中选择一个舒适度最大的座位,让这个人坐下去即可。注意到为了减少不必要的更新,我们可以限定j<=ceil(n/2)。
代码实现
#include <iostream>
#include <cstdio>
#include <queue>
#include <bitset>
#include <map>
#include <set>
#include <algorithm>
#include <cstdlib>
#include <unordered_map>
#include <unordered_set>
#include <cmath>
#include <vector>
#include <sstream>
using namespace std;
int n,m;
int comfort[4005]={0};
int dp[4005][2005]={0};
//dp[i][j] = max(dp[i-1][j],dp[i-2][j-1]+comfort[i]);
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
scanf("%d",&comfort[i]);
}
for(int i=1;i<=n;i++){
dp[i][1]=*max_element(comfort+1,comfort+i+1);
}
for(int i=2;i<=n;i++)
for(int j=2;j<=ceil(n/2.0);j++){
if(j<=ceil((i-1)/2.0))
dp[i][j] = max(dp[i-1][j],dp[i][j]);
if(j-1<=ceil((i-2)/2.0))
dp[i][j] = max(dp[i][j],dp[i-2][j-1]+comfort[i]);
}
printf("%d\n",dp[n][m]);
return 0;
}