Codeforces 455A Boredom 取數位dp

標籤：http   os   io   for   ar   amp   size   type   

題目連結：點擊開啟連結

給定一個n長的序列

刪除x這個數就能獲得x * x的個數 的分數，然後x+1和x-1這2個數會消失，即無法獲得這2個數的分數

問最高得分。

先統計每個數出現的次數，然後dp一下，對於每個數只有取或不取2種狀態。

#include <algorithm>#include <cctype>#include <cassert>#include <cstdio>#include <cstring>#include <climits>#include <vector>#include<iostream>#include <queue>using namespace std;#define ll long longll hehe;#define N 100005ll num[N], n;ll dp[N][2];void work(){for(ll i = 1; i < N; i++){dp[i][0] = max(dp[i-1][0], dp[i-1][1]);if(num[i]) {if(i-2>=0)dp[i][1] = max(dp[i-2][0], dp[i-2][1])+num[i]*i;dp[i][1] = max(dp[i][1], dp[i-1][0] + num[i]*i);}}cout<<max(dp[N-1][0], dp[N-1][1])<<endl;}int main(){  ll i, j, u;while(cin>>n){memset(num, 0, sizeof num);memset(dp, 0, sizeof dp);for(i = 1; i <= n; i++){scanf("%I64d", &u);num[u] ++;}work();}    return 0;}