2022-01-23:力扣425,单词方块。 给定一个单词集合 (没有重复),找出其中所有的 单词方块 。 一个单词序列形成了一个有效的单词方块的意思是指从第 k 行和第 k 列 (0 ≤ k < max(行数, 列数)) 来看都是相同的字符串。 例如,单词序列 ["ball","area","lead","lady"] 形成了一个单词方块,因为每个单词从水平方向看和从竖直方向看都是相同的。 注意: 单词个数大于等于 1 且不超过 500。 所有的单词长度都相同。 单词长度大于等于 1 且不超过 5。 每个单词只包含小写英文字母 a-z。
答案2022-01-23:
递归。所有单词的所有前缀都会成为key。
注意!课上介绍题目设定的时候,有一点点小错 题目描述如下: 给定n个字符串,并且每个字符串长度一定是n,请组成单词方阵,比如: 给定4个字符串,长度都是4,["ball","area","lead","lady"] 可以组成如下的方阵: b a l l a r e a l e a d l a d y 什么叫单词方阵?如上的方阵可以看到, 第1行和第1列都是"ball",第2行和第2列都是"area",第3行和第3列都是"lead",第4行和第4列都是"lady" 所以如果有N个单词,单词方阵是指: 一个NN的二维矩阵,并且i行和i列都是某个单词,不要求全部N个单词都在这个方阵里。 请返回所有可能的单词方阵。 示例: 输入: words = ["abat","baba","atan","atal"] 输出: [["baba","abat","baba","atal"],["baba","abat","baba","atan"]] 解释: 可以看到输出里,有两个链表,代表两个单词方阵 第一个如下: b a b a a b a t b a b a a t a l 这个方阵里没有atan,因为不要求全部单词都在方阵里 第二个如下: b a b a a b a t b a b a a t a n 这个方阵里没有atal,因为不要求全部单词都在方阵里 课上说的是:一个NN的二维矩阵,并且i行和i列都是某个单词,要求全部N个单词都在这个方阵里 原题说的是:一个N*N的二维矩阵,并且i行和i列都是某个单词,不要求全部N个单词都在这个方阵里 讲的过程没错,但是介绍题意的时候,这里失误了
代码用golang编写。代码如下:
package main
import "fmt"
func main() {
ret := wordSquares([]string{"abat", "baba", "atan", "atal"})
fmt.Println(ret)
}
func wordSquares(words []string) [][]string {
n := len(words[0])
// 所有单词,所有前缀字符串,都会成为key!
map0 := make(map[string][]string)
for _, word := range words {
for end := 0; end <= n; end++ {
prefix := word[0:end]
map0[prefix] = append(map0[prefix], word)
}
}
ans := make([][]string, 0)
process(0, n, map0, &([]string{}), &ans)
return ans
}
// i, 当前填到第i号单词,从0开始,填到n-1
// map, 前缀所拥有的单词
// path, 之前填过的单词, 0...i-1填过的
// ans, 收集答案
func process(i, n int, map0 map[string][]string, path0 *[]string, ans *[][]string) {
if i == n {
pathc := make([]string, len(*path0))
copy(pathc, *path0)
*ans = append(*ans, pathc)
} else {
// 把限制求出来,前缀的限制!
builder := make([]byte, 0)
for _, pre := range *path0 {
builder = append(builder, pre[i])
}
prefix := string(builder)
if _, ok := map0[prefix]; ok {
for _, next := range map0[prefix] {
*path0 = append(*path0, next)
process(i+1, n, map0, path0, ans)
*path0 = (*path0)[0 : len(*path0)-1]
}
}
}
}
执行结果如下: