Ruby中帶有兩個循環的兩個和的總和

Question

如果數組中的一對數相加爲零，我需要這兩個數的位置。如果沒有一對數字總和爲零，我應該返回nil。

迭代沒有發生在我的外循環：

def two_sum(nums) 
    # puts(nums) 
    l = nums.length 
    i = 0 
    j = 1 
    while(i < l) 
    while(j < l) 
     puts(nums[i] + nums[j]) 
     puts(nums[i]) 
     puts(nums[j]) 
     if(nums[i] + nums[j] == 0) 
     return ("[" + i.to_s + "," + j.to_s + "]") 
     puts("[" + i.to_s + "," + j.to_s + "]") 
     else 
     j = j + 1 
     end 
    end 
    i = i + 1 
    end 
end 

Answer 1

的問題是，你只在一開始設定的j = 1值，但是你需要它重置爲每個外迭代。只要將j = 1的

while(i < l) 

甚至更​​好以後：j = i + 1

Answer 2

這是更簡單的使用範圍和each;這使得代碼更清晰，更簡潔：

#!/usr/bin/env ruby 

def two_sum(nums) 
    (0...nums.length).each do |i| 
    ((i+1)...nums.length).each do |j| 
     return [i, j] if nums[i] + nums[j] == 0 
    end 
    end 
    nil 
end 

p two_sum([1, 2, 3, -1, 4]) # [0, 3] 
p two_sum([1, 2, 3])   # nil 
p two_sum([])    # nil 

Answer 3

由於✅已經回答了你的問題，我想提出一個替代方案：

def pair_sums_to_zero(arr) 
    h = arr.each_with_index.group_by { |n,_| n.abs }  
    return h[0].first(2).map(&:last) if h.key?(0) and h[0].size > 1 
    a = h.map { |k,v| v.uniq(&:first) }.find { |b| b.size == 2 } 
    a ? a.map(&:last) : nil 
end 

arr = [3,2,-4,-2,3,2] 
pair_sums_to_zero arr 
    #=> [1,3] 

步驟：

h = arr.each_with_index.group_by { |n,_| n.abs } 
    #=> {3=>[[3, 0], [3, 4]], 2=>[[2, 1], [-2, 3], [2, 5]], 4=>[[-4, 2]]} 
    h.key?(0) and h[0].size > 1 
    #=> false 
    c = h.map { |k,v| v.uniq(&:first) } 
    #=> [[[3, 0]], [[2, 1], [-2, 3]], [[-4, 2]]] 
    a = c.find { |b| b.size == 2 } 
    a ? a.map(&:last) : nil 
    #=> [[2, 1], [-2, 3]] 
    a ? a.map(&:last) : nil 
    #=> [1, 3]  

另一個例如：

arr = [3,0,2,-4,-6,0,3,0,2] 
pair_sums_to_zero arr 

h = arr.each_with_index.group_by { |n,_| n.abs }  
    #=> {3=>[[3, 0], [3, 6]], 0=>[[0, 1], [0, 5], [0, 7]], 2=>[[2, 2], [2, 8]], 
    # 4=>[[-4, 3]], 6=>[[-6, 4]]} 
h.key?(0) and h[0].size > 1 
    #=> true 
h[0].first(2).map(&:last) 
    #=> [1, 5] (returned)