Week10 作业 C - 拿数问题 II CodeForces - 455A

题目描述

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

Hint
对于第三个样例：先选任何一个值为2的元素，最后数组内剩下4个2。然后4次选择2，最终得到10分。

解题思路

给定 n 个数，每一步能拿走一个数，拿第 i 个数，就能得到相应的分数 Ai，但拿掉这个Ai 后，A(i+1) 和A(i-1) 就会变得不可拿，但A(i)仍可拿。我们稍微转换一下 ，令 A[i] 中存储的是值为i的元素的总的分数，再按拿数问题列转移方程即可。

因此，我们在判断是否拿走第i个数时，我们有两种选择，一是拿走dp[i-2]，那么我们最终的分数dp[i]=dp[i-2]+A[i]；一个是拿走dp[i-1]，那么本次就不能拿数了，dp[i]=dp[i-1]。所以，我们的状态转移方程为 dp[i] = max(dp[i-1], dp[i-2]+A[i]) 。最后的结果就是 max{ dp[i] }。

程序源码

#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
using namespace std;


long long A[100005] = { 0 }; // A[i] 中存储的是值为i的元素的总的分数
long long dp[100005] = { 0 };

int main()
{
	std::ios::sync_with_stdio(false);
	int n;
	cin >> n;
	for (int i = 1; i <= n; i++) { 
		int g;
		cin >> g;
		A[g] += g; //值为g的元素的总分加上g
	}

	
	dp[1] = A[1]; //拿数问题
	long long rs = A[1];
	for (int i = 2; i <= 100000; i++) { // dp[i] =  max(dp[i-1], dp[i-2]+A[i]) 
		long long rs1 = dp[i - 2] + A[i];
		if (dp[i - 1] > rs1) rs1 = dp[i - 1];
		dp[i] = rs1;
		if (rs1 > rs) rs = rs1;
	}

	cout << rs << endl; //最后的结果是 max{ dp[i] }
	return 0;
}