如何使用vector<vector<int>>找到严格高于矩阵次对角线的元素之和？

Question

如果输入是：

4 
1 6 3 1
6 1 3 1
1 3 1 6
3 1 6 1

我的输出应该是：

18 

1 + 6 + 3 + 6 + 1 + 1 = 18

第二条对角线是从右上角开始到左下角的对角线

       1
     3
   3
 3

我需要找到严格高于第二条对角线的元素的总和。

这是我目前的代码：

int n , sum = 0;
cin >> n;

vector<vector<int>> A(n);

for (auto rows : A)
   for (auto elements : rows)
       cin >> elements;

for(auto x : A) 
   sum += A[x] [n - x];

cout << sum;

我是矩阵向量的新手。另外，如果你们知道我可以在哪里提高我的矢量技能，并更好地理解 stl 矢量，您的建议将非常有帮助！

如果您能帮助我，将不胜感激！

谢谢大家的回答！

Answer 1

对于初学者，向量应定义为

vector<vector<int>> A(n, std::vector<int>( n ) );

这些循环应该使用对对象的引用

for (auto &rows : A)
   for (auto &elements : rows)
       cin >> elements;

要查找总和，您可以使用索引变量，例如

long long int sum = 0;

for ( auto n = A.size(); n-- != 0;  )
{
    for ( std::vector<int>::size_type i = 0; i < n; i++ )
    {
        sum += A[A.size() - n - 1][i];
    }
}

这是一个演示程序

#include <iostream>
#include <vector>
#include <cstdlib>
#include <ctime>

int main() 
{
    size_t n = 0;

    std::cout << "Enter the size of a matrix: ";

    std::cin >> n;

    std::vector<std::vector<int>> v( n, std::vector<int>( n ) );

    std::srand( ( unsigned int )std::time( nullptr ) );

    for ( auto &row : v )
    {
        for ( auto &item : row ) item = std::rand() % n;
    }

    for ( const auto &row : v )
    {
        for ( const auto &item : row ) std::cout << item << ' ';
        std::cout << '\n';
    }

    long long int sum = 0;

    for ( auto n = v.size(); n-- != 0; )
    {
        for ( std::vector<int>::size_type i = 0; i < n; i++ )
        {
            sum += v[v.size() - n - 1][i];
        }
    }

    std::cout << "\nThe sum is equal to " << sum << '\n';

    return 0;
}

它的输出可能看起来像

Enter the size of a matrix: 5
4 4 0 2 0 
1 4 4 3 3 
3 3 1 0 4 
2 0 2 3 2 
2 1 3 0 3 

The sum is equal to 27

Answer 2

使用索引：

...
for (int i = 0; i < A.size(); i++)
   for (int j = 0; j < A[i].size() - i; j++)
       sum += A[i][j];
...