LeetCode 5214. 最長定差子序列（Java）HashMap

題目： 5214. 最長定差子序列

給你一個整數數組 arr 和一個整數 difference，請你找出 arr 中全部相鄰元素之間的差等於給定 difference 的等差子序列，並返回其中最長的等差子序列的長度。

示例 1：
輸入：arr = [1,2,3,4], difference = 1
輸出：4
解釋：最長的等差子序列是 [1,2,3,4]。

示例 2：
輸入：arr = [1,3,5,7], difference = 1
輸出：1
解釋：最長的等差子序列是任意單個元素。

示例 3：
輸入：arr = [1,5,7,8,5,3,4,2,1], difference = -2
輸出：4
解釋：最長的等差子序列是 [7,5,3,1]。

提示：

• 1 <= arr.length <= 10^5
• -10^4 <= arr[i], difference <= 10^4

題解：

Map<key, value> 保存以 key 結束的序列的最大長度。

遍歷數組 arr，數 k 的前驅結點爲 pre = k - difference，所以以 k 結束的序列的長度加 1，更新最大值（返回值），並更新 map 中 key = k 時的 value。

時間複雜度： $O(n)$，不考慮 HashMap 中查找的時間複雜度
空間複雜度： 不肯定

Java：

class Solution {
    public int longestSubsequence(	int[] arr,
                                    int difference) {
        Map<Integer, Integer> map = new HashMap<>();
        int ret = 0;
        for (int k : arr) {
            int pre = k - difference;// 前驅
            // 序列增長數k，所以長度加1
            int len = map.getOrDefault(pre, 0) + 1;
            ret = Math.max(ret, len);
            map.put(k, len);// 更新map
        }
        return ret;
    }
}