Tridiagonale-matrix-algoritme

Uit testwiki
Naar navigatie springen Naar zoeken springen

Het tridiagonale-matrix-algoritme, kortweg TDMA genoemd, en ook bekend als het Thomas-algoritme, is een numerieke methode om een vierkant stelsel van lineaire vergelijkingen op te lossen dat wordt beschreven door een tridiagonale matrix. Dit is een matrix waarbij de elementen buiten de diagonaal en de twee nevendiagonalen alle gelijk aan nul zijn, zoals in onderstaande matrix. De methode is niets anders dan een toepassing van de Gauss-eliminatiemethode, met als doel de elementen onder de hoofddiagonaal nul te maken, waardoor het stelsel een geschikte vorm krijgt om te worden opgelost door achterwaartse substitutie.

Methode

Een tridiagonaal stelsel van n vergelijkingen en n onbekenden heeft als vorm:

aixi1+bixi+cixi+1=di

waarin a1=0 en cn=0. In matrixvorm wordt dit:

[b1c100a2b2c20a3b3c3000an1bn1cn1000anbn][x1x2x3xn]=[d1d2d3dn]

De elementen bi vormen hierbij samen de hoofdiagonaal of kortweg diagonaal. De elementen ai en ci bevinden zich op de zogenaamde nevendiagonalen, meer bepaald respectievelijk op de onderdiagonaal en de bovendiagonaal.

De oplossing van dergelijke stelsels kan worden uitgevoerd in O(n) bewerkingen in plaats van de normale O(n3) bewerkingen van de Gauss-methode. In een eerste stap worden de coëfficiënten ai weggewerkt, waarna via achterwaartse substitutie de oplossing wordt gevonden. Dit soort stelsels komt bijvoorbeeld voor bij de numerieke oplossing van de diffusievergelijking en bij het berekenen van natuurlijke splines.

In de eerste stap worden de coëfficiënten van het stelsel omgevormd, opdat alle elementen van de coëfficiëntenmatrix op de onderdiagonaal nul worden. De nieuwe coëfficiënten worden in onderstaande vergelijkingen met een accent aangeduid:

c'i={c1b1;i=1cibic'i1ai;i=2,3,,n1

en:

d'i={d1b1;i=1did'i1aibic'i1ai;i=2,3,,n.

Na deze voorwaartse stap volgt als tweede stap een achterwaartse substitutie die de oplossing levert:

xn=d'n
xi=d'ic'ixi+1; i=n1,n2,,1.

Concrete implementaties

Het Engelstalige artikel over dit onderwerp bevat voorbeelden van implementaties in de programmeertalen C, Matlab en Fortran. Bij het programmeren dient men ermee rekening te houden dat de nummering van de elementen van een vector of een matrix in sommige programmeertalen loopt van 1 tot n, en in andere van 0 tot n1. In het laatste geval dienen de uitdrukkingen in dit artikel aangepast te worden.

Afleiding

De afleiding van het tridiagonaal-matrix-algoritme is gebaseerd op Gauss-eliminatie, waarbij de elementen van de onderdiagonaal nul worden gemaakt door het uitvoeren van elementaire rijoperaties. Stel dat de onbekenden van het stelsel x1,,xn zijn en de vergelijkingen:

b1x1+c1x2=d1;i=1aixi1+bixi+cixi+1=di;i=2,,n1anxn1+bnxn=dn;i=n

Dan wordt de tweede vergelijking door middel van de eerste als volgt gewijzigd:

(vergelijking 2)b1(vergelijking 1)a2

Het resultaat van deze elementaire rijoperatie is:

(a2x1+b2x2+c2x3)b1(b1x1+c1x2)a2=d2b1d1a2
(b2b1c1a2)x2+c2b1x3=d2b1d1a2

Op deze manier is x1 verwijderd uit de tweede vergelijking. Met behulp van de tweede vergelijking wordt dan de derde op analoge manier behandeld, zodat:

(a3x2+b3x3+c3x4)(b2b1c1a2)((b2b1c1a2)x2+c2b1x3)a3=d3(b2b1c1a2)(d2b1d1a2)a3
(b3(b2b1c1a2)c2b1a3)x3+c3(b2b1c1a2)x4=d3(b2b1c1a2)(d2b1d1a2)a3

Dit systeem van rijoperaties wordt nu verder toegepast doorheen de matrix, waardoor alle elementen op de onderdiagonaal nul worden. Als gevolg hiervan bevat de n-de rij slechts één onbekende, namelijk xn, die dus direct kan gevonden worden. Vervolgens worden de andere onbekenden in volgorde van onder naar boven bepaald. Bij elke stap bevat de te behandelen vergelijking immers nog slechts één onbekende. De laatste onbekende die wordt bepaald is dus x1.

Indien de coëfficiënten van de gewijzigde vergelijkingen expliciet worden opgeschreven worden ze steeds ingewikkelder. Het is daarom beter de gewijzigde coëfficiënten, hieronder aangeduid met een tilde, recursief te definiëren.

a~i=0
b~1=b1
b~i=bib~i1c~i1ai
c~1=c1
c~i=cib~i1
d~1=d1
d~i=dib~i1d~i1ai

Om de oplossing nog vlotter te laten verlopen kan men ook nog b~i wegdelen (indien niet deze niet nul zijn). De aldus gewijzigde coëfficiënten, aangeduid met een asterisk, zijn:

a'i=0
b'i=1
c'1=c1b1
c'i=cibic'i1ai
d'1=d1b1
d'i=did'i1aibic'i1ai

Dit geeft het volgende stelsel, nog steeds in de oorspronkelijke onbekenden:

xi+c'ixi+1=d'i; i=1,,n1xn=d'n; i=n

De oplossingen zijn, startend vanaf xn en zo door achterwaartse substitutie naar x1:

xn=d'n
xi=d'ic'ixi+1; i=n1,n2,,1.

Varianten

In bepaalde situaties, zoals in het geval van periodieke randvoorwaarden, komt een lichtjes andere vorm van een tridiagonaal stelsel voor:

a1xn+b1x1+c1x2=d1aixi1+bixi+cixi+1=di,i=2,,n1anxn1+bnxn+cnx1=dn

In dat geval kan de formule van Sherlan-Morrison gebruikt worden om bijkomende bewerkingen tijdens de elementaire rijoperaties te vermijden. In andere gevallen komt een stelsel voor dat in matrixvorm een blokmatrix geeft. Ook in dit geval bestaan er aangepaste methoden van de Gauss-eliminatiemethode.

Referenties

Bronnen