코딩테스트/LV. 1

LV. 1 :: 제일 작은 수 제거하기

KUROMI98 2023. 1. 23. 13:55

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

function solution(arr) 
{
    var answer = [];
    var a = Math.min(...arr);
    // arr에서 가장 작은 값을 a라고 하자
    var index = arr.indexOf(a);
    // a의 index를 구한다.
    
    arr.splice(index, 1);
    // arr에서 a를 제거한다.
    
    if (arr.length == 0) {answer = [-1];}
    // 배열의 크기가 0인 경우 (배열의 원소가 한 개만 있는 경우) 답은 [-1]
    else {answer = arr;} // 0이 아닌 경우 답은 배열 그대로임
    
    return answer;
}

230916 C++

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

using namespace std;

vector<int> solution(vector<int> arr) { 
    int min = *std::min_element(arr.begin(), arr.end());
    int index = find(arr.begin(), arr.end(), min) - arr.begin();
    arr.erase(arr.begin() + index); 
    if (arr.size()==0) arr.push_back(-1);
    return arr;
}