这是悦乐书的第270次更新，第284篇原创
01 看题和准备
今天介绍的是LeetCode算法题中Easy级别的第136题（顺位题号是594）。我们定义一个和谐数组是一个数组，其最大值和最小值之间的差值恰好为1。给定一个整数数组，在其所有可能的子序列中找到其最长的和谐子序列的长度。例如：
输入：[1,3,2,2,5,2,3,7]
输出：5
说明：最长的和谐子序列是[3,2,2,2,3]。
注意：输入数组的长度不会超过20,000。
本次解题使用的开发工具是eclipse，jdk使用的版本是1.8，环境是win7 64位系统，使用Java语言编写和测试。
02 第一种解法
直接使用两层for循环，依次遍历元素，如果当前元素在数组中不存在与之值相差为1的数，标志flag就为false，那么此元素就不参与计算，反之存在，就把自己出现的次数与其值相差为1的元素的出现次数加起来，并且flag改为true，取最大值即可。
public int findLHS(int[] nums) {
if (nums == null || nums.length < 2) {
return 0;
}
int result = 0;
for(int i=0; i<nums.length; i++){
boolean flag = false;
int count = 0;
for (int j=0; j<nums.length; j++) {
if (nums[i] == nums[j]) {
count++;
} else if (nums[i] == nums[j]-1) {
flag = true;
count++;
}
}
if (flag) {
result = Math.max(result, count);
}
}
return result;
}
03 第二种解法
先对数组进行升序排序，借助Arrays.sort方法来实现，然后遍历数组的元素。
排完序后，可以先将两种特殊情况直接返回结果，如果排序后的数组第一个元素与最后一个元素值相等，直接返回0；如果数组第一个元素与最后一个元素值相差1，返回数组长度。
如果当前元素连续出现，我们需要先将其出现次数记录下来，使用变量preCount来存值，在遇到与其值相差为1的元素时，再将与其值相差为1的元素的出现次数记录下来，加上preCount的值，就表示此和谐数组的长度，再去与所有可能的和谐数组长度做比较取最大值。
如果当前元素不是连续出现，那么preCount的值就为1，表示当前元素只出现了一次，然后判断下一个元素是否与其值相差1，如果下一个元素符合条件，就找到该元素出现的次数，加上preCount的值，两次数之和与最大值取较大者。
public int findLHS2(int[] nums) {
if (nums == null || nums.length < 2) {
return 0;
}
Arrays.sort(nums);
if (nums[0] == nums[nums.length-1]) {
return 0;
}
if (nums[0] == nums[nums.length-1]-1) {
return nums.length;
}
int preCount = 1, result = 0;
for (int i = 0; i < nums.length; i++) {
int count = 1;
if (i > 0 && nums[i] - nums[i - 1] == 1) {
while (i < nums.length - 1 && nums[i] == nums[i + 1]) {
count++;
i++;
}
result = Math.max(result, count + preCount);
preCount = count;
} else {
while (i < nums.length - 1 && nums[i] == nums[i + 1]) {
count++;
i++;
}
preCount = count;
}
}
return result;
}
04 第三种解法
使用HashMap，以数组中的每个元素为key，其出现次数为value，然后遍历map中的每个key，如果map中含有比当前key值大1的key，那么将当前key所对应的value与当前key加1后所对应的value相加，就表示一组和谐数组，在所有可能的和谐数组中找出最大值即可。为了避免溢出，HashMap中的key的类型取Long。
public int findLHS3(int[] nums) {
if (nums == null || nums.length < 2) {
return 0;
}
int result = 0;
Map<Long, Integer> map = new HashMap<Long, Integer>();
for (long n : nums) {
map.put(n, map.getOrDefault(n, 0)+1);
}
for (long key : map.keySet()) {
if (map.containsKey(key+1)) {
result = Math.max(result, map.get(key)+map.get(key+1));
}
}
return result;
}
05 第四种解法
我们可以将上面的解法再精简下，变成一个循环。有一点需要注意，与当前元素能够组成和谐数组的数，可能比自己大1，也可能比自己小1，并且可能还在当前元素的后面出现，所以需要将两种情况都考虑进去。
public int findLHS4(int[] nums) {
if (nums == null || nums.length < 2) {
return 0;
}
HashMap<Long, Integer> map = new HashMap<Long, Integer>();
int result = 0;
for (long n: nums) {
map.put(n, map.getOrDefault(n, 0)+1);
if (map.containsKey(n+1)) {
result = Math.max(result, map.get(n) + map.get(n+1));
}
if (map.containsKey(n-1)) {
result = Math.max(result, map.get(n) + map.get(n-1));
}
}
return result;
}
06 小结
算法专题目前已日更超过四个月，算法题文章137+篇，公众号对话框回复【数据结构与算法】、【算法】、【数据结构】中的任一关键词，获取系列文章合集。
以上就是全部内容，如果大家有什么好的解法思路、建议或者其他问题，可以下方留言交流，点赞、留言、转发就是对我最大的回报和支持！