Code KATA/알고리즘 코드카타

[2025.01.08] 추억 점수

iiblueblue 2025. 1. 8. 09:35

문제 설명

사진들을 보며 추억에 젖어 있던 루는 사진별로 추억 점수를 매길려고 합니다. 사진 속에 나오는 인물의 그리움 점수를 모두 합산한 값이 해당 사진의 추억 점수가 됩니다. 예를 들어 사진 속 인물의 이름이 ["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 yearning photo result 설명
["may", "kein", "kain", radi"] [5, 10, 1, 3] [["may", "kein", "kain", "radi"],["may", "kein", "brin", "deny"],["kon", "kain", "may", "coni"]] [19, 15, 6] 첫 번째 사진 속 "may", "kein", "kain", "radi"의 그리움 점수를 합치면 19(5 + 10 + 1 + 3)점 입니다. 두 번째 사진 속 그리워하는 사람들인 "may"와 "kein"의 그리움 점수를 합치면 15(5 + 10)점입니다. 세 번째 사진의 경우 "kain"과 "may"만 그리워하므로 둘의 그리움 점수를 합한 6(1 + 5)점이 사진의 추억 점수입니다. 따라서 [19, 15, 6]을 반환합니다.
["kali", "mari", "don"] [11, 1, 55] [["kali", "mari", "don"], ["pony", "tom", "teddy"], ["con", "mona", "don"]] [67, 0, 55] 첫 번째 사진 속 그리워하는 사람들인 "kali", "mari", "don"의 그리움 점수를 합치면 67(11 + 1 + 55)점입니다. 두 번째 사진 속엔 그리워하는 인물이 없으므로 0점입니다. 세 번째 사진 속 그리워하는 사람은 "don"만 있으므로 55점입니다. 따라서 [67, 0, 55]를 반환합니다.
["may", "kein", "kain", "radi"] [5, 10, 1, 3] [["may"],["kein", "deny", "may"], ["kon", "coni"]] [5, 15, 0] 설명 생략

 

 

문제 풀이

풀이 언어 : C++

#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;
    int yearning_sum=0; // 사진 하나의 그리움 점수 합
    
    // 사진 별로 반복(행 별로)
    for(int row=0; row<photo.size(); row++)
    {
        yearning_sum=0; // 그리움 점수 합 초기화
        
        // name에 있는 이름 수 만큼 반복
        for(int i=0; i<name.size(); i++)
        {
            // 사진 속에 name안에 사람이 있다면
            if(find(photo[row].begin(), photo[row].end(), name[i])!=photo[row].end()) 
            {
                yearning_sum+=yearning[i]; // 그리움 점수 누적
            }
        }
        
        answer.push_back(yearning_sum); // answer에 그리움 점수 합 저장
    }
    return answer;
}

2차원 배열 photo는 한 행마다 사진 한장을 의미한다. 그렇기 때문에 photo의 한 행을 뽑아와서 name에 있는 이름들이 사진에 있는지 확인하고 사진에 있다면 사진에 있는 이름들의 그리움 점수를 더해주면 된다. 사진 하나마다 그리움 점수의 합을 변수에 저장하고 그 값을 answer 벡터에 하나씩 쌓아줄 것이다.

 

우선 사진 한장씩 보기 위해 photo의 row크기만큼 돌아가는 배열을 만든다. 그리고 한 사진의 그리움 점수의 합을 저장할 yearning_sum 변수를 만들어서 반복문이 시작할 때 0으로 초기화 해준다. 새로운 사진으로 넘기고 나서는 0으로 다시 초기화 되어야 하기 때문이다.

 

안쪽 반복문에서는 name의 길이만큼 반복문을 돌린다. find 함수를 이용하여 photo[row] 벡터에, 즉 사진에 name[i]가 있는지 확인한다. 만약 이름이 있다면 같은 위치에 있는 i번째 yearning 벡터에 들어있는 요소를 yearning_sum에 더해준다. name에 있는 모든 이름이 대조가 끝나면 반복문 밖으로 나와 answer에 yearning_sum을 저장한다. 이렇게 모든 사진을 돌면 answer이 완성된다.

 

 

문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/176963

 

프로그래머스

SW개발자를 위한 평가, 교육, 채용까지 Total Solution을 제공하는 개발자 성장을 위한 베이스캠프

programmers.co.kr