[백준 / node.js] 1152번: 단어의 개수

출처: 백준

1152번: 단어의 개수

문제

영어 대소문자와 띄어쓰기만으로 이루어진 문자열이 주어진다. 이 문자열에는 몇 개의 단어가 있을까? 이를 구하는 프로그램을 작성하시오. 단, 한 단어가 여러 번 등장하면 등장한 횟수만큼 모두 세어야 한다.

입력

첫 줄에 영어 대소문자와 띄어쓰기로 이루어진 문자열이 주어진다. 이 문자열의 길이는 1,000,000을 넘지 않는다. 단어는 띄어쓰기 한 개로 구분되며, 공백이 연속해서 나오는 경우는 없다. 또한 문자열의 앞과 뒤에는 공백이 있을 수도 있다.

출력

첫째 줄에 단어의 개수를 출력한다.

예제 입력 1

The Curious Case of Benjamin Button

예제 출력 1

6

예제 입력 2

Mazatneunde Wae Teullyeoyo

예제 출력 2

3

예제 입력 3

Teullinika Teullyeotzi

예제 출력 3

2

내 코드 1

const readline = require("readline");
const rl = readline.createInterface({
  input: process.stdin,
  output: process.stdout,
});

let input = [];

rl.on("line", function (line) {
  input = line.split("");
  rl.close();
}).on("close", function () {
  let count = 0;

  for (let i = 0; i < input.length; i++) {
    if (input[i] === " ") count++;
  }
  if (input[0] === " ") count--;

  if (input[input.length - 1] === " ") count--;

  console.log(count + 1);
  process.exit();
});

내 코드 2

const readline = require("readline");
const rl = readline.createInterface({
  input: process.stdin,
  output: process.stdout,
});

rl.on("line", function (line) {
  let word = [];
  word = line.trim().split(" ");
  if (word[0] === "") {
    console.log(0);
  } else console.log(word.length);
  rl.close();
}).on("close", function () {
  process.exit();
});

결과

> The Curious Case of Benjamin Button 6

>  Mazatneunde Wae Teullyeoyo 3

> Teullinika Teullyeotzi  2

 

정답률이 29%라서 쉽지 않겠다 싶었는데 생각보다 너무 쉬워서 왜 이게 29%지? 싶었다. 

주의해야 할 점이 있었는데 코드 1을 작성할 땐 몰랐는데 코드 2를 작성하면서 알게 되었다.

코드 1에서는 받은 문장을 공백을 기준으로 나눴다.

 

예제 2번의 ' Mazatneunde Wae Teullyeoyo' 같은 경우

[ ' ', 'M', 'a', 'z', 'a', 't', 'n', 'e', 'u', 'n', 'd', 'e', ' ', 'W', 'a', 'e', ' ', 'T', 'e', 'u', 'l', 'l', 'y', 'e', 'o', 'y', 'o' ]

이렇게 되는데 앞에 공백이 있는 것에 주의해야 한다.

이 배열을 돌면서 공백 ' '을 세서 카운트를 1씩 올렸다.

그리고 맨 앞 혹은 맨 뒤에 ' ' 이 있으면 카운트를 1씩 내렸다.

 

코드 2에서는 trim() 메서드를 사용했다.

word 배열에 line.trim().split(' ');을 하면

word에는

[ 'Mazatneunde', 'Wae', 'Teullyeoyo' ]가 된다.

여기서 이 배열의 길이만 세주면 되는데 주의할 점이 있다.

 

코드 1에서나 코드 2에서나 주의해야 할 점은 아무것도 입력받지 않았을 경우이다.

아무것도 입력받지 않아도 공백이 배열에 추가가 되어 1이 출력된다.

그래서 배열의 첫 번째 요소가 공백일 때의 조건을 추가해 주어야 한다.