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  }