139. Word Break #37
139. Word Break #37
Conversation
| func wordBreak(s string, wordDict []string) bool { | ||
| wordDictInitialsToWords := make(map[byte][]string) | ||
| for _, w := range wordDict { | ||
| wordDictInitialsToWords[w[0]] = append(wordDictInitialsToWords[w[0]], w) |
There was a problem hiding this comment.
これ、頭文字ごとに分類してから、HasPrefix のチェックしていますがなくてもいいのではないでしょうか。
状況によっては少し速くはなりますが。
There was a problem hiding this comment.
たしかに思っていたよりも速くならなさそうです。平均で高々26倍ですかね
There was a problem hiding this comment.
| ``` | ||
|
|
||
| #### 2c | ||
| - Trieを使う方法 |
There was a problem hiding this comment.
あまり時間がかかるようだったら生成AIに聞くと直してくれるでしょう。
最終的な目的は、解けるようになることで、解くことではないので、後から戻ってくるなり色々工夫してください。
| } | ||
|
|
||
| suffixes := []string{s} // used as a stack | ||
| checkedSuffixesMemo := make(map[string]struct{}) |
There was a problem hiding this comment.
checkedSuffixes でも意味が通りますし、シンプルでよいと思います。
| suffixes := []string{s} // used as a stack | ||
| checkedSuffixesMemo := make(map[string]struct{}) | ||
| for len(suffixes) > 0 { | ||
| top := suffixes[len(suffixes)-1] |
There was a problem hiding this comment.
stack の top という意味だと思うのですが、 stack から取り出した時点ですでに top ではなくなっていると思います。 suffix が入っていると思いますので、そのまま suffix という変数名にするとよいと思います。
|
|
||
| ```Go | ||
| func wordBreak(s string, wordDict []string) bool { | ||
| canBeSegmentedHead := make([]bool, len(s)+1) |
There was a problem hiding this comment.
ありがとうございます。ここはいい表現が思いついていませんでした
| top := suffixes[len(suffixes)-1] | ||
| suffixes = suffixes[:len(suffixes)-1] |
There was a problem hiding this comment.
Goだとpopがないようですが、ここでpop()という関数を自作しちゃった方が意図がわかりやすいような気もします
| if !strings.HasPrefix(top, w) { | ||
| continue | ||
| } | ||
| topSuffix := top[len(w):] |
There was a problem hiding this comment.
wordDictにある単語を除くという操作をしたことが、ちょっと読んでて時間かかってしまいました。(このまま下を読んでいくと、topSuffixがなんなのか途中で忘れて、わからなくなってくる)
個人的にはsuffixAfterRemoveとか、afterWordRemovalとかの方がわかりやすいかなと思いました。
There was a problem hiding this comment.
ご指摘ありがとうございます。自分で読み返してたしかにそうだと思いました。
|
|
||
| canBeSegmented := make([]bool, len(s)+1) | ||
| for i := 0; i < len(s); i++ { | ||
| if i != 0 && canBeSegmented[i] == false { |
| for _, child := range node.children { | ||
| if child.value == word[i] { | ||
| node = child | ||
| goto wordLoop |
There was a problem hiding this comment.
gotoはちょっとわかりにくいと思います。(foundとかのbool変数で管理した方がいいのかな)
| goto wordLoop | ||
| } | ||
| } | ||
| node.children = append(node.children, &TrieNode{word[i], []*TrieNode{}}) |
There was a problem hiding this comment.
childが見つからず新たに作った後もnodeをchildに移動しないと、正しく構築できない気がしますがどうでしょう
There was a problem hiding this comment.
はい、こちらのコードは、wordDictからTrie木を作る -> 出来上がったTrie木にsと合致するパスがあるかどうかを調べる、ということしかしていなかったです。Trie木を実装することに精一杯で問題を解くためにどう使ったら良いかが見えておらずでした
| trieRoot := InitTrie(wordDict) | ||
|
|
||
| var backtrack func(s string) bool | ||
| backtrack = func(s string) bool { |
There was a problem hiding this comment.
backtrackという関数名は、dfsなどと同様、ややわかりにくく感じました
| continue | ||
| } | ||
| topSuffix := top[len(w):] | ||
| if len(topSuffix) == 0 { |
There was a problem hiding this comment.
個人的には空文字列もstackに突っ込んでpop()した後にTrueの判定をした方が、Trueの条件が読んでいて早期にわかって安心感がある、かつfor文の中身が少なくなって見やすくなって好きですが、好みの範囲そうです。
| if canSplitAtIndex(index + len(word)) { | ||
| return true | ||
| } | ||
| memo[index+len(word)] = false |
There was a problem hiding this comment.
本当ですね。自分もレビュー時にこういうのに気づけるようになりたいです
|
|
||
| ```Go | ||
| func wordBreak(s string, wordDict []string) bool { | ||
| memo := make(map[int]bool) // falseのものだけmemoに追加 |
https://leetcode.com/problems/word-break/description/