NodeJS 빠른 입출력

수정

문자열을 공백 기준으로 나눠주는 IO 방식인데, V8이 문자열의 slice를 특수 처리하기 때문에 성능 향상을 보일 수 있다.

사용법

// 입력: kiwiyou 123
const io = new IO()
const a = io.token()
const b = io.number()
io.print(a)
io.byte(10)
io.print(b)
// 출력:
// kiwiyou
// 123

코드

function IO() {
  const { readFileSync, writeSync } = require('node:fs')
  const stdin = readFileSync(0)
  const text = stdin.toString('ascii')
  let buffer = ''
  process.on('exit', () => this.flush())
  let i = 0
  this.white = () => {
    while (stdin[i] <= 32) i++
  }
  this.token = () => {
    this.white()
    const s = i
    while (stdin[i] > 32) i++
    return text.slice(s, i)
  }
  this.number = () => {
    return this.token() | 0
  }
  this.print = (v) => {
    buffer += v
    if (buffer.length > 1 << 19) this.flush()
  }
  this.flush = () => {
    writeSync(1, buffer, 'ascii')
    buffer = ''
  }
  return this
}

(2025-05-17 수정) 현재는 입력만 남기고 number을 최적화한 아래 코드를 사용하고 있습니다.

function IO() {
  const { readFileSync, writeSync } = require('node:fs')
  this.writeSync = writeSync
  this.stdin = readFileSync(0)
  this.text = this.stdin.toString('ascii')
  this.i = 0
  this.white = () => {
    while (this.stdin[this.i] <= 32) this.i++
  }
  this.token = () => {
    this.white()
    const s = this.i
    while (this.stdin[this.i] > 32) this.i++
    return this.text.slice(s, this.i)
  }
  this.number = () => {
    this.white()
    let v = 0
    const neg = this.stdin[this.i] === 45
    if (neg) this.i++
    while (this.stdin[this.i] >= 48) v = v * 10 + (this.stdin[this.i++] - 48)
    return neg ? -v : v
  }
}