Commit5041074

authored

Merge pull requestneetcode-gh#328 from AgranatMarkit/main

Update 212-Word-Search-II.py

2 parents05acc27 +89f538b commit5041074Copy full SHA for 5041074

File tree

+16

-1

lines changed

+16

-1

lines changed

Lines changed: 16 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -2,14 +2,25 @@ class TrieNode:`
`2`	`2`	`def__init__(self):`
`3`	`3`	`self.children= {}`
`4`	`4`	`self.isWord=False`
	`5`	`+self.refs=0`
`5`	`6`
`6`	`7`	`defaddWord(self,word):`
`7`	`8`	`cur=self`
	`9`	`+cur.refs+=1`
`8`	`10`	`forcinword:`
`9`	`11`	`ifcnotincur.children:`
`10`	`12`	`cur.children[c]=TrieNode()`
`11`	`13`	`cur=cur.children[c]`
	`14`	`+cur.refs+=1`
`12`	`15`	`cur.isWord=True`
	`16`	`+`
	`17`	`+defremoveWord(self,word):`
	`18`	`+cur=self`
	`19`	`+cur.refs-=1`
	`20`	`+forcinword:`
	`21`	`+ifcincur.children:`
	`22`	`+cur=cur.children[c]`
	`23`	`+cur.refs-=1`
`13`	`24`
`14`	`25`
`15`	`26`	`classSolution:`
`@@ -24,14 +35,18 @@ def findWords(self, board: List[List[str]], words: List[str]) -> List[str]:`
`24`	`35`	`defdfs(r,c,node,word):`
`25`	`36`	`if (r<0orc<0or`
`26`	`37`	`r==ROWSorc==COLSor`
`27`		`-board[r][c]notinnode.childrenor (r,c)invisit):`
	`38`	`+board[r][c]notinnode.childrenor`
	`39`	`+node.children[board[r][c]].refs<1or`
	`40`	`+ (r,c)invisit):`
`28`	`41`	`return`
`29`	`42`
`30`	`43`	`visit.add((r,c))`
`31`	`44`	`node=node.children[board[r][c]]`
`32`	`45`	`word+=board[r][c]`
`33`	`46`	`ifnode.isWord:`
	`47`	`+node.isWord=False`
`34`	`48`	`res.add(word)`
	`49`	`+root.removeWord(word)`
`35`	`50`
`36`	`51`	`dfs(r+1,c,node,word)`
`37`	`52`	`dfs(r-1,c,node,word)`

Comments

(0)