[프로그래머스 / C++] 추억 점수

문제

추억 점수 : 문제 링크

문제 분석

• 사진들을 보며 추억에 젖어 있던 루는 사진별로 추억 점수를 매길려고 한다. 사진 속에 나오는 인물의 그리움 점수를 모두 합산한 값이 해당 사진의 추억 점수가 된다. 예를 들어 사진 속 인물의 이름이 ["may", "kein", "kain"]이고 각 인물의 그리움 점수가 [5점, 10점, 1점]일 때 해당 사진의 추억 점수는 16(5 + 10 + 1)점이 된다. 다른 사진 속 인물의 이름이 ["kali", "mari", "don", "tony"]이고 ["kali", "mari", "don"]의 그리움 점수가 각각 [11점, 1점, 55점]]이고, "tony"는 그리움 점수가 없을 때, 이 사진의 추억 점수는 3명의 그리움 점수를 합한 67(11 + 1 + 55)점이다.

• 그리워하는 사람의 이름을 담은 문자열 배열 name, 각 사람별 그리움 점수를 담은 정수 배열 yearning, 각 사진에 찍힌 인물의 이름을 담은 이차원 문자열 배열 photo가 매개변수로 주어질 때, 사진들의 추억 점수를 photo에 주어진 순서대로 배열에 담아 return하는 solution 함수를 완성

• 제한 사항

• 3 ≤ name의 길이 = yearning의 길이≤ 100
• 3 ≤ name의 원소의 길이 ≤ 7
• name의 원소들은 알파벳 소문자로만 이루어져 있다.
• name에는 중복된 값이 들어가지 않는다.
• 1 ≤ yearning[i] ≤ 100
• yearning[i]는 i번째 사람의 그리움 점수이다.
• 3 ≤ photo의 길이 ≤ 100
• 1 ≤ photo[i]의 길이 ≤ 100
• 3 ≤ photo[i]의 원소(문자열)의 길이 ≤ 7
• photo[i]의 원소들은 알파벳 소문자로만 이루어져 있다.
• photo[i]의 원소들은 중복된 값이 들어가지 않는다.

• name 내에 photo의 원소가 존재하는지 확인하기 위해 find() 함수를 사용하므로 algorithm 헤더를 include
• 사진들의 추억 점수를 저장할 정수형 벡터 answer을 초기화. 2중 for loop를 통해 photo의 모든 원소를 순환하고, 각 사진의 추억 점수를 임시로 저장할 정수형 변수 score를 0으로 초기화. if문과 find() 함수를 사용하여 name에 photo의 해당 인덱스의 원소가 존재하는지 확인하고 있다면, yearning 내 동일한 인덱스의 값을 score에 더하고 저장하고 내부 loop 탈출 후 score를 answer에 저장. loop 탈출 후 최종적으로 저장된 answer을 return

1. vector내에 해당 원소가 존재하는지 확인
=> find(v.begin(), v.end(), 찾을 대상)
1) return 값이 v.end()인 경우 => 해당 원소가 존재하지 않음
2) return 값이 v.end()가 아닌 경우 => 해당 원소가 존재함
2. vector내에서 해당 원소가 위치하는 인덱스 찾기
=> find(v.begin(), v.end(), 찾을 대상) - v.begin

풀이

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

using namespace std;

vector<int> solution(vector<string> name, vector<int> yearning, vector<vector<string>> photo) {
    vector<int> answer;
    
    for(int i = 0; i < photo.size(); ++i) {
        int score = 0;
        for(int j = 0; j < photo[i].size(); ++j) {
            if(find(name.begin(), name.end(), photo[i][j]) != name.end()) {
                score += yearning[find(name.begin(), name.end(), photo[i][j]) - name.begin()];
            }
        }
        answer.push_back(score);
    }
    return answer;
}