40 lines
823 B
Ruby
40 lines
823 B
Ruby
|
# This file contains the Ruby code from book of
|
||
|
|
# "Data Structures and Algorithms
|
||
|
|
# with Object-Oriented Design Patterns in Ruby"
|
||
|
|
# by Bruno R. Preiss.
|
||
|
|
#
|
||
|
|
# Copyright (c) 2004 by Bruno R. Preiss, P.Eng. All rights reserved.
|
||
|
|
|
||
|
|
class DenseMatrix < Matrix
|
||
|
|
|
||
|
|
def initialize(rows, cols)
|
||
|
|
super
|
||
|
|
@array = MultiDimensionalArray.new(rows, cols)
|
||
|
|
end
|
||
|
|
|
||
|
|
def [](i, j)
|
||
|
|
@array[i,j]
|
||
|
|
end
|
||
|
|
|
||
|
|
def []=(i, j, value)
|
||
|
|
@array[i,j] = value
|
||
|
|
end
|
||
|
|
|
||
|
|
def *(mat)
|
||
|
|
assert { numberOfColumns == mat.numberOfRows }
|
||
|
|
result = DenseMatrix.new(
|
||
|
|
numberOfRows, mat.numberOfColumns)
|
||
|
|
for i in 0 ... numberOfRows
|
||
|
|
for j in 0 ... mat.numberOfColumns
|
||
|
|
sum = 0
|
||
|
|
for k in 0 ... numberOfColumns
|
||
|
|
sum += self[i,k] * mat[k,j]
|
||
|
|
end
|
||
|
|
result[i,j] = sum
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
result
|
||
|
|
end
|
||
|
|
end
|