can read data from file.
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@4818 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
This commit is contained in:
parent
212d962a3b
commit
87704d6a3c
@ -6,43 +6,66 @@
|
|||||||
# Solves linear equation system(A*x = b) by LU decomposition method.
|
# Solves linear equation system(A*x = b) by LU decomposition method.
|
||||||
# where A is a coefficient matrix,x is an answer vector,b is a constant vector.
|
# where A is a coefficient matrix,x is an answer vector,b is a constant vector.
|
||||||
#
|
#
|
||||||
|
# USAGE:
|
||||||
|
# ruby linear.rb [input file solved]
|
||||||
|
#
|
||||||
|
|
||||||
require "bigdecimal"
|
require "bigdecimal"
|
||||||
require "ludcmp"
|
require "ludcmp"
|
||||||
|
|
||||||
include LUSolve
|
#
|
||||||
|
# NOTE:
|
||||||
|
# Change following BigDecimal::limit() if needed.
|
||||||
|
BigDecimal::limit(100)
|
||||||
|
#
|
||||||
|
|
||||||
def rd_order
|
include LUSolve
|
||||||
printf("Number of equations ?")
|
def rd_order(na)
|
||||||
n = gets().chomp.to_i
|
printf("Number of equations ?") if(na <= 0)
|
||||||
|
n = ARGF.gets().to_i
|
||||||
end
|
end
|
||||||
|
|
||||||
BigDecimal::limit(100)
|
na = ARGV.size
|
||||||
|
|
||||||
zero = BigDecimal::new("0.0")
|
zero = BigDecimal::new("0.0")
|
||||||
one = BigDecimal::new("1.0")
|
one = BigDecimal::new("1.0")
|
||||||
|
|
||||||
while (n=rd_order())>0
|
while (n=rd_order(na))>0
|
||||||
a = []
|
a = []
|
||||||
as= []
|
as= []
|
||||||
b = []
|
b = []
|
||||||
printf("\nEnter coefficient matrix element A[i,j]\n");
|
if na <= 0
|
||||||
for i in 0...n do
|
# Read data from console.
|
||||||
for j in 0...n do
|
printf("\nEnter coefficient matrix element A[i,j]\n");
|
||||||
printf("A[%d,%d]? ",i,j); s = gets
|
for i in 0...n do
|
||||||
a <<=BigDecimal::new(s);
|
for j in 0...n do
|
||||||
as<<=BigDecimal::new(s);
|
printf("A[%d,%d]? ",i,j); s = ARGF.gets
|
||||||
end
|
a << BigDecimal::new(s);
|
||||||
printf("Contatant vector element b[%d] ? ",i);b<<=BigDecimal::new(gets);
|
as << BigDecimal::new(s);
|
||||||
|
end
|
||||||
|
printf("Contatant vector element b[%d] ? ",i); b << BigDecimal::new(ARGF.gets);
|
||||||
|
end
|
||||||
|
else
|
||||||
|
# Read data from specified file.
|
||||||
|
printf("Coefficient matrix and constant vector.\n");
|
||||||
|
for i in 0...n do
|
||||||
|
s = ARGF.gets
|
||||||
|
printf("%d) %s",i,s)
|
||||||
|
s = s.split
|
||||||
|
for j in 0...n do
|
||||||
|
a << BigDecimal::new(s[j]);
|
||||||
|
as << BigDecimal::new(s[j]);
|
||||||
|
end
|
||||||
|
b << BigDecimal::new(s[n]);
|
||||||
|
end
|
||||||
end
|
end
|
||||||
printf "ANS="
|
|
||||||
x = lusolve(a,b,ludecomp(a,n,zero,one),zero)
|
x = lusolve(a,b,ludecomp(a,n,zero,one),zero)
|
||||||
p x
|
printf("Answer(x[i] & (A*x-b)[i]) follows\n")
|
||||||
printf "A*x-b\n"
|
|
||||||
for i in 0...n do
|
for i in 0...n do
|
||||||
s = zero
|
printf("x[%d]=%s ",i,x[i].to_s)
|
||||||
for j in 0...n do
|
s = zero
|
||||||
|
for j in 0...n do
|
||||||
s = s + as[i*n+j]*x[j]
|
s = s + as[i*n+j]*x[j]
|
||||||
end
|
end
|
||||||
p s-b[i]
|
printf(" & %s\n",(s-b[i]).to_s)
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
Loading…
x
Reference in New Issue
Block a user