[백준] 14425 문자열 집합 JavaScript

혅·2024년 4월 11일

JavaScript

백준, 프로그래머스 -JavaScript-

목록 보기

88/215

문제

총 N개의 문자열로 이루어진 집합 S가 주어진다.

입력으로 주어지는 M개의 문자열 중에서 집합 S에 포함되어 있는 것이 총 몇 개인지 구하는 프로그램을 작성하시오.

입력

첫째 줄에 문자열의 개수 N과 M (1 ≤ N ≤ 10,000, 1 ≤ M ≤ 10,000)이 주어진다.

다음 N개의 줄에는 집합 S에 포함되어 있는 문자열들이 주어진다.

다음 M개의 줄에는 검사해야 하는 문자열들이 주어진다.

입력으로 주어지는 문자열은 알파벳 소문자로만 이루어져 있으며, 길이는 500을 넘지 않는다. 집합 S에 같은 문자열이 여러 번 주어지는 경우는 없다.

출력

첫째 줄에 M개의 문자열 중에 총 몇 개가 집합 S에 포함되어 있는지 출력한다.

예제 입력

5 11
baekjoononlinejudge
startlink
codeplus
sundaycoding
codingsh
baekjoon
codeplus
codeminus
startlink
starlink
sundaycoding
codingsh
codinghs
sondaycoding
startrink
icerink

예제 출력

내가 했던 풀이 방법

N과 M을 공백을 기준으로 해 분할해 0번 index를 N으로 1번 index를 M으로 한다.
두 번째 줄부터 입력받은 모든 문자열을 배열에 저장한 뒤 trim을 이용해 공백을 제거해준다.
2번에서 저장한 배열을 0부터 N-1까지 잘라준 배열과 잘라지지 않은 배열을 각각 N과 M에 저장한다. 이때, N은 set으로 변환해준다. (배열의 indexOf는 효율이 낮기 때문에 Set으로 변환하여 has를 이용해 효율을 높여주고자 함 (4번참고))
M 배열을 순환하면서 N에 해당 요소가 있는 경우 answer를 1씩 증가시켜준다.

코드

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

for (let i in input) {
  input[i] = input[i].trim();
}

let N = input.splice(0, parseInt(n_m.split(' ')[0].trim()));
N = new Set(N);
let M = input.splice(0);

let answer = 0;
for (let i in M) {
  if (N.has(M[i])) {
    answer++;
  }
}

console.log(answer);