455/A. Boredom

codeforces百题计划第一周(2)

线性dp，记录每个数次出现的次数b[i]，dp[i]的状态是处理完前i个数字后的最大值。

转移方程：dp[i]=max(dp[i-1],dp[i-2]*i*b[i])

注意会爆int

#include <bits/stdc++.h>
using namespace std;

const int maxn=1e5+5;
long long  a[maxn];
long long  b[maxn];
long long  dp[maxn]; 
int main(){
    int n;
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        b[a[i]]++;
    }
    //操作了前i后的最好结果 
    dp[1]=b[1];
    //cout <<b[100000]<<endl;
    for(int i=2;i<=100000;i++)
    {
        dp[i]=max(dp[i-1],dp[i-2]+(i*b[i])); 
        //cout <<dp[i]<<endl;
    }
    cout <<dp[100000]<<endl;
    
    return 0;
}