[프로그래머스]다단계 칫솔 판매 -JAVA

[프로그래머스]다단계 칫솔 판매

programmers.co.kr/learn/courses/30/lessons/77486

코딩테스트 연습 - 다단계 칫솔 판매

풀이

이 문제는 올해 프로그래머스 Dev-Matching 문제라고 한다. 문제에서 주어진 조건대로 따라가며 풀었고 별다른 자료구조를 사용하지 않고 그리디하게 풀었다.

 

조건에 맞게 반복문을 돌리면서 현재 판매원이 가져갈 금액을 더해주고, 금액을 현재 금액의 10퍼센트로 바꾸어 주면서 연산을 했다. 

 

테스트 코드는 모두 통과되지만, 걸리는 시간을 보니 그리 효율적인 코드는 아닌 것 같다. 더 효율적인 다른 방법이 없는지 고민해 보아야 겠다.

 

코드

import java.util.*;
 
class Solution {
    public int[] solution(String[] enroll, String[] referral, String[] seller, int[] amount) {
        int[] result = new int[enroll.length];
        
        for(int i = 0; i < seller.length; i++) {
            int cost = amount[i] * 100;
            String currentSeller = seller[i];
            
            while(!currentSeller.equals("-")) {
                int idx = 0;  //enroll index를 찾음
                for(int j = 0; j < enroll.length; j++) {
                    if(currentSeller.equals(enroll[j])) {
                        idx = j;
                        break;
                    }
                }
                
                String parent = referral[idx]; // 상위 판매원
                double tenPercent = cost * 0.1; // 현재 급액의 10프로
                int sellerCost = cost - (int) tenPercent; //현재 판매원이 가져가는 금액
                result[idx] += sellerCost;
                
                currentSeller = parent;
                cost = (int) tenPercent;
            }
            
        }
        return result;
    }
}