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

YH J·2023년 9월 18일
0

프로그래머스

목록 보기
146/168

문제 링크

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

내 풀이

굳이 배열을 만들어서 할 필요는 없고 규칙만 찾으면 된다.
예시에서 1행은 1, 2, 3
2행은 2, 2, 3
3행은 3, 3, 3 이다.
행의 1열은 행의 숫자를 따라가고, 그 뒤의 열들은 행 이상이면 원래 열의 숫자, 행 이하면 행의 숫자를 따라간다.
left부터 right까지 for문을 돌리면서
a는 index가 몇번째 행인지
b는 index가 몇번째 열인지 구한 값이다.
행의 값이 열의 값 이상이면 행값을, 미만이면 열 값을 push_back한다.

내 코드

#include <string>
#include <vector>
#include <iostream>
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 a = i / n + 1;
        long long b = i % n + 1;
        a >= b ? answer.push_back((int)a) : answer.push_back((int)(b));
    }
    
    return answer;
}

다른 사람의 풀이

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>

using namespace std;

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

    return answer;
}

다른 사람의 풀이 해석

구한 뒤 비교하는 것 없이 그냥 max를 사용했다.

profile
게임 개발자 지망생

0개의 댓글