LU decomposition: Difference between revisions

m
m (→‎{{header|zkl}}: change formating)
m (→‎{{header|Sidef}}: updated code)
Line 2,898:
<lang ruby>func is_square(m) { m.all { .len == m.len } }
func matrix_zero(n, m=n) { m.of { n.of(0) } }
func matrix_ident(n) { n.of {|i| [(i-1)n.of(0)..., {|j| i==j ? 1, (n-i).of(: 0)...] } } }
 
 
func pivotize(m) {
var size = m.len
var id = matrix_ident(size)
for i in (^size) {
var max = m[i][i]
var row = i
for j in range(i, .. size-1) {
if (m[j][i] > max) {
max = m[j][i]
Line 2,912:
}
}
if (row  != i) {
id.swap(row, i)
}
Line 2,918:
return id
}
 
 
func mmult(a, b) {
var p = []
for r,c in (^a ~X ^b[0]) {
for i in (^b) {
p[r][c]  := 0 += (a[r][i] * b[i][c])
}
}
return p
}
 
 
func lu(a) {
is_square(a) || die "Defined only for square matrices!";
Line 2,936:
var L = matrix_ident(n)
var U = matrix_zero(n)
for i,j in (^n ~X ^n) {
if (j >= i) {
U[i][j] = (Aʼ[i][j] - (^i->map { U[_][j] * L[i][_] }.summap(0^i).sum))
} else {
L[i][j] = ((Aʼ[i][j] - (^j->map { U[_][j] * L[i][_] }.summap(0^j).sum)) / U[j][j])
}
}
return [P, Aʼ, L, U]
}
 
 
func say_it(message, array) {
say "\n#{message}"
Line 2,952:
}
}
 
 
var t = [[
%n(1 3 5),
Line 2,963:
%n( 2 5 7 1),
]]
 
 
for test (t) {
t.each { |test|
say_it('A Matrix', test);
for a in,b (['P Matrix', 'Aʼ Matrix', 'L Matrix', 'U Matrix'] ~Z lu(test)) {
say_it(a[0], a[1]b)
}
}</lang>
2,747

edits