程序设计思维 week10 作业C-拿数问题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 ≤ 10e5)

Output

输出一个整数：n（你能得到的最大分值）。

Sample Input

1 2 1 3 2 2 2 2 3

Sample Output

10

思路

动态规划。
使用sum[i]记录数字 i 出现的次数。dp[i]为考虑sum[1…i]时能拿到的最大分数。初始化dp[1]=sum[1]。转移方程为：dp[i] = max(dp[i-1], dp[i-2]+A[i]) 。答案为dp[Max]，其中Max为读入的数字中最大的。

代码

#include <cstdio>
#include <algorithm>
using namespace std;
const int maxn=1e5+5;
long long sum[maxn];
long long dp[maxn];

int main(){
    long long n,Max=0,a;
    scanf("%lld",&n);
    for(int i=0;i<n;i++){
        scanf("%lld",&a);
        sum[a]++;
        Max=max(Max,a);
    }
    dp[1]=sum[1];
    for(long long i=2;i<=Max;i++)
        dp[i]=max(dp[i-1],dp[i-2]+sum[i]*i);
    printf("%lld",dp[Max]);
    return 0;
}

总结

一开始sum[]声明成了int，结果在test42 wa。后来就全改成了long long。
题目链接