[C++] 프로그래머스 커피 심부름

문제설명

팀의 막내인 철수는 아메리카노와 카페 라테만 판매하는
카페에서 팀원들의 커피를 사려고 합니다.
아메리카노와 카페 라테의 가격은 차가운 것과 뜨거운 것 상관없이 각각 4500, 5000원입니다.
각 팀원에게 마실 메뉴를 적어달라고 하였고,
그 중에서 메뉴만 적은 팀원의 것은 차가운 것으로 통일하고
"아무거나"를 적은 팀원의 것은 차가운 아메리카노로 통일하기로 하였습니다.

각 직원이 적은 메뉴가 문자열 배열 order로 주어질 때,
카페에서 결제하게 될 금액을 return 하는 solution 함수를 작성해주세요.
order의 원소는 아래의 것들만 들어오고, 각각의 의미는 다음과 같습니다.

order의 원소	의미
“iceamericano”, “americanoice”	차가운 아메리카노
“hotamericano”, “americanohot”	따뜻한 아메리카노
“icecafelatte”, “cafelatteice”	차가운 카페 라테
“hotcafelatte”, “cafelattehot”	따뜻한 카페 라테
“americano”	아메리카노
“cafelatte”	카페 라테
“anything”	아무거나

제한사항

• 1 ≤ order의 길이 ≤ 1,000

입출력 예

order	result
[“cafelatte”, “americanoice”, “hotcafelatte”, “anything”]	19000
[“americanoice”, “americano”, “iceamericano”]	13500

풀이

order 배열의 원소에서 메뉴를 찾아내어 합산 금액을 return하는 구현 문제입니다. find 함수를 이용하여 메뉴를 찾고 합산하는 로직을 구현하여 문제를 풀 수 있었습니다.

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

int solution(vector<string> order) {
    int answer = 0;
    
    for(int i = 0; i < order.size(); i++)
    {
        if(order[i].find("anything") != string::npos || order[i].find("americano") != string::npos)
        {
            answer += 4500;
            continue;
        }
        
        if(order[i].find("cafelatte") != string::npos)
        {
            answer += 5000;
            continue;
        }
    }
    
    return answer;
}

결과