【2021届百度春季实习笔试题】取数

题目描述

一行数字a1，a2，a3，……，an，有M个回合，每个回合必须从中选择一个数字擦去它，接着每个回合后剩下来的每个数字ai都要递减一个相应的值bi。如此重复M个回合，所有你擦去的数字之和就是你所得的分数。
给出n，m，a数列和b数列，求所得的最多的分数。

Input

输入文件的第一行是一个整数n（1<=n<=100），表示数字个数
第二行一个整数m（1<=m<=n），表示回合数
接下来一行有n个不超过500的正整数，a1，a2，a3，……，an表示原始序列，最后一行有n个不超过500的正整数，b1，b2，b3，……，bn，表示每回合每个数字递减的值。

Output

输出文件只有一个整数，表示最大的可能得分。

Sample Input

3
3
10 20 30
4 5 6

Sample Output

47

思路

先确定取数字的顺序，然后dp。

代码

#include <iostream>
#include <iomanip>
#include <algorithm>
#include <cstdio>
#include <vector>
using namespace std;

struct Node{
    int a;
    int b;
};

static bool cmp(Node A, Node B) {
    return A.b > B.b;
}

int main() {
    int n, m;
    while(cin >> n >> m) {
        Node nodes[105];
        for (int i=0; i<n; ++i) {
            cin >> nodes[i].a;
        }
        for (int i=0; i<m; ++i) {
            cin >> nodes[i].b;
        }

        sort(nodes, nodes+n, cmp);
        vector<vector<int> > dp(n+1, vector<int>(m+1, 0));

        for (int i=1; i<=n; ++i) {
            for (int j=1; j<=m; ++j) {
                dp[i][j] = max(dp[i-1][j], dp[i-1][j-1]+nodes[i-1].a-nodes[i-1].b*(j-1));
            }
        }

        cout << dp[n][m] << endl;
    }
    return 0;
}

/*
5
1
10 20 30 40 50
9 5 6 7 8
*/

自闭了。。。笔试的时候愣是想不到n和m不相等的情况。。。
枯了。。。