Làm thế nào để vuông mỗi phần tử của một mảng trong lớp Array trong Ruby?

Question

Một phần của mã của tôi là như sau:

class Array 
    def square! 
    self.map {|num| num ** 2} 
    self 
    end 
end 

Khi tôi gọi:

[1,2,3].square! 

tôi mong muốn nhận được [1,4,9], nhưng thay vào đó tôi nhận được [1,2,3 ]. Tại sao điều này là trường hợp? Khi tôi gọi:

[1,2,3].map {|num| num ** 2} 

ngoài phương pháp lớp học, tôi nhận được câu trả lời đúng.

Answer 1

Bạn phải sử dụng Array#map!, không phải Array#map.

Array#map ->Gọi khối đã cho một lần cho mỗi phần tử của tự. Tạo một mảng mới chứa các giá trị được trả về bởi khối.

Array#map! ->Gọi khối đã cho một lần cho mỗi phần tử của bản thân, thay thế phần tử bằng giá trị được khối trả về.

class Array 
    def square! 
    self.map! {|num| num ** 2} 
    end 
end 

[1,2,3].square! #=> [1, 4, 9]