C语言编写能加减乘矩阵的程序

C语言编写能加减乘矩阵的程序

#include "stdafx.h"
#include<stdlib.h>

#include <stdio.h>
//新建
int **newArr(int X, int Y) {
int **p, i;

p = (int **)malloc(X * sizeof(int *));
for (i = 0; i < X; i++) {
p[i] = (int *)malloc(Y * sizeof(int));
}
return p;
}

//释放内存
void deleteArr(int **p, int X) {
int i;
for (i = 0; i < X; i++)
free(p[i]);
free(p);
}

//计算矩阵相加    参数1:矩阵A 参数2:矩阵B 参数3:矩阵行数,列数
int **addArr(int **A, int **B,int X,int Y)
{
int **p, i,j;
p = (int **)malloc(X * sizeof(int *));
for (i = 0; i < X; i++)
p[i] = (int *)malloc(Y * sizeof(int));

for (i = 0; i < X; i++)
{
for (j = 0; j < Y; j++)
p[i][j] = A[i][j] + B[i][j];
}
return p;
}

//计算矩阵相减    参数1:矩阵A 参数2:矩阵B 参数3:矩阵行数,列数
int **subArr(int **A, int **B, int X, int Y)
{
int **p, i, j;
p = (int **)malloc(X * sizeof(int *));
for (i = 0; i < X; i++)
p[i] = (int *)malloc(Y * sizeof(int));

for (i = 0; i < X; i++)
{
for (j = 0; j < Y; j++)
p[i][j] = A[i][j] - B[i][j];
}
return p;
}

//计算矩阵相乘    参数1:矩阵A 参数2:矩阵B 参数3,4,5,6:矩阵行数,列数
int **multiplyArr(int **A, int **B, int aX, int aY, int bX, int bY)
{
//A[aX][aY] * B[bX][bY] =C[aX][bY]
int **p, i,j,k;
p = (int **)malloc(aX * sizeof(int *));
for (i = 0; i < aX; i++)
p[i] = (int *)malloc(bY * sizeof(int));

for (i = 0; i < aX; i++)
{
for (j = 0; j < bY; j++)
{
p[i][j] = 0;
}

}

for (i = 0; i<aX; i++)
{
for (j = 0; j<bY; j++)
{
for (k = 0; k<aY; k++)
{
p[i][j] += A[i][k] * B[k][j];
}
}
}
return p;
}

//打印矩阵  参数1:矩阵  参数2,3:矩阵行数,列数
void printArr(int **arr,  int X, int Y)
{
int i, j;
printf("输出矩阵开始:\n");
for (i = 0; i < X; i++) {
for (j = 0; j < Y; j++) {
printf(" %5d", arr[i][j]);
}
printf("\n");
}
printf("输出矩阵结束:\n");
}

void main()
{
int **aArr, **bArr, i, j, aX, aY, bX, bY;

printf("输入矩阵A的行数:\n");
scanf("%d", &aX);
printf("输入矩阵A的列数:\n");
scanf("%d", &aY);

printf("输入A矩阵:\n");
aArr = newArr(aX, aY);
for (i = 0; i < aX; i++) {
for (j = 0; j < aY; j++) {
scanf("%d", &aArr[i][j]);
}
}

printf("输入矩阵B的行数:\n");
scanf("%d", &bX);
printf("输入矩阵B的列数:\n");
scanf("%d", &bY);

printf("输入B矩阵:\n");
bArr = newArr(bX, bY);
for (i = 0; i < bX; i++)
{
for (j = 0; j < bY; j++)
scanf("%d", &bArr[i][j]);
}

//打印 A,B 矩阵
printArr(aArr, aX, aY);
printArr(bArr, bX, bY);

//计算
//判断是否能进行加减运算
if (aX==bX&&aY==bY)
{
printf("矩阵A,B可进行加减运算\n");
//矩阵相加
int **resultAdd = addArr(aArr, bArr, aX, aY);
//打印矩阵相加结果
printArr(resultAdd, aX, aY);
//释放内存
deleteArr(resultAdd, aX);

//矩阵相减
int **resultSub = subArr(aArr, bArr, aX, aY);
//打印矩阵相减结果
printArr(resultSub, aX, aY);
//释放内存
deleteArr(resultSub, aX);
}
else
{
printf("矩阵A,B不可进行加减运算\n");
}

//判断是否能进行乘法运算
if (aY==bX)
{
printf("矩阵A,B可进行乘法运算\n");

int **resultMultiply = multiplyArr(aArr, bArr, aX, aY,bX,bY);
//打印矩阵相减结果
printArr(resultMultiply, aX, bY);
//释放内存
deleteArr(resultMultiply, aX);
}
else
{
printf("矩阵A,B不可进行乘法运算\n");
}
//释放内存
deleteArr(aArr, aX);
deleteArr(bArr, bX);
}