最长重复字符串题解 golang

最长重复字符串题解

package main

import (
	"fmt"
	"strings"
)

type Index map[int]int

type Counter map[string]Index

var c = make(Counter)

func setRecord(match string, index int) {
	i, ok := c[match]
	if !ok {
		i = make(Index)
		c[match] = i
		return
	}

	i[index]++
}

func filterOverlap() {
	var keys []string
	for k := range c {
		keys = append(keys, k)
	}

	for _, xkey := range keys {
		for k := range c {
			if strings.HasPrefix(xkey, k) && xkey != k {
				// fmt.Printf("delete key=%v\n", k)
				delete(c, k)
				continue
			}
		}
	}
}

func calc(sample []byte) {
	var i, n, xi, xj, yi, yj int
	n = len(sample)

	var a, b string

	for i = 0; i < n; i++ {
		for xi = i; xi < n-1; xi++ {
			for xj = xi + 1; xj < n; xj++ {
				for yi = xj + 1; yi < n; yi++ {

					yj = yi + (xj - xi)
					a = string(sample[xi:xj])
					b = string(sample[yi:yj])
					if a == b {
						setRecord(a, xi)
						setRecord(b, yi)
					}

				}
			}
		}
	}
}

func main() {
	var s = "12342342341234"

	calc([]byte(s))

	filterOverlap()

	for k, v := range c {
		fmt.Printf("%v:%v\n", k, len(v))
	}
}