[TIL] 241227 (프로그래머스 리스트 자르기)

·2024년 12월 26일

TIL

목록 보기
262/268
post-thumbnail

오늘 한 일

  • 알고리즘 코드카타
    • 리스트 자르기

알고리즘 코드카타

리스트 자르기

문제

정수 n과 정수 3개가 담긴 리스트 slicer 그리고 정수 여러 개가 담긴 리스트 num_list가 주어집니다. slicer에 담긴 정수를 차례대로 a, b, c라고 할 때, n에 따라 다음과 같이 num_list를 슬라이싱 하려고 합니다.

  • n = 1 : num_list의 0번 인덱스부터 b번 인덱스까지
  • n = 2 : num_list의 a번 인덱스부터 마지막 인덱스까지
  • n = 3 : num_list의 a번 인덱스부터 b번 인덱스까지
  • n = 4 : num_list의 a번 인덱스부터 b번 인덱스까지 c 간격으로

올바르게 슬라이싱한 리스트를 return하도록 solution 함수를 완성해주세요.

풀이

function solution(n, slicer, num_list) {
    const [a,b,c]=slicer
    if (n===1) return num_list.slice(0,b+1)
    else if (n===2) return num_list.slice(a)
    else if (n===3) return num_list.slice(a,b+1)
    else if (n===4) {
        let result = []
        for (let i=a; i<=b; i+=c){
            result.push(num_list[i])
        }
        return result;
    }
    
}

각 조건에 따라 if문을 설정하여 문제를 풀었다.

다른 사람의 풀이

function solution(n, slicer, num_list) {
    let [a, b, c] = [...slicer];

    switch(n) {
        case 1:
            return num_list.slice(0, b + 1);
        case 2:
            return num_list.slice(a);
        case 3:
            return num_list.slice(a, b + 1);
        case 4:
            return num_list.slice(a, b + 1).filter((_, idx) => !(idx % c));
    }
}

filter로 설정하는 방법을 처음에 생각했다가 for문으로 바꿨는데 idx를 c로 나눈 나머지를 통해 구할 수 있었다. 다음에는 switch문 사용과 함께 이 방법도 사용해봐야겠다.

profile
웹 프론트엔드 개발자

0개의 댓글