程序设计第十周 C - 拿数问题 II

问题描述

YJQ 上完第10周的程序设计思维与实践后，想到一个绝妙的主意，他对拿数问题做了一点小修改，使得这道题变成了 拿数问题 II。

给一个序列，里边有 n 个数，每一步能拿走一个数，比如拿第 i 个数， Ai = x，得到相应的分数 x，但拿掉这个 Ai 后，x+1 和 x-1 (如果有 Aj = x+1 或 Aj = x-1 存在) 就会变得不可拿（但是有 Aj = x 的话可以继续拿这个 x）。求最大分数。

本题和课上讲的有些许不一样，但是核心是一样，需要你自己思考。
Input
第一行包含一个整数 n (1 ≤ n ≤ 105)，表示数字里的元素的个数第二行包含n个整数a1, a2, …, an (1 ≤ ai ≤ 105)
Output
输出一个整数：n你能得到最大分值。
Example
Input

2
1 2

Output

2

Input

3
1 2 3

Output

4

Input

9
1 2 1 3 2 2 2 2 3

Output

10

想法

课上学习了对于一个数组但，拿掉这个A[i] 后，A[i+1] 和A[i-1] 就会变得不可拿给,这样就可以 定义dp[i] 为仅考虑A[1…i] ，能拿到的最大分数
⚫ 状态转移方程为 dp[i] = max(dp[i-1], dp[i-2]+A[i])
现在可以将这个问题转化为拿相邻数的问题，我们可以将数值转换为数组的索引，这样就转换为了所学问题，因为可以重复选数，在转换索引时我们直接令 nu[t]+=t;
更新完所有的状态后，输出最大值就是所求答案。

代码

    #include <iostream>
    #include<cstdio>
    #include<algorithm>
    #define ll long long
    using namespace std;
    ll vis[100100] = {0},nu[100100] = {0},ma;
    int n,le=100100,ri,t;
    int pre[100100];
    int main()
    {
        cin >> n;
        for (int i = 1; i <=n; i++)
        {
            scanf("%d", &t);
            nu[t]+=t;
            le=min(le,t);
            ri=max(ri,t);
        }
        //1-1e5
        vis[le]=nu[le];
        pre[le] = 0; 
         for (int i = le + 1; i <= ri; i++)
        {
            vis[i]=max(vis[i-1],vis[i-2]+nu[i]);
            // dp[i] =  max(dp[i-1], dp[i-2]+A[i])  
        }
        for (int i = le; i <= ri; i++)
        {
            ma = ma > vis[i] ? ma : vis[i];
        }
        cout << ma << endl;
       // system("pause");
        return 0;
    }