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

문제

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

프로그래머스

 

문제 분석

 

그림과 같이 판매원은 발생한 이익의 10% 를 자신을 조직에 참여시킨 추천인에게 배분하고 남은 돈을 갖는다.

 

판매원 배열 , 추천자 배열 , 판매자 배열, 판매 개수 배열들이 주어졌을 때 최종적으로 판매원들이 어느정도의 수익금을 갖는지 계산하는 문제이다.

 

 

문제 접근

 

판매원 사이에 트리 관계가 있다. 

그래서 판매원을 하나의 class 로 만들어 문제를 풀어볼까 생각했지만 HashMap 을 잘 이용하면 굳이 필요 없을 것 같다는 생각이 들었다.

<판매원 이름 , 추천인 이름> ,  <판매원 이름, 총 수익> 이렇게 2개의 Map 을 잘 사용하면 문제를 풀 수 있을 거 같다는 판단이 들었다.

 

주어진 정보들로 우선 Map 을 구성하고

판매원의 수익을 계산할 때는 판매금의 0.9 만 수익에 추가하고 Map을 통해 자신의 추천인을 찾아 그 추천인에게 0.1을 추가하는 것을 추천인이 없는 판매자에 도달할 때 까지 반복하여 계산하였다.

 

코드

import java.util.*;

class Solution {
    public int[] solution(String[] enroll, String[] referral, String[] seller, int[] amount) {
        Map<String, String> referralMap = new HashMap<>();
        Map<String, Integer> profitMap = new HashMap<>();

        for (int i = 0; i < enroll.length; i++) {
            referralMap.put(enroll[i], referral[i]);
            profitMap.put(enroll[i], 0);
        }

        for (int i = 0; i < seller.length; i++) {
            String name = seller[i];
            int profit = amount[i] * 100;

            while (referralMap.get(name) != null && profit != 0) {
                profitMap.put(name, profitMap.get(name) + (profit - (int) (profit * 0.1)));

                name = referralMap.get(name);
                profit = (int) (profit * 0.1);
            }
        }

        int[] result = new int[enroll.length];

        for (int i = 0; i < enroll.length; i++) {
            result[i] = profitMap.get(enroll[i]);
        }

        return result;
    }
}