[프로그래머스] 코딩테스트 연습 : 다단계 칫솔 판매 - C++[CPP]

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

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

---

⊙ 문제

⊙ 제한사항

⊙ 입출력 예

⊙ 입출력 예 설명

---

 

⊙ 문제 접근 과정

 

map을 사용하여 문제를 풀었다.

이익에 대해 계산해주는 재귀 함수를 따로 만들어주어 판 사람의 수만큼 반복해준다.

 

자세한 설명은 코드 줄마다 전부 주석을 달아놨으니 참고하면 될 것 같다!

---

 

⊙ 문제 풀이

 

#include <iostream>
#include <vector>
#include <map>

using namespace std;
#define MAX 10001

map<string,int> m;
int parent[MAX];
int profits[MAX];

void divideProfit(int indx, int profit) {
    if(indx == -1) return; //최상위 포식자면 종료

    int stolen = profit/10; //10%를 상위 포식자에게 준다
    profits[indx] += profit - stolen; //내 이익

    if(!stolen) return; //줄 profit이 없다면 stop

    divideProfit(parent[indx], stolen); //재귀
}

vector<int> solution(vector<string> enroll, vector<string> referral, vector<string> seller, vector<int> amount) {
    vector<int> answer;

    for (int i = 0; i < enroll.size(); i++)
        m[enroll[i]] = i; //name, index 저장

    for (int i = 0; i < referral.size(); i++) {
        if (referral[i] == "-") parent[i] = -1; //최상위 포식자
        else parent[i] = m[referral[i]]; //상위 포식자 저장
    }
 
    for (int i = 0; i < seller.size(); i++)
        divideProfit(m[seller[i]], amount[i] * 100); // 이익 계산(칫솔 개당 100원)
 
    for (int i = 0; i < enroll.size(); i++)
        answer.push_back(profits[i]); //결과
 
    return answer;
}

---

⊙ 마무리

 

 

NONE

 

좋아요는 로그인하지 않아도 누를 수 있습니다!