挑选(pick)

1s/128MB

【题目背景】

NOIP2017 马上就要到了,丁爷爷想要从他的小朋友里挑选出一些厉害的来参加NOIP。

 【题目描述】

丁爷爷共有 n 个小朋友,按编号 1 . . . n 从左到右排成一行。每个小朋友都有一个实力值,第 i 个小朋友的实力值为 wi。

丁爷爷作为一名爷爷,自然可以随意地挑选小朋友。但是如果他挑选的小朋友太靠  近,就免不了给人一种内定、钦点的感觉了。他并不想这么做,于是他制定出了一个  方案。

他给每个小朋友设置了一个气场值 ci,当丁爷爷挑选出 i 号小朋友参加 NOIP 后, 他会把它右边的ci个小朋友(不包被挑选的小朋友本身)都自动淘汰。这些被淘汰的小朋友加上被丁爷爷挑选出的小朋友立即被移出队伍。接着,丁爷爷就可以继续他的挑选工作。需要注意的是,如果此时右边的小朋友个数不足ci,那么丁爷爷是不能进行这次挑选的。

作为丁爷爷的粉丝,你和 Yazid 都对丁爷爷的挑选工作很感兴趣。

你想知道丁爷爷小朋友们最高的实力值总和,而 Yazid则对挑选小朋友的本质不同的方案数很感兴趣。(两种挑选方案本质相同,当且仅当两种方案最终挑选出的小朋友完全一致。)

众所周知,Yazid 是一名辣鸡蒟蒻,于是你就需要同时求解这两个问题。由于方案数可能很大,所以对于 Yazid 的问题,你只需要回答对 998244353 取模的结果即可。

 【输入格式】

从文件 pick.in 中读入数据。

第一行一个正整数 n,表示小朋友的数目。

接下来一行 n 个用空格隔开的非负整数 w1 . . . wn,依次描述 1 号小朋友至n号小朋友的实力值。

接下来一行 n 个用空格隔开的非负整数 c1 . . . cn,依次描述 1 号小朋友至n号小朋友的气场值。

 【输出格式】

输出到文件 pick.out 中。

输出一行 2 个用空格隔开的整数,第一个整数表示挑选出的小朋友实力值总和的最大值,第二个整数表示本质不同的挑选方案数对 998244353 取模的结果。

【样例 1 输入】

3

7 2 3

2 0 0

【样例 1 输出】

7 5

【样例 2 输入】

18

27 68 75 76 75 75 68 40 33 62 58 88 4 75 766 84 52

1 2 3 3 2 4 1 2 3 3 5 1 3 1 1 3 2 1

【样例 2 输出】

490 3348

【样例 3】

见选手目录下的 pick/pick3.in 与 pick/pick3.ans。

 【提示】

样例 1 解释:你可以挑选出 {}, {1}, {2},{3},{2, 3} 这些小朋友的子集,因此方案数为

5。其中,{1} 这个子集的小朋友实力值总和最大,总和为 7(只有 1 号小朋友)。

 【子任务】

对于10% 的数据,保证 n ≤ 9。对于 30% 的数据,保证 n ≤ 18。对于 60% 的数据,保证 n ≤ 200。

对于另外20% 的数据,保证所有的 wi= 1,保证所有的 ci= 1。对于 95% 的数据,保证n≤ 3, 000。

对于 100% 的数据,保证 n ≤ 5, 000,wi ≤ 10^7,ci <n。

由这个题意似乎可以往一种常见的dp去想那就是---括号序列!
这里我们显然可以转化成若第i个填左括号,那么就要有ci个右括号来与之匹配。
那不就非常简单了吗?!
dp[i][j]表示前i个小朋友中左括号与右括号的差为j的最大实力值。
则dp[i][j]=max{dp[i-1][j+1](取右括号情况),dp[i-1][j-c[i]]+wi}
if j==0 dp[i][j]=max{dp[i][j],dp[i-1][j]}可以不取。
辣么最后的结果就是dp[n][0]啦~
初值dp[0][0]=0就ok啦。
g[i][j]表示前i个小朋友中左括号与右括号的差为j的方案总数。
也是一样的道理,将max改成累加即可。
这道题的变形有点考思路,但变形好之后就很基本了qwq下次要记住啊...

code

#include <cstdio>
#include <algorithm>
#define ll long long
using namespace std;
const int mod = 998244353;
const int MAXN = 5005;

inline ll read() {
    ll k = 0, f = 1;
    char ch = getchar();
    while (ch < '0' || ch > '9') {
        if (ch == '-')
            f = -1;
        ch = getchar();
    }
    while (ch >= '0' && ch <= '9') {
        k = (k << 1) + (k << 3) + (ch & 15);
        ch = getchar();
    }
    if (f == -1) k = ~k + 1;
    return k;
}

inline void write(ll x) {
    if (x < 0)
        x = -x, putchar('-');
    if (x > 9)
        write(x / 10);
    putchar(x % 10 + '0');
}

int n;
int now;
ll w[MAXN],c[MAXN];
ll dp[2][MAXN],g[2][MAXN];
int main(){
    freopen("pick.in","r",stdin);
    freopen("pick.out","w",stdout);
    //n = read();
    scanf("%d",&n);

    for (int i = 1 ; i <= n ; i++){
        //w[i] = read();
        scanf("%d",&w[i]);
    }
    for (int i = 1 ; i <= n ; i++){
        //c[i] = read();
        scanf("%d",&c[i]);
    }

    g[0][0] = 1;

    for (int i = 1 ; i <= n ; i++){
        now = 1 - now;
        for (int j = 0 ; j <= n ; j++){
            if (j == 0){

                dp[now][j] = max(dp[1 - now][j],dp[1 - now][j + 1]);
                if(j >= c[i]) dp[now][j] = max(dp[now][j],dp[1 - now][j - c[i]]+  w[i]);

                g[now][j] = (g[1 - now][j] + g[1 - now][j + 1]) % mod;
                if(j >= c[i]) g[now][j] = (g[now][j] + g[1 - now][j - c[i]]) % mod;

            }

            else{

                dp[now][j] = dp[1 - now][j + 1];
                if(j >= c[i]) dp[now][j] = max(dp[now][j],dp[1 - now][j - c[i]] + w[i]);

                g[now][j] = g[1 - now][j + 1];
                if(j >= c[i]) g[now][j] = (g[now][j] + g[1 - now][j - c[i]]) % mod;
            }
        }
    }

    //write(dp[now][0]),putchar(32),write(g[now][0]);
    printf("%lld %lld",dp[now][0],g[now][0]);
    return 0;
}
12-25 17:43