- Notifications
You must be signed in to change notification settings - Fork 4
/
Copy path1027-longest-arithmetic-subsequence.rb
69 lines (54 loc) · 1.7 KB
/
1027-longest-arithmetic-subsequence.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
# frozen_string_literal: true
# 1027. Longest Arithmetic Subsequence
# Medium
# https://leetcode.com/problems/longest-arithmetic-subsequence
=begin
Given an array nums of integers, return the length of the longest arithmetic subsequence in nums.
Note that:
* A subsequence is an array that can be derived from another array by deleting some or no elements without changing the order of the remaining elements.
* A sequence seq is arithmetic if seq[i + 1] - seq[i] are all the same value (for 0 <= i < seq.length - 1).
Example 1:
Input: nums = [3,6,9,12]
Output: 4
Explanation: The whole array is an arithmetic sequence with steps of length = 3.
Example 2:
Input: nums = [9,4,7,2,10]
Output: 3
Explanation: The longest arithmetic subsequence is [4,7,10].
Example 3:
Input: nums = [20,1,15,3,10,5,8]
Output: 4
Explanation: The longest arithmetic subsequence is [20,15,10,5].
Constraints:
* 2 <= nums.length <= 1000
* 0 <= nums[i] <= 500
=end
# @param {Integer[]} nums
# @return {Integer}
deflongest_arith_seq_length(arr)
tab=Array.new(arr.length){ |h,k| Hash.new(1)}
(1...arr.length).eachdo |i|
(0...i).eachdo |j|
a,b=arr[i],arr[j]
diff=b - a
tab[i][diff]=tab[j][diff] + 1
end
end
max=0
tab.eachdo |hash|
max_val=hash.values.max || 0
max=max_valifmax_val > max
end
max
end
# ********************#
# TEST #
# ********************#
require"test/unit"
classTest_longest_arith_seq_length < Test::Unit::TestCase
deftest_
assert_equal4,longest_arith_seq_length([3,6,9,12])
assert_equal3,longest_arith_seq_length([9,4,7,2,10])
assert_equal4,longest_arith_seq_length([20,1,15,3,10,5,8])
end
end