洛谷P2389 电脑班的裁员 （动态规划）

2020.6.17
这道题比较曲折，我刚开始看范围以为是个区间dp，后来看到n = 1000, k = 500，所有状态1000 x 1000 x 500 貌似开不下。那就没法整了，然后想到了记忆化搜索。首先对于所有的位置，有两种选项，选或者不选，如果选的话肯定是看从上一位选的情况或者不选的情况加上现在位置上的值转移过来，如果不选的话那就从上一个位置选或者不选转移过来，这就行了。然后比较下在n这个地方是选还是不选好。结果记忆化调了半天就50分。。。。然后我看到代码恍然大悟，这状态转移的方法不都给我了么还用个球的记忆化啊（错误 ）因为我看到三维数组下意识地去想n^3解法，然后完全没发现第三个状态只有2个。。。。。
那就简单了，写了一发过了。

upd: 我忘考虑s == 0的时候了，抱歉，题解已更正

我觉得没问题啊，但是记忆化就是不对。。。。有大佬知道我哪里错了欢迎指出！！
代码：
50分的记忆化（更正，刚删掉了点东西，70了）：

#include <bits/stdc++.h>
using namespace std;
#define limit (1000 + 5)//防止溢出
#define INF 0x3f3f3f3f
#define inf 0x3f3f3f3f3f
#define lowbit(i) i&(-i)//一步两步
#define EPS 1e-6
#define FASTIO  ios::sync_with_stdio(false);cin.tie(0);
#define ff(a) printf("%d\n",a );
#define pi(a,b) pair<a,b>
#define rep(i, a, b) for(int i = a; i <= b ; ++i)
#define per(i, a, b) for(int i = b ; i >= a ; --i)
#define mint(a,b,c) min(min(a,b), c)
#define MOD 998244353
#define FOPEN freopen("C:\\Users\\tiany\\CLionProjects\\acm_01\\data.txt", "rt", stdin)
typedef long long ll;
typedef unsigned long long ull;
ll read(){
    ll sign = 1, x = 0;char s = getchar();
    while(s > '9' || s < '0' ){if(s == '-')sign = -1;s = getchar();}
    while(s >= '0' && s <= '9'){x = x * 10 + s - '0';s = getchar();}
    return x * sign;
}//快读
void write(ll x){
    if(x < 0) putchar('-'),x = -x;
    if(x / 10) write(x / 10);
    putchar(x % 10 + '0');
}

int n, k,m;
int dp[limit][limit][2];//对应
int a[limit];
int dfs(int cur, int s, int flag){
    if(dp[cur][s][flag] || cur < 1 || !s)return dp[cur][s][flag];//如果越界或者已经被搜过
    int ans = 0;
    if(flag)ans = max(dfs(cur- 1, s - 1, !flag), dfs(cur - 1, s ,flag) ) + a[cur];
    else ans = max(dfs(cur - 1,s,flag), dfs(cur - 1,s,!flag));
    return dp[cur][s][flag] = ans;
}
int main() {
#ifdef LOCAL
    FOPEN;
#endif
    n = read(), k = read();
    rep(i ,1, n){
        a[i] = read();
    }
    write(max(dfs(n + 1 , k, 1),dfs(n + 1,k,0)));
    return 0;
}

AC的dp：

#include <bits/stdc++.h>
using namespace std;
#define limit (1000 + 5)//防止溢出
#define INF 0x3f3f3f3f
#define inf 0x3f3f3f3f3f
#define lowbit(i) i&(-i)//一步两步
#define EPS 1e-6
#define FASTIO  ios::sync_with_stdio(false);cin.tie(0);
#define ff(a) printf("%d\n",a );
#define pi(a,b) pair<a,b>
#define rep(i, a, b) for(int i = a; i <= b ; ++i)
#define per(i, a, b) for(int i = b ; i >= a ; --i)
#define mint(a,b,c) min(min(a,b), c)
#define MOD 998244353
#define FOPEN freopen("C:\\Users\\tiany\\CLionProjects\\acm_01\\data.txt", "rt", stdin)
typedef long long ll;
typedef unsigned long long ull;
ll read(){
    ll sign = 1, x = 0;char s = getchar();
    while(s > '9' || s < '0' ){if(s == '-')sign = -1;s = getchar();}
    while(s >= '0' && s <= '9'){x = x * 10 + s - '0';s = getchar();}
    return x * sign;
}//快读
void write(ll x){
    if(x < 0) putchar('-'),x = -x;
    if(x / 10) write(x / 10);
    putchar(x % 10 + '0');
}

int n, k,m;
int dp[limit][limit][2];//对应
int a[limit];
int main() {
#ifdef LOCAL
    FOPEN;
#endif
    n = read(), k = read();
    rep(i ,1, n){
        a[i] = dp[i][1][1] = read();
    }
    
    rep(i ,1, n){
        rep(j ,1, k){
            dp[i][j][0] = max(dp[i - 1][j][0] , dp[i-1][j][1]);//选和不选
            dp[i][j][1] = max(dp[i - 1][j - 1][0] + a[i] , dp[i - 1][j][1] + a[i]);
        }
    }
    int maxx = max(dp[n][k][0], dp[n][k][1]);
    write(maxx);
    return 0;
}