第十周作业——拿数问题 II

拿数问题 II

一、题目
　　YJQ 上完第10周的程序设计思维与实践后，想到一个绝妙的主意，他对拿数问题做了一点小修改，使得这道题变成了 拿数问题 II。
　　给一个序列，里边有 n 个数，每一步能拿走一个数，比如拿第 i 个数， Ai = x，得到相应的分数 x，但拿掉这个 Ai 后，x+1 和 x-1 (如果有 Aj = x+1 或 Aj = x-1 存在) 就会变得不可拿（但是有 Aj = x 的话可以继续拿这个 x）。求最大分数。
　　
二、输入
　　第一行包含一个整数 n (1 ≤ n ≤ 105)，表示数字里的元素的个数
　　第二行包含n个整数a1, a2, …, an (1 ≤ ai ≤ 105)
　　
三、输出
　　输出一个整数：n你能得到最大分值。
　　
四、样例输入输出
Input

9
1 2 1 3 2 2 2 2 3

Output

10

样例解释
　　对于第三个样例：先选任何一个值为2的元素，最后数组内剩下4个2。然后4次选择2，最终得到10分。
　　
五、解题思路
　　动态规划。
　　首先对数据进行处理。将所有相等的数据合并起来，存入a数组。例如共有4个2，则a[2]=8。即可将问题转为变化前的拿数问题。
　　转移方程：dp[i]=max{dp[i-1],dp[i-2]+a[i]},即不选择这个数和选择这个数的分数中取较大的。
　　
六、样例代码

#include<iostream>
using namespace std;
int n;
long long int maxi=0;
long long int a[100050];
long long int dp[100050];

void solve(long long int x)
{
	dp[0]=0;
	dp[1]=a[1];
	for(long long int i=2;i<=x;i++)
	{
		dp[i]=max(dp[i-1],dp[i-2]+a[i]);
	}
}

int main()
{
	cin>>n;
	long long int x;
	for(int i=0;i<100050;i++) //初始化 
	{
		a[i]=0;
	}
	
	for(int i=1;i<=n;i++)
	{
		cin>>x; 
		a[x]+=x;
		maxi=max(maxi,x);
	} 
//	for(int i=1;i<=maxi;i++) cout<<a[i]<<endl;
	solve(maxi);
//	for(int i=1;i<=maxi;i++) cout<<dp[i]<<" ";
    cout<<dp[maxi]<<endl;
}