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

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

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

import java.util.*;
class Solution {
    public int[] solution(String[] enroll, String[] referral, String[] seller, int[] amount) {
        HashMap<String, Integer> hs = new HashMap<>();
        Node person[] = new Node[enroll.length+1];
        for(int i=0; i<enroll.length+1; i++){
            person[i] = new Node();    
        } 
        enroll(enroll, hs);
        union(enroll, referral, hs, person);        
        calculate(seller, amount, hs, person);
        int[] answer = new int[enroll.length];
        for(int i=0; i<enroll.length; i++){
            answer[i] = person[i+1].result;
        }
        return answer;
    }
    
    private void calculate(String[] seller, int[] amount, HashMap<String,Integer> hs ,Node[] person){
        int sellerSize = seller.length;
        for(int i=0; i<sellerSize; i++){
        	int value = 100*amount[i];
            int remain = value * 10 / 100;
            if(remain > 0){
                 person[hs.get(seller[i])].result += (value - remain);
                 dfs(person[hs.get(seller[i])].parent, remain);
            }
            else{
                 person[hs.get(seller[i])].result += value;            
            }
        }
    }
    
    private void dfs(Node cur, int value){
    	if(cur.parent == null) {
    		return;
    	}
        int remain = value * 10 / 100;
        if(remain > 0) {
            cur.result += value - remain;
            dfs(cur.parent, remain);
        }
        else{
            cur.result += value;
        }
    }
    
    private void union(String[] enroll, String[] referral, HashMap<String,Integer> hs, Node[] person){
        int personSize = enroll.length; 
        for(int i=0; i<personSize; i++){
            if(referral[i].equals("-")){
                person[hs.get(enroll[i])].parent = person[0];
            }
            else{
                person[hs.get(enroll[i])].parent = person[hs.get(referral[i])];
            }
        }
    }
    
    private void enroll(String[] enroll, HashMap<String,Integer> hs){
        hs.put("center", 0);
        int idx = 1;
        for(String name : enroll){
            hs.put(name, idx++);
        }
    }
    
    class Node{
        Node parent;
        int result;
        int amount;
    }
    
}

우선 사람들의 이름을 HashMap을 이용해서 인덱스화하여 사용했다.

판매량이 주어지면 그 판매한 사람에서 시작하여 부모에게 10%값을 전달해야 하기 때문에

노드 클래스에는 parent와 현재 노드의 총액, 판매량을 정의했다.

 

1. 이름을 인덱스로

2. 각 노드의 부모를 등록

3. amount를 순회하면서 10%를 부모에게 전달