[C++] 프로그래머스 이차원 배열 대각선 순회하기

문제설명

2차원 정수 배열 board와 정수 k가 주어집니다.
i + j <= k를 만족하는 모든 (i, j)에 대한
board[i][j]의 합을 return 하는 solution 함수를 완성해 주세요.

제한사항

• 1 ≤ board의 길이 ≤ 100
• 1 ≤ board[i]의 길이 ≤ 100
  ◦ 1 ≤ board[i][j] ≤ 10,000
  ◦ 모든 board[i]의 길이는 같습니다.
• 0 ≤ k < board의 길이 + board[i]의 길이

입출력 예

board	k	result
[[0, 1, 2],[1, 2, 3],[2, 3, 4],[3, 4, 5]]	2	8

풀이

i + j <= k를 만족하는 모든 (i, j)에 대한 board[i][j]의 합을 구하는 구현 문제입니다.

#include <string>
#include <vector>

using namespace std;

int solution(vector<vector<int>> board, int k) {
    int answer = 0;
    
    for(int i = 0; i < board.size(); i++)
    {
        for(int j = 0; j < board[i].size(); j++)
        {
            if(i + j <= k)
                answer += board[i][j];
        }
    }
    
    return answer;
}

결과