codeforces455——A. Boredom(线性DP)
原题链接
题意:
思路:
记录每个数的出现次数为a[i]。
考虑dp。
dp[i][j]表示选到i并且i的状态为j的最大价值。
当j==0时,表示不选i,此时i-1选不选都可以。
当j ==1时,表示选i,这时候i-1必须不选,选i的贡献是i*a[i]。
遍历到1e5就好了。
也可以将第二维去掉。
代码:
#pragma GCC optimize(3)
#pragma GCC optimize("Ofast","unroll-loops","omit-frame-pointer","inline")
#pragma GCC optimize(2)
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll,ll>PLL;
typedef pair<int,int>PII;
typedef pair<double,double>PDD;
#define I_int ll
inline ll read()
{
ll x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
char F[200];
inline void out(I_int x) {
if (x == 0) return (void) (putchar('0'));
I_int tmp = x > 0 ? x : -x;
if (x < 0) putchar('-');
int cnt = 0;
while (tmp > 0) {
F[cnt++] = tmp % 10 + '0';
tmp /= 10;
}
while (cnt > 0) putchar(F[--cnt]);
//cout<<" ";
}
ll ksm(ll a,ll b,ll p){ll res=1;while(b){if(b&1)res=res*a%p;a=a*a%p;b>>=1;}return res;}
const ll inf = 0x3f3f3f3f3f3f3f3f;
const int maxn=1e5+100,N=1e6+100;
const double PI = atan(1.0)*4;
const double eps=1e-6;
ll n,a[maxn],dp[maxn][2];
int main(){
n=read();
for(int i=1;i<=n;i++){
int x=read();
a[x]++;
}
ll res=-1;
for(int i=1;i<=100000;i++){
dp[i][0]=max(dp[i-1][0],dp[i-1][1]);
dp[i][1]=max(dp[i][1],dp[i-1][0]+i*a[i]);
res=max(res,max(dp[i][1],dp[i][0]));
}
out(res);
return 0;
}