solv.c 1.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172
  1. /* solv.c CCMATH mathematics library source code.
  2. *
  3. * Copyright (C) 2000 Daniel A. Atkinson All rights reserved.
  4. * This code may be redistributed under the terms of the GNU general
  5. * public license. ( See the gpl.license file for details.)
  6. * ------------------------------------------------------------------------
  7. */
  8. #include <stdlib.h>
  9. #include "ccmath.h"
  10. int solv(double *a, double *b, int n)
  11. {
  12. int i, j, k, lc;
  13. double *ps, *p, *q, *pa, *pd;
  14. double *q0, s, t, tq = 0., zr = 1.e-15;
  15. q0 = (double *)calloc(n, sizeof(double));
  16. for (j = 0, pa = a, pd = a; j < n; ++j, ++pa, pd += n + 1) {
  17. if (j) {
  18. for (i = 0, q = q0, p = pa; i < n; ++i, p += n)
  19. *q++ = *p;
  20. for (i = 1; i < n; ++i) {
  21. lc = i < j ? i : j;
  22. for (k = 0, p = pa + i * n - j, q = q0, t = 0.; k < lc; ++k)
  23. t += *p++ * *q++;
  24. q0[i] -= t;
  25. }
  26. for (i = 0, q = q0, p = pa; i < n; ++i, p += n)
  27. *p = *q++;
  28. }
  29. s = fabs(*pd);
  30. lc = j;
  31. for (k = j + 1, ps = pd; k < n; ++k) {
  32. if ((t = fabs(*(ps += n))) > s) {
  33. s = t;
  34. lc = k;
  35. }
  36. }
  37. tq = tq > s ? tq : s;
  38. if (s < zr * tq) {
  39. free(q0);
  40. return -1;
  41. }
  42. if (lc != j) {
  43. t = b[j];
  44. b[j] = b[lc];
  45. b[lc] = t;
  46. for (k = 0, p = a + n * j, q = a + n * lc; k < n; ++k) {
  47. t = *p;
  48. *p++ = *q;
  49. *q++ = t;
  50. }
  51. }
  52. for (k = j + 1, ps = pd, t = 1. / *pd; k < n; ++k)
  53. *(ps += n) *= t;
  54. }
  55. for (j = 1, ps = b + 1; j < n; ++j) {
  56. for (k = 0, p = a + n * j, q = b, t = 0.; k < j; ++k)
  57. t += *p++ * *q++;
  58. *ps++ -= t;
  59. }
  60. for (j = n - 1, --ps, pd = a + n * n - 1; j >= 0; --j, pd -= n + 1) {
  61. for (k = j + 1, p = pd, q = b + j, t = 0.; k < n; ++k)
  62. t += *++p * *++q;
  63. *ps -= t;
  64. *ps-- /= *pd;
  65. }
  66. free(q0);
  67. return 0;
  68. }