题意:N个歹徒去一个餐馆,旅馆门有k个状态,每个歹徒拥有自己需要的状态(价格)和繁荣度(价值),歹徒在Ti 时刻到餐厅来(多个歹徒可能同时来餐馆),若此时刻门的状态与歹徒的状态相同,则歹徒就进入餐馆,同时餐馆获得这个歹徒相应的繁荣度(价值),若歹徒到来时门的状态与歹徒的状态不同,则歹徒离开且不再回来。门的状态在每个单位时间里改变一个单位(加一或减一),可以保持当前的打开程度不变。
刚开始参考背包问题的模型,想的是dp[i][j],i表示第i个歹徒,j表示j时刻,但是这样的话dp值不只要保存当前的最大价值,还要保存当前门的状态,所以dp的定义是这样的
struct State{
int val;
int sta;
} dp[333][33333];
然后空间优化一下
struct State{
int val;
int sta;
} dp[33333];
然后状态转移会有两层循环,第一层历遍每个匪徒(0<i<=100),第二层历遍每个时间(0<j<=30000),
其实这样做很蠢,
第一,歹徒进入餐厅后的状态(j>demo[i].t)与当前歹徒完全无关,所以第二层循环就可以改为(0<j<demo[i].t);
第二,歹徒进入餐厅的顺序是固定的,完全可以先给歹徒排序,而且歹徒们是离散地进入餐厅地,所以(0<j<demo[i].t)里面的很多状态其实没有必要判断。所以第二层循环可以改成
for (int j = 0; j < i; j++)
这时候dp数组的下表也没必要保存时间了,直接保存第几个歹徒就行了。
State dp[333];
既然保存的是第几个歹徒,那我就可以直接知道这个歹徒的t值,所以可以这样
int dp[333];
这时候,代码已经和最大上升子序列很相似了,关键代码就可以这样写
for (int i = 0; i < n; i++) {
if (demo[i].t >= demo[i].s) dp[i] = demo[i].p;
for (int j = 0; j < i; j++) {
if (dp[j] == 0) continue;
if (abs(demo[i].s - demo[j].s) <= demo[i].t - demo[j].t)
dp[i] = max(dp[i], dp[j] + demo[i].p);
}
}
完整代码
#include <iostream>
#include <cstdio>
#include <iomanip>
#include <string>
#include <cstdlib>
#include <cstring>
#include <queue>
#include <set>
#include <vector>
#include <map>
#include <algorithm>
#include <cmath>
#include <stack>
#include <stdlib.h>
#include <stdio.h>
#define INF 0x3f3f3f3f
#define LINF 0x3f3f3f3f3f3f3f3f
#define ll long long
#define ull unsigned long long
#define uint unsigned int
#define l(x) x<<1
#define r(x) x<<1|1
#define ms(a,b) memset(a,b,sizeof(a))
using namespace std;
struct Gan {
int t, p, s;
};
int n, k, t;
Gan demo[200];
int dp[333];
bool cmp(const Gan& a, const Gan& b) {
return a.t < b.t;
}
int main(){
scanf("%d%d%d", &n, &k, &t);
for (int i = 0; i < n; i++)
scanf("%d", &demo[i].t);
for (int i = 0; i < n; i++)
scanf("%d", &demo[i].p);
for (int i = 0; i < n; i++)
scanf("%d", &demo[i].s);
sort(demo, demo + n,cmp);
memset(dp, 0, sizeof(dp));
for (int i = 0; i < n; i++) {
if (demo[i].t >= demo[i].s) dp[i] = demo[i].p;
else continue;
for (int j = 0; j < i; j++) {
if (dp[j] == 0) continue;
if (abs(demo[i].s - demo[j].s) <= demo[i].t - demo[j].t)
dp[i] = max(dp[i], dp[j] + demo[i].p);
}
}
int res = 0;
for (int i = 0; i < n; i++) {
res = max(res, dp[i]);
}
printf("%d\n", res);
return 0;
}