프로그래머스 n^2 배열 자르기

 

재밌는 문제였다.

나는 되게 금방 풀었는데, 다른 사람들의 후기를 보니 쉽지 않은 문제였나보다.

또한 내 풀이가 엄청난 최적화 풀이인 것 같아서 뿌듯.

#include <string>
#include <vector>
#define MAX(a, b) ((a >= b)? a : b)

using namespace std;

vector<int> solution(int n, long long left, long long right) {
    
    vector<int> answer;
    
    for(long long i = left; i <= right; i++)
        answer.push_back(MAX(i/n, i%n) +1);
    
    return answer;
}