백준 16165 < 걸그룹 마스터 준석이 > JavaScript

📍문제 링크

https://www.acmicpc.net/problem/16165

16165번: 걸그룹 마스터 준석이

 

📍알고리즘 분류

- 자료구조

- 해시를 사용한 집합과 맵 (해시 맵 또는 해시 테이블)

 

📍문제 풀이

- 주어지는 인풋 값을 어떤 형태로 저장할까? 고민이 필요

- 팀 멤버가 주어지면 그룹 이름을 출력해야 하고,

그룹 이름이 주어지면, 팀 전체 멤버를 출력해야 한다.

- { 그룹 이름: [ 팀 멤버1, ... ] } 형태의 해시 맵으로 저장하면 좋을 것 같다.

 

📍코드 (JavaScript)

const [in1, ...in2] = require('fs')
  .readFileSync('/dev/stdin')
  .toString()
  .trim()
  .split('\n');

const [N, M] = in1.split(' ').map(Number);

// 전역에 객체를 생성하고, 키값은 그룹명, 밸류값은 멤버명이 담긴 배열
const data = {};

// data 완성
for (let i = 0; i < N; i++) {
  const teamName = in2.shift();
  const memberNum = +in2.shift();
  data[teamName] = in2.splice(0, memberNum);
}

const answer = [];

// 제시되는 조건에 따라 답을 리턴하는 함수
const tester = (keyword, type) => {
  if (type) {
    for (let team in data) {
      if (data[team].includes(keyword)) return team;
    }
  }
  for (let team in data) {
  	// 팀 멤버 전체를 리턴하는 경우, 줄바꿈해서 리턴
    if (keyword === team) return data[team].sort().join('\n');
  }
};

for (let i = 0; i < M; i++) {
  const keyword = in2.shift();
  const type = +in2.shift();
  answer.push(tester(keyword, type));
}

console.log(answer.join('\n'));

 

📍리뷰

- 다음부터는 전역 변수는 한 곳에 모으자! 가독성을 위해