#include "iostream.h"
#include "math.h"

void main()
{
    //program d1r3
    //driver program for routine tridag
    int i,j,l,n = 3;
    double a[4], b[4], c[4], r[4], u[4], a1[4][4], x[4];
    //輸入已知的方程組的系數矩陣
    a1[1][1] = 1; a1[1][2] = 2; a1[1][3] = 0;
    a1[2][1] = 2; a1[2][2] = 2; a1[2][3] = 3;
    a1[3][1] = 0; a1[3][2] = 3; a1[3][3] = 3;
    //輸入已知的方程組的右端向量
    r[1] = 1;
    r[2] = 2;
    r[3] = 3;
    cout<<endl;
    cout<<"已知的方程組的右端向量"<<endl;
	cout.width(10);
    cout<<r[1]<<endl;
	cout.width(10);
    cout<<r[2]<<endl;
	cout.width(10);
    cout<<r[3]<<endl;
    for (i = 2; i<=n; i++)
	{
        a[i] = a1[i][i - 1];
    }
    for (i = 1; i<=n - 1; i++)
	{
        c[i] = a1[i + 1][i];
	}
    for (i = 1; i<=n; i++)
	{
        b[i] = a1[i][i];
    }
    tridag(a, b, c, r, u, n);
    cout<<endl;
    cout<<"計算出的方程組的解"<<endl;
	cout.width(10);
    cout<<u[1]<<endl;
	cout.width(10);
    cout<<u[2]<<endl;
	cout.width(10);
    cout<<u[3]<<endl;
    //將計算出的解乘以系數矩陣，以驗證計算結果正確
    for (l = 1; l<=n; l++)
	{
        x[l] = 0.0;
        for (j = 1; j<=n; j++)
		{
            x[l] = x[l] + a1[l][j] * u[j];
        }
    }
    cout<<endl;
    cout<<"計算出的解乘以系數矩陣的結果"<<endl;
	cout.width(10);
    cout<<x[1]<<endl;
	cout.width(10);
    cout<<x[2]<<endl;
	cout.width(10);
    cout<<x[3]<<endl;
	
}