public class Heat2 {
    public static void main(String[] args) {
        double Te= 40.0, Tn=60.0, Tw=80.0, Ts=20.0;     // Tiempos en los extremos
        final int col= 5;
        final int row= 5;
        final int n= col * row;
        
        // Crea a, una matriz de coeficientes
        Matrix a= new Matrix(n,n);
        for (int i=0; i < n; i++)
            for (int j=0; j < n; j++) {
                if (i==j)
                    a.setElement(i, j, 4.0);
                // Complete este código: defina el elemento en -1.0
                // si el nodo i está al lado del nodo j
                // Cuidado con los extremos (p. ej., el nodo 5 no es
                // adyacente al nodo 4)
                else
                    a.setElement(i, j, 0.0);
            }
        System.out.println("A:");
        a.print();
        
        // Crea x, matriz de incógnitas
        Matrix x= new Matrix(n, 1);
        for (int i=0; i < n; i++)
            x.setElement(i, 0, 0.0);
        System.out.println("X inicial:");
        x.print();
        
        // Crea b, matriz de valores
        Matrix b= new Matrix(n, 1);
        for (int i=0; i < n; i++) {
            b.setElement(i, 0, 0.0);
            if (i < col)
                b.setElement(i, 0, b.getElement(i,0)+Tn);
            // Complete este código para el resto de extremos; sin ‘elses’!
            // Siempre suma el extremo a b
        }
        System.out.println("b:");
        b.print();
        
        GaussMain.gaussian(a, b, x);
        System.out.println("X final:");
        x.print();
        System.out.println("Rejilla de temperaturas:");
        // Muestra los resultados en una rejilla de 5 x 5. Utiliza enteros.
    }
}