📜  求平方矩阵的两个对角线之和的乘积

📅  最后修改于: 2021-04-29 04:36:05             🧑  作者: Mango

给定一个方阵由垫子大小的整数N×N个,任务是计算它的对角线的总和之间的产品。
例子:

Input: mat[][] = {{5, 8, 1}, 
                   {5, 10, 3}, 
                   {-6, 17, -9}}
Output: 30
Sum of primary diagonal = 5 + 10 + (-9) = 6.
Sum of secondary diagonal = 1 + 10 + (-6) = 5.
Product = 6 * 5 = 30.

Input: mat[][] = {{22, -8, 11}, 
                   {55, 87, -1}, 
                   {-61, 69, 19}}
Output: 4736

天真的方法:遍历整个矩阵并找到对角线元素。计算平方矩阵的两个对角线的和。然后,仅取所获得的两个和的乘积。
时间复杂度: O(N 2 )
天真的方法:通过观察对角线元素索引中的模式,仅对角线元素而不是整个矩阵进行遍历。
以下是此方法的实现:

CPP
// C++ program to find the product
// of the sum of diagonals.
 
#include 
using namespace std;
 
// Function to find the product
// of the sum of diagonals.
long long product(vector> &mat, int n)
{
    // Initialize sums of diagonals
    long long d1 = 0, d2 = 0;
 
    for (int i = 0; i < n; i++)
    {
        d1 += mat[i][i];
        d2 += mat[i][n - i - 1];
    }
     
    // Return the answer
    return 1LL * d1 * d2;
}
 
// Driven code
int main()
{
    vector> mat = {{ 5, 8, 1},
                               { 5, 10, 3},
                               { -6, 17, -9}};
                                
    int n = mat.size();
     
    // Function call
    cout << product(mat, n);
     
    return 0;
}


Java
// Java program to find the product
// of the sum of diagonals.
 
 
class GFG{
  
// Function to find the product
// of the sum of diagonals.
static long product(int [][]mat, int n)
{
    // Initialize sums of diagonals
    long d1 = 0, d2 = 0;
  
    for (int i = 0; i < n; i++)
    {
        d1 += mat[i][i];
        d2 += mat[i][n - i - 1];
    }
      
    // Return the answer
    return 1L * d1 * d2;
}
  
// Driven code
public static void main(String[] args)
{
    int [][]mat = {{ 5, 8, 1},
                               { 5, 10, 3},
                               { -6, 17, -9}};
                                 
    int n = mat.length;
      
    // Function call
    System.out.print(product(mat, n));
      
}
}
 
// This code is contributed by 29AjayKumar


Python3
# Python3 program to find the product
# of the sum of diagonals.
 
# Function to find the product
# of the sum of diagonals.
def product(mat,n):
 
    # Initialize sums of diagonals
    d1 = 0
    d2 = 0
 
    for i in range(n):
 
        d1 += mat[i][i]
        d2 += mat[i][n - i - 1]
 
    # Return the answer
    return d1 * d2
 
 
# Driven code
if __name__ == '__main__':
    mat = [[5, 8, 1],
        [5, 10, 3],
        [-6, 17, -9]]
 
    n = len(mat)
 
    # Function call
    print(product(mat, n))
     
# This code is contributed by mohit kumar 29


C#
// C# program to find the product
// of the sum of diagonals.
using System;
 
class GFG{
 
// Function to find the product
// of the sum of diagonals.
static long product(int [,]mat, int n)
{
    // Initialize sums of diagonals
    long d1 = 0, d2 = 0;
 
    for (int i = 0; i < n; i++)
    {
        d1 += mat[i, i];
        d2 += mat[i, n - i - 1];
    }
     
    // Return the answer
    return 1L * d1 * d2;
}
 
// Driven code
public static void Main(String[] args)
{
    int [,]mat = {{ 5, 8, 1},
                    { 5, 10, 3},
                    { -6, 17, -9}};
                                 
    int n = mat.GetLength(0);
     
    // Function call
    Console.Write(product(mat, n));
}
}
 
// This code is contributed by Princi Singh


Javascript


输出:
30

时间复杂度: O(N)