[프로그래머스] 코딩테스트 Lv.1 추억점수

 

 

 

 

 

 

 

 

 

자바

알아둬야 할 개념

1. HashMap : HashMap은 Map 인터페이스를 상속하고 있기에 Map의 성질을 그대로 가지고 있고, 
                       Map은 키와 값으로 구성된 Entry객체를 저장하는 구조를 가지고 있는 자료구조이다.

HashMap

 

2. int[숫자] answer : 숫자 길이만큼의 정수 값이 들어가 있는 배열

 

import java.util.*;
class Solution {
    public int[] solution(String[] name, int[] yearning, String[][] photo) {
        int[] answer = new int[photo.length] ;
        HashMap<String, Integer> map = new HashMap<>();
        for(int i = 0; i < name.length; i++){
            map.put(name[i], yearning[i]);
            };
        for(int v = 0; v < photo.length; v++ ){
            for(int k = 0; k < photo[v].length; k++){
                 if(map.containsKey(photo[v][k])){
                   answer[v] += map.get(photo[v][k]);  
                 };
            };
        };
        System.out.println(map);
        return answer;
    }
}

 

파이썬

def solution(name, yearning, photo):
    answer = []
    for k in photo:
        num = 0
        for v in k:
            if v in name:
                num += yearning[name.index(v)]
        answer.append(num)
        
    return answer