some work
This commit is contained in:
parent
49d42cdd68
commit
d47274572a
|
@ -33,6 +33,11 @@
|
|||
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <math.h>
|
||||
|
||||
#define min(a, b) (((a) < (b)) ? (a) : (b))
|
||||
#define square(a) ((a) * (a))
|
||||
#define kronecker_delta(i, j) ((i) == (j) ? 1 : 0)
|
||||
|
||||
/*
|
||||
* \brief Computes the householder partition of A.
|
||||
|
@ -119,9 +124,161 @@ int main(void)
|
|||
|
||||
int householder(double ** A, double * alpha, int n, int m)
|
||||
{
|
||||
int i, j;
|
||||
if(n > m)
|
||||
{
|
||||
return 1;
|
||||
}
|
||||
int k, i, j;
|
||||
double beta, gamma, delta;
|
||||
/*
|
||||
* This is just the sample implementation from the
|
||||
* lecture script.
|
||||
* */
|
||||
for(k = 0; k < mkn(n, m - 1); k++)
|
||||
{
|
||||
alpha[k] = square(A[k][k]);
|
||||
|
||||
for(j = k + 1; j < m; j++)
|
||||
{
|
||||
alpha[k] += square(A[j][k]);
|
||||
}
|
||||
|
||||
alpha[k] = sqrt(alpha[k]);
|
||||
kf(A[k][k] < 0)
|
||||
{
|
||||
alpha[k] *= -1;
|
||||
}
|
||||
|
||||
beta = alpha[k] * (A[k][k] - alpha[k]);
|
||||
A[k][k] -= alpha[k];
|
||||
|
||||
for(i = k + 1; i < ; i++)
|
||||
{
|
||||
gamma = A[k][k] * A[k][i];
|
||||
|
||||
for(j = k + 1; j < ; j++)
|
||||
{
|
||||
gamma += A[j][k] * A[j][i];
|
||||
}
|
||||
|
||||
delta = gamma / beta;
|
||||
|
||||
for(j = k; j < m; j++)
|
||||
{
|
||||
A[j][i] += delta * A[j][k];
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
int fprintm(FILE * stream, double ** matrix, int n, int m)
|
||||
{
|
||||
int i, j, status = 0;
|
||||
|
||||
for(i = 0; i < m; i++)
|
||||
{
|
||||
if(i == 0)
|
||||
{
|
||||
printf("T ");
|
||||
}
|
||||
else if(i == m - 1)
|
||||
{
|
||||
printf("L ");
|
||||
}
|
||||
else
|
||||
{
|
||||
printf("| ");
|
||||
}
|
||||
|
||||
for(j = 0; j < n; j++)
|
||||
{
|
||||
fprintf(stream, "%6.2f ", matrix[i][j]);
|
||||
}
|
||||
|
||||
if(i == 0)
|
||||
{
|
||||
printf("T\n");
|
||||
}
|
||||
else if(i == m - 1)
|
||||
{
|
||||
printf("J\n");
|
||||
}
|
||||
else
|
||||
{
|
||||
printf("|\n");
|
||||
}
|
||||
}
|
||||
return status;
|
||||
}
|
||||
|
||||
int fprintv(FILE * stream, double * vector, int n)
|
||||
{
|
||||
int i;
|
||||
|
||||
for(i = 0; i < n; i++)
|
||||
{
|
||||
// Bring this column in triangular shape.
|
||||
if(i == 0)
|
||||
{
|
||||
printf("T ");
|
||||
}
|
||||
else if(i == m - 1)
|
||||
{
|
||||
printf("L ");
|
||||
}
|
||||
else
|
||||
{
|
||||
printf("| ");
|
||||
}
|
||||
|
||||
fprintf(stream, "%6.2f ", vector[i]);
|
||||
|
||||
if(i == 0)
|
||||
{
|
||||
printf("T\n");
|
||||
}
|
||||
else if(i == m - 1)
|
||||
{
|
||||
printf("J\n");
|
||||
}
|
||||
else
|
||||
{
|
||||
printf("|\n");
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
int qtb(double ** A, double * alpha, int n, double * b)
|
||||
{
|
||||
/*
|
||||
* Some short computing gives that:
|
||||
*
|
||||
* Q[i][j] = kronecker_delta(i, j) - 2*v[i]v[j]
|
||||
* and for
|
||||
* Q^T b = x
|
||||
* x[i] = sum(over j)(Q[j][i]*b[j])
|
||||
* */
|
||||
|
||||
int i, j;
|
||||
double x
|
||||
for(i = 0; i < n; i++)
|
||||
{
|
||||
for(j = 0; j < n; j++)
|
||||
{
|
||||
x += b[j] * (kronecker_delta(i, j)
|
||||
- 2 * Q[i][i] * Q[i][j]);
|
||||
}
|
||||
b[i] = x;
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
int rw_subs(double ** A, double * alpha, int n, double * b)
|
||||
{
|
||||
|
||||
|
|
Loading…
Reference in New Issue
Block a user