Commit eee7127

authored

Create 904.py

1 parent a68dc6c commit eee7127Copy full SHA for eee7127

File tree

+21

-0

lines changed

+21

-0

lines changed

Lines changed: 21 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,21 @@`
	`1`	`+class Solution:`
	`2`	`+ def totalFruit(self, tree):`
	`3`	`+ """`
	`4`	`+ :type tree: List[int]`
	`5`	`+ :rtype: int`
	`6`	`+ """`
	`7`	`+ j = 0`
	`8`	`+ dic = {}`
	`9`	`+ maxlen = -float('inf')`
	`10`	`+ for i in range(len(tree)):`
	`11`	`+ if tree[i] not in dic:`
	`12`	`+ dic[tree[i]] = 1`
	`13`	`+ else:`
	`14`	`+ dic[tree[i]] += 1`
	`15`	`+ while len(dic) > 2:`
	`16`	`+ dic[tree[j]] -= 1`
	`17`	`+ if dic[tree[j]] == 0:`
	`18`	`+ del dic[tree[j]]`
	`19`	`+ j += 1`
	`20`	`+ maxlen = max(maxlen, i-j+1)`
	`21`	`+ return maxlen if maxlen != -float('inf') else 0`

Comments

(0)