[알고리즘 || 프로그래머스] 체육복

이거 풀긴 나름 빨리 풀었는데, 

코드 정리하는 걸 배웠다.

 

* 2개 이상 가지고 있는 애 -> 오른쪽으로 전달, 

* 2개 이상 가지고 있는 애 <- 왼쪽으로 전달

 

이 두가지가 순서가 바뀌면 다른 결과가 나오기 때문에 순서를 바꾼 결과의 수를 비교해서 결과를 도출했다. 

 

function solution(n, lost, reserve) {
    const swts = Array.from({ length: n }, (v, i) => lost.includes(i + 1) ? 0 : 1).map((v, i) => reserve.includes(i + 1) ? v += 1 : v)
    const upper = (swts) => {
        for (let i = 0; i < swts.length; i++) {
            if (swts[i] === 0&& swts[i + 1] > 1) {
                    swts[i] += 1;
                    swts[i + 1] -= 1;
            }
        }
        return swts;
    }
    const lower = (swts) => {
        for (let i = 1; i < swts.length; i++) {
            if (swts[i] === 0&&swts[i - 1] > 1) {
                    swts[i] += 1;
                    swts[i - 1] -= 1;
            }
        }
        return swts;
    }
    const upperNumber = (swts) => lower(upper(swts)).filter(v => v !== 0).length;
    const lowerNumber = (swts) => upper(lower(swts)).filter(v => v !== 0).length;
    return Math.max(upperNumber([...swts]), lowerNumber([...swts]))
}

 

 

오늘도 비난좌가 도와주셨다. 햅삐

 

 

function solution(n, lost, reserve) {
    let students = Array(n).fill(1);
    // 체육복을 잃어버린 학생 처리
    lost.forEach(v => students[v-1]--);
    // 여벌 체육복을 가지고 있는 학생 처리
    reserve.forEach(v => students[v-1]++);
    
    students.forEach((v, i) => {
        // 이 학생의 바로 앞에 여별의 체육복을 가진 학생이 있으면 세비지 합니다.
        if (!v && students[i-1] === 2) {
            students[i-1]--;
            students[i]++;
        }
        else if (!v && students[i+1] === 2) {
         // 이 학생의 바로 뒤에 여별의 체육복을 가진 학생이 있으면 세비지 합니다.
            students[i+1]--;
            students[i]++;
        }
    });

    return students.filter(v => v > 0).length;
}

 

 

 

비난님 뇌 새비지 하고 싶다...