编程零基础算法 | 四、循环和选择结构——1572. 矩阵对角线元素的和

一、题目链接

1572. 矩阵对角线元素的和

二、题目简介

给你两个整数，n 和 start 。
数组 nums 定义为：nums[i] = start + 2*i（下标从 0 开始）且 n == nums.length 。
请返回 nums 中所有元素按位异或（XOR）后得到的结果。

三、源码讲解

int diagonalSum(int** mat, int matSize, int* matColSize){
    int sum = 0;
    int sec = matSize - 1;
    for(int i = 0; i < matSize; i++) {
        sum += mat[i][i];
        if(i != sec) sum += mat[i][sec];
        sec--;
    }
    return sum;
}

最先想到的就是双重循环判断对角线的下标，时间复杂度为O^2

也可以只用一个循环，左下对角线行列相等， 右下对角线行递增、列递减即可得到如下代码