Codeforces Round 260 (Div. 1)A. Boredom(dp)
最开始写了一发贪心wa了,然后这种选和不选的组合优化问题,一般是考虑动态规划
d
p
[
i
]
[
0
]
:
dp[i][0]:
dp[i][0]:表示第i个数不选的最大值
d
p
[
i
]
[
1
]
:
dp[i][1]:
dp[i][1]:表示第i个数选的最大值
考虑转移:
d
p
[
i
]
[
0
]
=
m
a
x
(
d
p
[
i
−
1
]
[
1
]
,
d
p
[
i
−
1
]
[
0
]
)
dp[i][0]=max(dp[i-1][1],dp[i-1][0])
dp[i][0]=max(dp[i−1][1],dp[i−1][0])
d
p
[
i
]
[
1
]
=
d
p
[
i
−
1
]
[
1
]
+
a
[
i
]
∗
i
dp[i][1]=dp[i-1][1]+a[i] * i
dp[i][1]=dp[i−1][1]+a[i]∗i
需要将每一个数用一个桶统计次数
因为n比较小。
最后答案在
d
p
[
n
]
[
0
]
和
d
p
[
n
]
[
1
]
dp[n][0]和dp[n][1]
dp[n][0]和dp[n][1]两者中取一个最大值即可
#include <bits/stdc++.h>
#define int long long
#define rep(i,a,b) for(int i = (a); i <= (b); ++i)
#define fep(i,a,b) for(int i = (a); i >= (b); --i)
#define pii pair<int, int>
#define pll pair<long long, long long>
#define ll long long
#define db double
#define endl '\n'
#define x first
#define y second
#define pb push_back
using namespace std;
const int N=1e6+10,mod=1e9+7;
int dp[N][2];
int a[N];
void solve()
{
int n;cin>>n;
int m=0;
rep(i,1,n){
int x;cin>>x;
a[x]++;
m=max(m,x);
}
rep(i,1,m){
dp[i][0]=max(dp[i-1][0],dp[i-1][1]);
dp[i][1]=dp[i-1][0]+a[i]*i;
}
cout<<max(dp[m][0],dp[m][1])<<endl;
}
signed main(){
ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
// freopen("1.in", "r", stdin);
int _;
// cin>>_;
// while(_--)
solve();
return 0;
}