github.com/d-tsuji/suffixarray@v0.0.0-20200625031310-5b0c40604e73/cmd/main/main.go (about) 1 package main 2 3 import ( 4 "bufio" 5 "fmt" 6 "os" 7 8 "github.com/d-tsuji/suffixarray" 9 ) 10 11 func main() { 12 r := bufio.NewReader(os.Stdin) 13 w := bufio.NewWriter(os.Stdout) 14 defer w.Flush() 15 16 var s string 17 var q int 18 19 fmt.Fscan(r, &s) 20 fmt.Fscan(r, &q) 21 22 index := suffixarray.New(s) 23 24 for i := 0; i < q; i++ { 25 var p string 26 fmt.Fscan(r, &p) 27 28 res := index.LookupAll(p) 29 if len(res) > 0 { 30 fmt.Fprintln(w, 1) 31 } else { 32 fmt.Fprintln(w, 0) 33 } 34 } 35 }