2次元配列のポインタを引数にして、関数で値を書き換えたい。

Question

main関数内で2次元配列を宣言した時、main関数の外に定義した関数内でその2次元配列の値を書き換えたいです。つまりはこの2次元配列のポインタを引数にして関数に渡すと思ったのですが、うまくいきませんでした。どうすれば良いのか教えてください。
下のコードは簡易的に作ったものですみません。こんな感じのことをしたいです。

int b(int *a) {
 a[0][0] = 1;

 return 0;
}

int main(void) {
  int a[2][3];
  b(a);
  
  return 0;
}

Comments

[episteme]

与えたいのは R行C列のint行列:int [R][C] なんでしょうけど、C は固定値ですか?

[K.M.PEANUTS]

固定値です。

Answer 1

サイズ可変版

#include <stdio.h>

typedef int element;

/* "?行ld列 の行列 body" の row行col列に val をsetする */
void set_at(void* body, int ld, int row, int col, element val) {
  ((element*)body)[row*ld+col] = val;
}

/* "?行ld列 の行列 body" の row行col列の値を getする */
element get_at(void* body, int ld, int row, int col) {
  return ((element*)body)[row * ld + col];
}

int main() {
  element mtx[4][3];
  /* mtx[y][x] = x+y; */
  for ( int y = 0; y < 4; ++y ) {
    for ( int x = 0; x < 3; ++x ) {
      set_at(mtx, 3, y, x, x+y);
    }
  }

  /* できたかな? */
  for (int y = 0; y < 4; ++y) {
    for (int x = 0; x < 3; ++x) {
      printf("%2d ", mtx[y][x] /* get_at(mtx, 3, y, x)*/ );
    }
    printf("\n");
  }
  return 0;
}

Answer 2

int b(int a[][3]) { // あるいは a[2][3]
 a[0][0] = 1;

 return 0;
}

でしょうか。最初の添え字だけは省略できます。

関数側でサイズを持ちたくない場合は、一次元化して位置を計算。

int b(int *a, int n) {
 a[3*n+2] = 1;

 return 0;
}

int main(void) {
  int a[10][20]={0};
  b((int*)a, sizeof a[0]/sizeof a[0][0]);

  printf("%d\n",a[3][2]);
  return 0;
}

Answer 3

素直に、関数をint b(int a[2][3]) { ではダメ?

配列サイズが固定で無い場合、ちょっと考慮が必要ですが。

Answer 4

説明がややこしくなっちゃうのですが…

int b(int (*a)[10])
{
  a[0][0] = 1;
}

int main(void)
{
  int array[50][10];
  b(array);
}

int (*a)[10]では、aは「int型の要素が10個の配列」を指すポインタとなります。
この[10]の部分を可変にすることはできません。

Comments

[K.M.PEANUTS]

1次元の配列を10個、それぞれの配列の先頭要素のアドレスを渡したということですね！
納得しましたありがとうございます！！

[thkana]

> この[10]の部分を可変にすることはできません。
gccでcだと、
int b(int n, int (*a)[n]);
が通ったりします。