594.最长和谐子序列
题目描述
和谐数组是指一个数组里元素的最大值和最小值之间的差别正好是1。
现在,给定一个整数数组,你需要在所有可能的子序列中找到最长的和谐子序列的长度。
示例 1:
输入: [1,3,2,2,5,2,3,7]
输出: 5
原因: 最长的和谐数组是:[3,2,2,2,3].
解题思路
- 用map存储数字及其个数
- 和谐数组中元素差不超过1,所以遍历map中的每个元素,判断map中是否存在其相邻元素。如果有,统计其出现次数,更新子序列长度
class Solution {
public:
int findLHS(vector<int>& nums) {
if(nums.empty()) return 0;
map<int,int>mp;
for(int i=0;i<nums.size();i++){
mp[nums[i]]++;
}
int ans=0;
for(auto it=mp.begin();it!=mp.end();it++){
if(mp.find(it->first+1)!=mp.end()){
ans=max(ans,mp[it->first+1]+mp[it->first]);
}
}
return ans;
}
};