[프로그래머스 / C++] n^2 배열 자르기

문제

n^2 배열 자르기 : 문제 링크

문제 분석

• 정수 n, left, right가 주어진다. 다음 과정을 거쳐서 1차원 배열을 만들고자 한다.

1. n행 n열 크기의 비어있는 2차원 배열을 만든다.
2. i = 1, 2, 3, ..., n에 대해서, 다음 과정을 반복힌다.

• 1행 1열부터 i행 i열까지의 영역 내의 모든 빈 칸을 숫자 i로 채운다.

3. 1행, 2행, ..., n행을 잘라내어 모두 이어붙인 새로운 1차원 배열을 만든다.
4. 새로운 1차원 배열을 arr이라 할 때, arr[left], arr[left+1], ..., arr[right]만 남기고 나머지는 지운다.

• 정수 n, left, right가 매개변수로 주어진다. 주어진 과정대로 만들어진 1차원 배열을 return 하도록 solution 함수를 완성

• 제한 사항

• 1 ≤ n ≤ 10^7
• 0 ≤ left ≤ right < n^2
• right - left < 10^5

• 주어진 과정대로 만들어진 1차원 배열을 저장할 정수형 벡터 answer을 초기화. for loop의 초기화식을 i = left, 조건식을 i <= right로 설정하여 주어진 범위만큼 순회. 현재 인덱스를 n으로 나눈 몫은 x, 나머지는 y 좌표가 되므로 각각을 long long형 변수 x, y에 저장. if문을 사용하여 y크가면 y에 1을 더한 값을 이외의 경우에는 else문을 사용하여 x에 1을 더한 값을 answer에 저장. loop 탈출 후, 최종적으로 저장된 answer을 return

풀이

#include <vector>

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) {
        long long x = i / n;
        long long y = i % n;
        if(x < y) answer.push_back(y + 1);
        else answer.push_back(x + 1);
    }
    return answer;
}