[动态规划] 电影院选座

描述

在一个大型的电影院中，有一排共 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;
}