洛谷P2389 电脑班的裁员

题目背景

隔壁的新初一电脑班刚考过一场试，又到了BlingBling的裁员时间，老师把这项工作交给了ZZY来进行。而ZZY最近忙着刷题，就把这重要的任务交（tui）给了你。

题目描述

ZZY有独特的裁员技巧：每个同学都有一个考试得分ai（-1000<=ai<=1000），在n个同学（n<=500）中选出不大于k段（k<=n）相邻的同学留下，裁掉未被选中的同学，使剩下同学的得分和最大。要特别注意的是，这次考试答错要扣分【不要问我为什么】，所以得分有可能为负。

输入格式

第一行为n，k，第二行为第1~n位同学的得分。

输出格式

一个数s，为最大得分和。

输入输出样例

输入 #1

5 3
1 -1 1 -1 1

输出 #1

3

说明/提示

2014彭鲲志：“题目这么短一看就很水。”

Code:

#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;
}