Level 0. 이차원 배열 대각선 순회하기

Level 0. 이차원 배열 대각선 순회하기

 

 

범위만 넘어가지 않게 조심하면 된다.

 

#include <iostream>
#include <vector>
#include <numeric>

using namespace std;

int solution(vector<vector<int>> board, int k)
{
    int answer = 0;
    int row = board.size(), col = board.front().size();

    ++k;

    for (int i = 0; i < row && i < k; ++i)
    {
        int c = (col <= k - i) ? col : k - i;

        answer += accumulate(board[i].begin(), board[i].begin() + c, 0);
    }

    return answer;
}