Commit 5758600

authored

Create Solution.py

1 parent 7719c39 commit 5758600Copy full SHA for 5758600

File tree

+17

-0

lines changed

+17

-0

lines changed

Lines changed: 17 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,17 @@`
	`1`	`+class Solution:`
	`2`	`+ def makesquare(self, matchsticks: List[int]) -> bool:`
	`3`	`+ N = len(matchsticks)`
	`4`	`+ basket, rem = divmod(sum(matchsticks), 4)`
	`5`	`+ if rem or max(matchsticks) > basket: return False`
	`6`	`+`
	`7`	`+ @lru_cache(None)`
	`8`	`+ def dfs(mask):`
	`9`	`+ if mask == 0: return 0`
	`10`	`+ for j in range(N):`
	`11`	`+ if mask & 1<<j:`
	`12`	`+ neib = dfs(mask ^ 1<<j)`
	`13`	`+ if neib >= 0 and neib + matchsticks[j] <= basket:`
	`14`	`+ return (neib + matchsticks[j]) % basket`
	`15`	`+ return -1`
	`16`	`+`
	`17`	`+ return dfs((1<<N) - 1) == 0`

Comments

(0)