AtCoder の ABC049C - 白昼夢 の問題がACできない

Question

AtCoder で ABC049C - 白昼夢 の問題がACできず困っております。
https://atcoder.jp/contests/abs/tasks/arc065_a

いくつかの文字列で試しましたが、全て問題なく動作します。
テストケースが公開されていないので何が問題なのか分かりず、ご助力いただけますと幸いです。

// https://atcoder.jp/contests/abs/tasks/arc065_a
func main() {
	fmt.Print(f(sc))
}

func f(sc *bufio.Scanner) string {
	sc.Split(bufio.ScanWords)

	S := nextLine(sc)

	var s string
	s = S
	for len(s) > 0 {
		found := false
		for _, v := range []string{"dream", "dreamer", "erase", "eraser"} {
			if strings.HasSuffix(s, v) {
				s = strings.TrimSuffix(s, v)
				found = true
				break
			}
		}
		if !found {
			break
		}
	}

	if len(s) == 0 {
		return "YES"
	}

	return "NO"
}

var sc = bufio.NewScanner(os.Stdin)

func nextLine(sc *bufio.Scanner) string {
	sc.Scan()
	return sc.Text()
}

Comments

[Zuishin]

※ 削除

[melian]

※ 削除

Answer 1

bufio.Scanner の内部バッファの容量はデフォルトで 64 kb です。そのため、65537 バイト以上の文字列が入力されると 65537 バイト以降は切り捨てられます。(sc.Scan() が false を返します) なので、内部バッファを充分な容量を持つスライスに置き換えます。

func nextLine(sc *bufio.Scanner) string {
	const maxLen int = 100001  // 1 ≦ ∣S∣ ≦ 10^5
	buf := make([]byte, 0, maxLen)
	sc.Buffer(buf, maxLen)
	sc.Scan()
	return sc.Text()
}

Comments

[atcoder-beginer]

ありがとうございます！
いただいた内容で再トライしてみたところ、ACできました！
入力の読み取りは共通関数化しており今まで問題なかったため、メインロジックばかり見ていました、、
理解が足りておらず自分では一生ハマりっぱなしになるとことでした、、
勉強になりました、ありがとうございます！