在C中乘以矩阵和向量

时间:2012-10-23 05:58:38

标签: c arrays 2d malloc

我想将矩阵和向量相乘,我已经编写了函数来将矩阵和向量存储在malloc数组中。对于这个函数,我需要使用malloc创建另一个数组来存储我的答案。然后进行计算(http://www.facstaff.bucknell.edu/mastascu/elessonsHTML/Circuit/MatVecMultiply.htm)

#include <stdlib.h>
/* Multiply a matrix by a vector. */
double *matrix_vector_multiply(int rows, int cols,
               double **mat, double *vec){

 // creating an vecotr to hold the answer first

 double *ans= malloc(rows* sizeof(double));

 // do the multiplication:
 mulitply **mat and *vec (mat = the matrix and vec is the vector)

 for (rows=0; rows< ; rows++)
    for (cols=0; cols< ; cols++)
        ans[rows] = ans[rows] + vec[rows][cols] * mat[rows];    

 //not sure if it is right

 // then store the answer back to the ans array            


}

主要功能:

double *matrix_vector_multiply(int rows, int cols,
               double **mat, double *vec);

int main(){
  double *answer = matrix_vector_multiply(rows, cols, matrix, vector);

  printf("Answer vector: \n");
  print_vector(rows, answer);
  return 0;
 }

不确定如何使用指针进行此乘法然后将其存储回来.. 任何帮助,将不胜感激!谢谢!

编辑:乘法函数:

#include <stdlib.h>
/* Multiply a matrix by a vector. */
double *matrix_vector_multiply(int rows, int cols,
               double **mat, double *vec){

double *ans = malloc(rows * sizeof (double));   
int i;  
for (i=0; i<rows; rows++)
    for (i=0; i<cols; cols++)
        ans[rows] = ans[rows] + vec[rows][cols] * mat[rows];    

return ans;            

}

但我在第12行收到错误,下标值是数组也不是指针

1 个答案:

答案 0 :(得分:2)

你的职能有几个错误:

  1. 迭代for循环中矩阵的变量不应该是传递给函数的参数。试试这样的事情: for(y=0;y<rows;y++)

  2. 您必须在for循环中交换vecmat

  3. 您必须将答案向量初始化为0(另一个用于循环)

  4. 你必须在乘法结束时返回答案(return ans;

  5. 希望有所帮助, 扬