AtCoderで fmt.Scanと bufio.Scanner の読み取り差でエラーが生じる原因を知りたい

Question

前提

AtCoderで初心者向けの課題を解いています。
白昼夢 の課題を Goで実装しているのですが、
標準入力の受け取り方によって 全問AC（正答）の場合と、半数が WA（誤答）の場合に分かれます。

具体的には ① fmt.Scan で読み取ると正答となり、② bufio.Scanner で読み取ると誤答が混ざります。
なぜ後者だと誤答が紛れ込むのかが知りたいです。

尚、その際の入力パターンは不明です。

実現したいこと

• bufio.Scanner を用いて処理した場合の問題点を知りたい
• bufio.Scanner をどの様に用いれば正答を得られるのか知りたい

発生している問題・エラーメッセージ

• 後述のコードで、fmt.Scan で読み込む readA() 関数を用いると全問正答 (AC) となるが、② bufio.Scanner で読み込む readB()を用いると 19パターン中 8パターン で (WA) が紛れてしまう
• 繰り返しになるが、AtCoder側の入力パターンは不明

該当のソースコード

// 白昼夢
// https://atcoder.jp/contests/abs/tasks/arc065_a

package main

import (
	"bufio"
	"fmt"
	"os"
	"strings"
)

func main() {
	var str = readB() // readA() では正答、readB() では誤答が混ざる（AC x 11, WA x 8)

	if check(str) {
		fmt.Print("YES")
	} else {
		fmt.Print("NO")
	}
}

func check(line string) bool {
	keywords := []string{"dream", "dreamer", "erase", "eraser"}
ChallengeLoop:
	for line != "" {
		for _, k := range keywords {
			if strings.HasSuffix(line, k) {
				line = strings.TrimSuffix(line, k)
				continue ChallengeLoop
			}
		}
		return false
	}
	return true
}

func readA() string {
	var str string
	fmt.Scan(&str)
	return str
}

func readB() string {
	lines := read()
	var str = ""
	if len(lines) > 0 {
		elm := strings.Split(strings.TrimSpace(lines[0]), " ")
		str = elm[0]
	}
	return str
}

func read() []string {
	s := bufio.NewScanner(os.Stdin)
	var lines []string
	for s.Scan() {
		lines = append(lines, s.Text())
	}
	return lines
}

試したこと

• fmt.Scan は標準入力にスペースが含まれる場合、トークン分割して最初のトークンを返す仕様なので、bufio.Scannerを用いる readB()内でも、読み取った最初の１行に対し、スペースで trimした後、最初のトークンを返すようにしている
• readB() の実装を以下のように単純化しても結果は変わらなかった

func readB() string {
	s := bufio.NewScanner(os.Stdin)
	s.Scan()
	return s.Text()
}

補足情報（FW/ツールのバージョンなど）

• go 1.14

Answer 1

bufio.Scanner の初期バッファが不足していることが原因でした。以下のコードで解決しました。

func read() []string {
	buff := make([]byte, 100000)
	s := bufio.NewScanner(os.Stdin)
	s.Buffer(buff, 100000) // バッファを設定
	var lines []string
	for s.Scan() {
		lines = append(lines, s.Text())
	}
	return lines
}

Answer 2

原因は分からないですが bufio.NewScanner は繰り返しの読み込みのようなので、1個の入力だけが与えられるこの問題には合ってなさそうですね

Comments

[koooyooo]

ありがとうございます。
おっしゃるとおりで、この問題には不要ですね。原因が分からない内は `fmt.Scan` を使っておこうと思います。