jayati

Matrix Diagonal Sum

May 7th, 2023
235
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 0.64 KB | None | 0 0
  1. class Solution {
  2. public:
  3.     int diagonalSum(vector<vector<int>>& mat) {
  4.         int sum=0;
  5.         for(int i=0;i<mat.size();i++)
  6.         {
  7.             for(int j=0;j<mat[i].size();j++)
  8.             {
  9.                 if(i==j)
  10.                 {
  11.                     sum+=mat[i][j];
  12.                 }
  13.             }
  14.         }
  15.         int i=0;
  16.         int j=mat.size()-1;
  17.         while(i<mat.size())
  18.         {
  19.             sum+=mat[i][j];
  20.             i++;
  21.             j--;
  22.         }
  23.         if(mat.size()%2!=0)
  24.         {
  25.             int k=mat.size();
  26.             sum-=mat[k/2][k/2];
  27.         }
  28.         return sum;
  29.        
  30.     }
  31. };
Advertisement
Add Comment
Please, Sign In to add comment