2次元配列を関数の引数にする方法について
以下のプログラムはSEIRモデル(https://ja.wikipedia.org/wiki/SEIR%E3%83%A2%E3%83%87%E3%83%AB)を2つたて、それらを相互作用させたときのA,Bそれぞれの解をルンゲクッタ法を用いて求めています。
ここで一次元配列a[i],b[j]を2次元配列a[2][4](i:領域A,B,j:SEIR)とし使用した場合、
関数の定義のところなどどのように書けばよいのでしょうか?
C言語初心者でして、お見苦しい点多々あると思いますが、何卒ご教示お願いします。。。
#include <stdio.h>
#include <math.h>
double alf=0.05, bet=0.005, gam=0.05; ///alf:移行率,bet:感染率,gam:除外率
double f1(double t1,double a0,double a1,double a2,double a3){return -bet*a0*a2;}
double f2(double t1,double a0,double a1,double a2,double a3){return bet*a0*a2-alf*a1;}
double f3(double t1,double a0,double a1,double a2,double a3){return alf*a1-gam*a2;}
double f4(double t1,double a0,double a1,double a2,double a3){return gam*a2;} //箱Aの関数
double g1(double t1,double b0,double b1,double b2,double b3){return -bet*b0*b2;}
double g2(double t1,double b0,double b1,double b2,double b3){return bet*b0*b2-alf*b1;}
double g3(double t1,double b0,double b1,double b2,double b3){return alf*b1-gam*b2;}
double g4(double t1,double b0,double b1,double b2,double b3){return gam*b2;} //箱Bの関数
int main(void) {
double t1,dt,t1max;
double k1[4],k2[4],k3[4],k4[4],l1[4],l2[4],l3[4],l4[4],a[4],b[4],da[4],db[4] ;
double (*pf[4])(double,double,double,double,double)={f1,f2,f3,f4};
double (*pg[4])(double,double,double,double,double)={g1,g2,g3,g4};
int i; //宣言
t1 = 0.0;
dt = 0.1;
t1max = 20.0; //時間初期値
a[0] = 2000.0;
a[1] = 300.0;
a[2] = 400.0;
a[3] = 300.0; //箱A初期値(a[0]:感受性人口、a[1]:潜伏人口、a[2]:感染人口、a[3]:隔離人口)
b[0] = 1000.0;
b[1] = 200.0;
b[2] = 300.0;
b[3] = 500.0; //箱B初期値(b[0]:感受性人口、b[1]:潜伏人口、b[2]:感染人口、b[3]:隔離人口)
printf("%4.2f,%10.3f,%10.3f,%10.3f,%10.3f,%10.3f,%10.3f,%10.3f,%10.3f,%10.3f\n",t1,a[0],a[1],a[2],a[3],b[0],b[1],b[2],b[3],a[0]+a[1]+a[2]+a[3]+b[0]+b[1]+b[2]+b[3]);
for(t1=0.0;t1<=t1max;t1+=dt) {
for (i=0;i<4;i++) k1[i]=dt*pf[i](t1,a[0],a[1],a[2],a[3]);
for (i=0;i<4;i++) k2[i]=dt*pf[i](t1+dt/2.0,a[0]+k1[0]/2.0,a[1]+k1[1]/2.0,a[2]+k1[2]/2.0,a[3]+k1[3]/2.0);
for (i=0;i<4;i++) k3[i]=dt*pf[i](t1+dt/2.0,a[0]+k2[0]/2.0,a[1]+k2[1]/2.0,a[2]+k2[2]/2.0,a[3]+k2[3]/2.0);
for (i=0;i<4;i++) k4[i]=dt*pf[i](t1+dt,a[0]+k3[0],a[1]+k3[1],a[2]+k3[2],a[3]+k3[3]);
for (i=0;i<4;i++) l1[i]=dt*pg[i](t1,b[0],b[1],b[2],b[3]);
for (i=0;i<4;i++) l2[i]=dt*pg[i](t1+dt/2.0,b[0]+l1[0]/2.0,b[1]+l1[1]/2.0,b[2]+l1[2]/2.0,b[3]+l1[3]/2.0);
for (i=0;i<4;i++) l3[i]=dt*pg[i](t1+dt/2.0,b[0]+l2[0]/2.0,b[1]+l2[1]/2.0,b[2]+l2[2]/2.0,b[3]+l2[3]/2.0);
for (i=0;i<4;i++) l4[i]=dt*pg[i](t1+dt,b[0]+l3[0],b[1]+l3[1],b[2]+l3[2],b[3]+l3[3]);
for (i=0;i<=3;i++) a[i]=a[i]+((k1[i]+2.0*k2[i]+2.0*k3[i]+k4[i])/6.0);
for (i=0;i<=3;i++) b[i]=b[i]+((l1[i]+2.0*l2[i]+2.0*l3[i]+l4[i])/6.0); //4次のルンゲクッタ
da[0]=-0.05*a[0];
da[1]=-0.05*a[1];
da[2]=-0.01*a[2];
da[3]=-0.05*a[3]; //AからBへの移動量
db[0]=-0.05*b[0];
db[1]=-0.05*b[1];
db[2]=-0.02*b[2];
db[3]=-0.05*b[3]; //BからAへの移動量
if (t1 < 15.0) { //移動開始
for (i=0;i<=3;i++) {
a[i]=a[i]+da[i];
b[i]=b[i]-da[i]; ////AからBへ移動後のAにおけるSEIRの数,AからBへ移動後のBにおけるSEIRの数
}
} else {
for (i=0;i<=3;i++) {
a[i]=a[i]-db[i]; ////BからAへの移動後のAにおけるSEIRの数
b[i]=b[i]+db[i]; ////BからAへの移動後のBにおけるSEIRの数
}
}
printf("%4.2f,%10.3f,%10.3f,%10.3f,%10.3f,%10.3f,%10.3f,%10.3f,%10.3f,%10.3f\n",t1+0.1,a[0],a[1],a[2],a[3],b[0],b[1],b[2],b[3],a[0]+a[1]+a[2]+a[3]+b[0]+b[1]+b[2]+b[3]);
}
return 0;
}
お礼
いつもありがとうございます。早速試してみます。