[Programmers] n^2 배열 자르기(Lv.2)

풀이 소요시간 : 45분

사실 풀이하는데 알고리즘이 따로 필요하지는 않았는데, 인덱스부분을 잘못 건드리다보니 시간초과가 발생했다. 0 부터 nxn 까지 모두 탐색하지 않고, left 부터 right 까지만 탐색하는 코드로 수정하니 시간초과가 해결되었다.

전체 코드

#include <string>
#include <vector>

using namespace std;

long long N;
vector<int> Vector;

vector<int> solution(int n, long long left, long long right) {
    
    N = n;
    // 전역 변수 세팅
    
    
    for(long long i = left; i <= right; i++)
    {
        int Row = i / N;
        int Col = i % N;
        
        if(Col <= Row)
        {
            Vector.push_back(Row + 1);
        }
        else
        {
            Vector.push_back(Col + 1);
        }
        
    }
    
    return Vector;
}