[프로그래머스] 시소 짝꿍 JS

문제링크

 

풀이 방법

쌍의 개수를 구하는 것이었기에 조합을 떠올렸습니다
두 수의 2,3,4배수 값 중 하나라도 같으면 됩니다
나올 수 있는 비율은 2:3, 2:4, 3:2, 3:4, 4:2, 4:3
쌍이기 때문에 모든 비율에 대한 수를 찾지 않고 특정수를 기준으로 큰 수만 구하기로 했습니다

 

function solution(weights) {
    
    //쌍을 구하라 = 조합
    //두 수의 2,3,4배수 값이 하나라도 같으면 됨
    //나올 수 있는 비율은 2:3, 2:4, 3:2, 3:4, 4:2, 4:3이지만
    //쌍이기 때문에 하나의 수를 기준으로 큰 수만 구하기
    
    let cnt=0;
    const table = {};
    
    weights.forEach(w=>{
        table[w]=(table[w]||0)+1;
    });
    


    for(let key in table){
        
        //같은 수가 있는 경우 nC2
        cnt+=table[key]*(table[key]-1)/2; 
        
        //조합이니까 모든 비율에 대해 구하지 않고 더 큰 수에 대해서만 구함
        //2배수가 있는 경우 4a=2b
        if(table[key*2]) 
            cnt+=table[key*2]*table[key];
        
        //3a=2b
        if(table[key/3*2])
            cnt+=table[key/3*2]*table[key];
        //4a=3b
        if(table[key/4*3])
            cnt+=table[key/4*3]*table[key];
    }
    
    
    return cnt;
    
}