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

문제

이차원 배열 대각선 순회하기 : 문제 링크

문제 분석

• 2차원 정수 배열 board와 정수 k가 주어진다. i + j <= k를 만족하는 모든 (i,j)에 대한 board[i][j]의 합을 return
• 문제 조건을 만족하는 board[i][j]의 합을 저장할 정수형 변수 answer을 0으로 초기화. 모든 (i,j)를 순환하기 위해 2중 for loop를 사용. 첫번째 loop는 board, 두번째 loop는 board[i]의 크기만큼 순환. if문을 통해 i + j <= k의 조건을 만족한다면, 해당하는 board[i][j]를 answer에 저장. 최종적으로 저장된 answer을 return

풀이

#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;
}