[C++] 프로그래머스 문제풀이 LEVEL 2 n^2 배열 자르기

 

 

https://school.programmers.co.kr/learn/courses/30/lessons/87390

프로그래머스

 

 

#include <bits/stdc++.h>

using namespace std;

vector<int> solution(int n, long long left, long long right) {
    vector<int> answer;
    answer.reserve(n);
    
    for (long long i = left; i <= right; ++i){
        int r = i / n;
        int c = i % n;
        
        if (r == c){
            answer.push_back(r + 1);
        }
        else {
            answer.push_back(std::max(r, c) + 1);
        }
    }
    return answer;
}