Alex_tz307

Codeforces Boredom(260, div1, A)

Sep 18th, 2020 (edited)
130
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 0.90 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2. #define int long long
  3.  
  4. using namespace std;
  5.  
  6. inline void max_self(int& a, int b) {
  7.     a = max(a, b);
  8. }
  9.  
  10. int32_t main() {
  11.     ios_base::sync_with_stdio(false);
  12.     cin.tie(nullptr);
  13.     cout.tie(nullptr);
  14.     int N;
  15.     cin >> N;
  16.     vector < int > freq(100001);
  17.     for(int i = 0; i < N; ++i) {
  18.         int x;
  19.         cin >> x;
  20.         ++freq[x];
  21.     }
  22.     vector < pair < int , int > > a;
  23.     for(int i = 1; i < 100001; ++i)
  24.         if(freq[i])
  25.             a.emplace_back(i, freq[i]);
  26.     vector < int > dp(a.size());
  27.     for(int i = 0; i < (int)a.size(); ++i) {
  28.         int prev = i - 1;
  29.         if(prev >= 0 && a[prev].first + 1 == a[i].first)
  30.             --prev;
  31.         dp[i] = a[i].first * a[i].second;
  32.         if(prev >= 0)
  33.             dp[i] += dp[prev];
  34.         if(i != 0)
  35.             max_self(dp[i], dp[i - 1]);
  36.     }
  37.     cout << dp[dp.size() - 1];
  38. }
  39.  
Advertisement
Add Comment
Please, Sign In to add comment