LeeCode:最长和谐子序列（C++）

问题描述

和谐数组是指一个数组（长度>=2）里元素的最大值和最小值之间的差别正好是1。

现在，给定一个整数数组，你需要在所有可能的子序列中找到最长的和谐子序列的长度。
如果找不到这样的数组，则输出0。

示例 :

输入: [1,3,2,2,5,2,3,7]

输出: 5

原因: 最长的和谐数组是：[3,2,2,2,3].

说明: 输入的数组长度最大不超过20,000.

输入说明

首先输入整数数组的长度n，2<=n<=10000

然后输入n个整数，以空格分隔。

8
1 3 2 2 5 2 3 7

输出说明

输出最长的和谐子序列的长度

5

代码思路

和谐数组中最大值和最小值之间的差别正好为1，因此数组中所包含的元素在数值上只能是相邻的，可以用map统计数值出现的次数并按照key值的大小进行排序；然后遍历map，若两个key之间的差为一，则将他们出现的次数相加，即为和谐数组的长度，在下一次遍历时若次数之和大于现在的长度则更新结果。

#include<iostream>
#include<string>
#include<map>
#include<vector>
using namespace std;

int findLHS(vector<int>& nums){
	map<int,int> cnt;
	for(auto num:nums)
		cnt[num]++;
	auto pre=cnt.begin();
	auto it=cnt.begin();
	int res=0;
	for(it++;it!=cnt.end();it++){
		if(it->first-pre->first==1)
			res=max(it->second+pre->second,res);
		pre++;
	}
	return res;
}
int main(){
	int n,data;
	vector<int> nums;
	cin>>n;
	for(int i=0;i<n;i++){
		cin>>data;
		nums.push_back(data);
	}
	int res=findLHS(nums);
	cout<<res<<endl;
	return 0;
}