Week10 作业 C - 拿数问题 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分。

二、思路概述

• 先将存储数字的数组用sort函数排序，然后遍历这个数组，每次碰到一个数字，就记录这个数字的大小、数字的数量和大小的乘积，放进一个新的node型数组mp。
• 设计状态：定义fi 表示从 mp[0]到mp[i]范围内取的和最大的方程；
  初始化：f0 = mp[0].sum；
  转移过程：若第i个数比它前一个数大1，则f[i]=max(f[i - 1], f[i- 2] + mp[i].sum)。否则f[i]=f[i - 1] + mp[i].sum。
  输出答案：f[tol]//f数组的最末一个数值

三、细节

1. 由于最后的输出结果最大值是1e5*1e5=1e10，超出了int的数据范围，所以与之相关的数据类型应该选择long long。
2. 1e5，数组却只开了10050（五位数），所以oj系统显示超时，自己也看不出来，在这里感谢我的助教，帮我看出来这个弱智的错误。

四、完整代码

#include<iostream>
#include<algorithm>
//#include<map>
using namespace std;

struct node{
	long long shu,sum;
}mp[100050];

int main() {
	long long n;
	scanf("%lld",&n) ;
	long long *a =new long long[n];
	for (long long i = 0; i < n; i++) {
		scanf("%lld",&a[i]) ;
	}
	sort(a, a + n);
	
	//map<long long, long long> mp;//first是数的值，second是数的数量
	long long cnt ;
	long long i = 0;
	long long num;
	long long toll=0;
	while (1) {
		cnt = 1;
		num = a[i];
		i++;
		while(a[i] == num&&i<n) {
			i++;
            cnt++;
			//if (i == n)break;
		}
		mp[toll].shu = num;
		//mp[toll].num = cnt;
		mp[toll].sum=num*cnt;
		toll++;
		if (i == n)break;
	}
	
	long long f[100050]; //= new long long[mp.size()];
	
	//auto it1 = mp.begin();auto it2= mp.begin();//it2是后面一个
	long long sum = 0;
	//long long f1 = 0; long long f2 = 0;
	long long tol = 0;
    //long long tol2=0;
	f[0] = mp[0].sum;
	
	while (tol<toll){
		tol++;
		if (mp[tol-1].shu==mp[tol].shu-1) {
			if (tol == 1)f[tol] = max(f[tol - 1],mp[tol].sum);
			else f[tol] = max(f[tol - 1], f[tol - 2] + mp[tol].sum);
		}
		else {
			f[tol] = f[tol - 1] + mp[tol].sum;
		}
	}
	printf("%lld\n",f[tol]);
//	cout << f[tol] << endl;

	return 0;
}