在一维数组中乘以矩阵

时间:2017-10-30 20:28:14

标签: c arrays matrix matrix-multiplication

void multiply(int a[], int row1, int col1, int b[], int row2, int col2) {
    int d[size];
    for (int i = 0; i < row1; i++) {
        for (int j = 0; j < col2; j++) {
            int sum = 0.0;
            for (int k = 0; k < col2; k++)
                sum = sum + a[i * col1 + k] * b[k * col2 + j];
            d[i * col2 + j] = sum;
        }
    }
    for (int i = 0; i < size; i++) {
        if (i % col2 == 0) {
            printf("\n");
        }
        printf("%d ", d[i]);
    }
}

我将此作为多个两个一维数组的函数,这些数组应该是矩阵。我使用的是在线编译器,在运行代码时它什么都没给我。

编辑:

#include <stdio.h>
#include <stdlib.h>
#include <stdbool.h>

int size = 8;

void multiply(int a[], int row1, int col1, int b[], int row2, int col2) {
    int d[size];
    for (int i = 0; i < row1; i++) {
        for (int j = 0; j < col2; j++) {
            int sum = 0.0;
            for (int k = 0; k < col2; k++) {
            sum = sum + a[i * col1 + k] * b[k * col2 + j];
            }
            d[i * col2 + j] = sum;
        }
    }
    for (int i = 0; i < size; i++) {
    if (i % col2 == 0) {
      printf("\n");
    }
    printf("%d ", d[i]);
  }
}

int main() {
    int a[size] = {
    1,  2,  3,  4, // 0  1  2  3
    5,  6,  7,  8  // 4  5  6  7
    };
    int c[size] = {
    1,  4,  // 0  1
    2,  3,  // 2  3
    3,  2,  // 4  5
    4,  1   // 6  7
    };
    printf("Multipying Matrices\n");
    multiply(a, 2, 4, c, 4, 2);
}

这是我运行的整个代码,用于测试Costantino Grana的乘法函数,但编辑它是因为我没有进入内存分配。

此代码的结果是:

5 10
17 38
2 4
6 8

这已经错了,因为两个矩阵的乘法应该产生一个2x2矩阵。

1 个答案:

答案 0 :(得分:2)

以下是您应该发布的the correct reference的固定版本。

#include <assert.h>
#include <stdio.h>
#include <stdlib.h>

void multiply(int *a, int row1, int col1, int *b, int row2, int col2) 
{
    assert(col1 == row2);

    int size = row1*col2;
#ifdef _MSC_VER
    int *d = malloc(size * sizeof *d);
#else
    int d[size];
#endif
    for (int i = 0; i < row1; i++) {
        for (int j = 0; j < col2; j++) {
            int sum = 0;
            for (int k = 0; k < col1; k++)
                sum = sum + a[i * col1 + k] * b[k * col2 + j];
            d[i * col2 + j] = sum;
        }
    }

    for (int i = 0; i < size; i++) {
        if (i % col2 == 0) {
            printf("\n");
        }
        printf("%d ", d[i]);
    }

#ifdef _MSC_VER
    free(d);
#endif
}

int main(void)
{
    int a[] = {
        1, 2, 3,
        4, 5, 6,
    };
    int b[] = {
        7, 10,
        8, 11,
        9, 12,
    };

    multiply(a, 2, 3, b, 3, 2);
}

除了避免使用VLA的解决方案之外,错误在于内部循环,它应该运行到col1或row2,而不是col2。

顺便说一下:我避免传递col1 row2。这些应该是一样的。我将a[]b[]替换为*a*b,这更贴近真实的东西(我的意见)。此外,你应该避免混合计算和打印。让函数返回一个新矩阵和另一个函数来打印矩阵。 最后,使用结构将数据指针,行和列保持在一起。