题目描述
给你一个正方形矩阵 mat,请你返回矩阵对角线元素的和。
请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和。
输入:mat = [[1,2,3],
[4,5,6],
[7,8,9]]
输出:25
解释:对角线的和为:1 + 5 + 9 + 3 + 7 = 25
请注意,元素 mat[1][1] = 5 只会被计算一次。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/matrix-diagonal-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路
由于矩阵是正方形矩阵,因此行列相等,我们一行一行加,
先将第一行参与运算的元素相加结果保存,再加第二行,直到最后一行
定义两个指针start和end,start指向头元素,end指向末尾
将第一行两个指针所指的元素相加,
mat[start][start]+mat[start][end]
start++移到第二行,end-- 移到前一列
继续相加
直到start移动到最后一行 (下面代码存在问题)
#define _CRT_SECURE_NO_WARNINGS
#include<stdio.h>
int diagonalSum(int** mat, int matSize, int* matColSize)
{
int start = 0;
int end = *matColSize - 1;
int sum = 0;
for ( start = 0; start < matSize; start++,end--)
{
sum = sum + mat[start][start];
sum = sum + mat[start][end];
}
return sum;
}
int main()
{
//int mat[][3] = { {1, 2, 3},{4, 5, 6},{7, 8, 9}};
int mat[][5] = { {0,1,2,3,4},{5,6,7,8,9},{10,11,12,13,14},{15,16,17,18,19},{20,21,22,23,24}};
int matSize = sizeof(mat)/ sizeof(mat[0]);//计算二维数组有多少行
int matColSize = sizeof(mat[0]) / sizeof(mat[0][0]); // 计算二维数组有多少列
int sum = diagonalSum(&mat, matSize, &matColSize);
printf("%d", sum);
}
但是这样就会存在一个问题
当 start=2时,end=2,
sum = sum + mat[start][start];
sum = sum + mat[start][end];
数组最中间的元素被加了两次
这时 我们可以重新定义一个变量,把最中间的元素拿出来,先不参与相加,等整个矩阵计算完,再将矩阵最中间的元素加到最终结果里面。
修改后代码
#define _CRT_SECURE_NO_WARNINGS
#include<stdio.h>
int diagonalSum(int** mat, int matSize, int* matColSize)
{
int start = 0;
int end = *matColSize - 1;
int count = 0;
int sum = 0;
for ( start = 0; start < matSize; start++,end--)
{
if (start==end)
{
count = mat[start][end];
}
else
{
sum = sum + mat[start][start];
sum = sum + mat[start][end];
}
}
sum = count + sum;
return sum;
}
int main()
{
//int mat[][3] = { {1, 2, 3},{4, 5, 6},{7, 8, 9}};
int mat[][5] = { {0,1,2,3,4},{5,6,7,8,9},{10,11,12,13,14},{15,16,17,18,19},{20,21,22,23,24}};
int matSize = sizeof(mat)/ sizeof(mat[0]);//计算二维数组有多少行
int matColSize = sizeof(mat[0]) / sizeof(mat[0][0]); // 计算二维数组有多少列
int sum = diagonalSum(&mat, matSize, &matColSize);
printf("%d", sum);
}
这样就可以解决中间元素被加两次的问题
如果矩阵的行列数为偶数则不存在这个问题,用第一份代码也可以