[C++] 프로그래머스 - 메뉴 리뉴얼

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

코딩테스트 연습 - 메뉴 리뉴얼

<풀이>

1. 주어진 메뉴의 개수만큼 가장 많이 선택된 메뉴 조합을 구한다.

2. 메뉴 조합을 사전순으로 정렬하고 반환한다.

 

<해법>

1. 주문한 메뉴 조합에서 코스 개수에 따른 조합 개수 구하는 방법

=> 주문한 메뉴 조합이 "ABCDE"이고 만들려고하는 코스 개수가 "2"개라면 만들 수 있는 모든 코스는 AB, AC, AD, AE, BC, BD, ... 이 있습니다. 이렇게 만들어진 코스 이름과 코스 개수를 저장해두어야 합니다. 따라서, 저는 map을 사용하여 만들어진 코스 이름과 코스 개수를 저장해두었습니다.

예를 들어, 메뉴조합은 "ABCDE, AB, CDE"이고 코스개수는 "2"개 라고 한다면

ABCDE -> AB, AC, AD, AE, BC, BD, ...

AB -> AB

CDE -> CD, CE, DE

메뉴조합은 위와 같이 나타낼 수 있습니다.

이를 map에 [AB : 2], [AC : 1], [AD: 1] 방식으로 저장합니다.

 

2. 가장 많이 함께 주문된 메뉴 구성 구하는 방법

=> 메뉴 구성과 개수를 map에 저장해두었습니다. map에 있는 개수를 기준으로 정렬하여야 하므로, map에 있는 데이터를 vector로 옮기고 정렬을 진행하였습니다. 이 후, 가장 많이 함께 구성된 메뉴를 answer에 넣어줍니다.

 

 

#include <iostream>
#include <string>
#include <vector>
#include <map>
#include <algorithm>
 
using namespace std;
 
map<string, int> m;
 
bool cmp(pair<string, int> a, pair<string, int> b) {
    return a.second > b.second;
}
 
void findAllCourse(string order, int size, string s, int index) {
 
    //종료조건
    if (s.length() == size) {
        map<string, int>::iterator iter;
 
        iter = m.find(s);
 
        //맵에 코스가 이미 존재하는 경우 -> 개수 + 1
        if (iter != m.end()) {
            iter->second++;
        }
        //맵에 코스가 존재하지 않는 경우 -> 맵에 삽입
        else {
            m.insert(pair<string, int>(s, 1));
        }
 
        return;
    }
 
    for (int i = index; i < order.length(); i++) {
        findAllCourse(order, size, s + order[i], i + 1);
    }
}
 
vector<string> solution(vector<string> orders, vector<int> course) {
 
    vector<string> answer;
 
    //코스 크기 만큼 모두 해보기
    for (int i = 0; i < course.size(); i++) {
 
        m.clear();
 
        for (int j = 0; j < orders.size(); j++) {
 
            //메뉴 정렬(ex. CBA -> ABC)
            sort(orders[j].begin(), orders[j].end());
 
            //주문한 조합에서 코스 개수만큼 골라서, 모든 조합 만들기 -> 맵에 저장
            if (course[i] <= orders[j].length()) {
                findAllCourse(orders[j], course[i], "", 0);
            }
        }
 
        //map -> vector 옮기기
        vector<pair<string, int>> v(m.begin(), m.end());
 
        //만들어진 조합 개수에 따라 정렬
        sort(v.begin(), v.end(), cmp);
 
        if (!v.empty()) {
            //최대 개수 크기만큼 정답에 저장
            int biggest = v[0].second;
 
            //최소 2명 이상 손님으로부터 주문된 조합이어야 하므로
            if (biggest >= 2) {
                for (int i = 0; i < v.size(); i++) {
                    if (v[i].second == biggest) {
                        answer.push_back(v[i].first);
                    }
                    else {
                        break;
                    }
                }
            }
        }
    }
 
    //정렬
    sort(answer.begin(), answer.end());
 
    //반환
    return answer;
}

 

브루트포스에 대해 알아볼 수 있는 문제였습니다.