프로그래머스

[프로그래머스_Level1] 그리움 점수

빙수빈수 2023. 4. 25. 13:26

[문제]

 사진들을 보며 추억에 젖어 있던 루는 사진별로 추억 점수를 매길려고 합니다. 사진 속에 나오는 인물의 그리움 점수를 모두 합산한 값이 해당 사진의 추억 점수가 됩니다. 예를 들어 사진 속 인물의 이름이 ["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]의 원소들은 중복된 값이 들어가지 않습니다.

 

[코드]

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 i=0;i<photo.length;i++) {
	    	  int sum=0;
	    	  for(int j=0;j<photo[i].length;j++) {
	    		  if(map.containsKey(photo[i][j])) // 그리움 점수가 있는 사람만 점수 누적
	    			  sum+=map.get(photo[i][j]);
	    	  }
	    	  answer[i]=sum;
	      }
	      
	      return answer;
	}

 

[고찰]

 이번 문제는 HashMap을 사용하여 풀 수 있었다. 각 사람마다 그리움 점수를 HashMap에 저장한 후 그리움 점수가 있는 사람만 점수를 누적해주었다.