WEEK10 作业 C - 拿数问题 II

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 ≤ 10⁵)，表示数字里的元素的个数
第二行包含n个整数a1, a2, …, an (1 ≤ ai ≤ 10⁵)

Output

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

Sample Input 1

2 
1 2

Sample Output 1

2

Sample Input 2

3 
1 2 3

Sample Output 2

4

Sample Input 3

9 
1 2 1 3 2 2 2 2 3

Sample Output 3

10

Hint

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

题解

这道题是动态规划的应用。
课上的题目是第i个数取了，第i+1和第i-1个数就不能取。这道题的区别是大小为i的数取了，大小为i+1和i-1的数就不能取。（且值最大为500000，a数组不会占用很多内存）所以我们把数的值的大小作为数组的下标，即a[i]表示值为i的数有多少个。score数组表示的是前i个数中能拿到的最大分数（第i个数不一定拿了）。
初始化时，score数组全都赋值为0，然后把score[mn]=mnscorep[mn]。（方便后面使用，后面循环从mn+1开始）。
状态转移方程为score[i]=max(score[i-1],score[i-2]+ia[i])。
一开始做的时候，没有记录mn和mx，直接从1循环到100000，但是后来发现还是有mn和mx比较方便，循环的时候极大地减少时间。
注意，编译器选择Microsoft Visual C++时，若代码中出现了scanf()，则需要在前面加上#define _CRT_SECURE_NO_WARNINGS 1，否则会编译出错。
每次做题都要注意数据范围，比如本次就要开long long。

代码

#define _CRT_SECURE_NO_WARNINGS 1
#include <iostream>
#include <algorithm>
#include <stdio.h>
#include <string.h>
using namespace std;
int n,mn=100050,mx=0;
long long a[100020],score[100020];
int main(int argc, char** argv) {
	scanf("%d",&n);
	memset(a,0,sizeof(a));
	for(int i=1;i<=n;i++){
		int in;
		scanf("%d",&in);
		a[in]++;
		mx=max(mx,in);
		mn=min(mn,in);
	}
	score[mn]=mn*a[mn];
	for(int i=mn+1;i<=mx;i++){
		score[i]=max(score[i-1],score[i-2]+i*a[i]);
	}
	cout<<score[mx]<<endl;
	return 0;
}
//状态转移方程！！！ 
//注意数据范围 
//记录下最小值最大值