问题描述 :
和谐数组是指一个数组(长度>=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
//利用vector对map进行排序
#include<iostream>
#include<vector>
#include<map>
#include<algorithm>
using namespace std;
bool cmp(pair<int,int> a,pair<int,int> b){
return a.first<b.first;
}
int findLHS(vector<int> &nums){
map<int,int> m;
for(int i=0;i<nums.size();i++){
m[nums[i]]++;
}
vector<pair<int,int>> v(m.begin(),m.end());
sort(v.begin(),v.end(),cmp);
int res=0;
for(vector<pair<int,int>>::iterator it=v.begin()+1;it!=v.end();it++){
if(abs((it-1)->first-it->first)==1){
res=max(res,(it-1)->second+it->second);
}
}
return res;
}
int main(){
int n,num;
cin>>n;
vector<int> nums;
for(int i=0;i<n;i++){
cin>>num;
nums.push_back(num);
}
int res=findLHS(nums);
cout<<res;
return 0;
}