week10 C - 拿数问题 II

题目：
YJQ 上完第10周的程序设计思维与实践后，想到一个绝妙的主意，他对拿数问题做了一点小修改，使得这道题变成了 拿数问题 II。

给一个序列，里边有 n 个数，每一步能拿走一个数，比如拿第 i 个数， Ai = x，得到相应的分数 x，但拿掉这个 Ai 后，x+1 和 x-1 (如果有 Aj = x+1 或 Aj = x-1 存在) 就会变得不可拿（但是有 Aj = x 的话可以继续拿这个 x）。求最大分数。

Input

第一行包含一个整数 n (1 ≤ n ≤ 105)，表示数字里的元素的个数
第二行包含n个整数a1, a2, …, an (1 ≤ ai ≤ 105

Output

输出一个整数：n你能得到最大分值。

Example
Input
2
1 2
Output
2
Input
3
1 2 3
Output
4
Input
9
1 2 1 3 2 2 2 2 3
Output
10

代码解析：
开一个a[100010]的int数组，初始化为0，每次输入
ai的时候，a[ai]++，把问题简化，这样当我们取a[i]的时候，a[i-1]和a[i+1]就变得不可取，
假设dp[i]是当取到a[i]时所得分数的最大值。
求出状态转移方程dp[i]=max(dp[i-1],dp[i-2]+(long long)i*a[i]);

代码很简洁：

#include<iostream>
#include<cmath>
#include<cstring>
using namespace std;
int main()
{
 int n;
 int a[100001];
 long long dp[100001];
 cin>>n;
 memset(a,0,sizeof(a));
 memset(dp,0,sizeof(dp));
 for(int i=0;i<n;i++){
  int temp;
  cin>>temp;
  a[temp]++; 
 }
 dp[1]=a[1];
 for(int i=2;i<=100000;i++){
  dp[i]=max(dp[i-1],dp[i-2]+(long long)i*a[i]);
 }
 cout<<dp[100000];
 return 0;
}