Commit f6b8acf

committed

Time: 80 ms (94.77%), Space: 14 MB (62.21%) - LeetHub

1 parent f7abe5a commit f6b8acfCopy full SHA for f6b8acf

File tree

+10

-0

lines changed

+10

-0

lines changed

Lines changed: 10 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,10 @@`
	`1`	`+class Solution:`
	`2`	`+ def threeSumMulti(self, A: List[int], target: int) -> int:`
	`3`	`+ c = collections.Counter(A)`
	`4`	`+ res = 0`
	`5`	`+ for i, j in itertools.combinations_with_replacement(c, 2):`
	`6`	`+ k = target - i - j`
	`7`	`+ if i == j == k: res += c[i] * (c[i] - 1) * (c[i] - 2) // 6`
	`8`	`+ elif i == j != k: res += c[i] * (c[i] - 1) // 2 * c[k]`
	`9`	`+ elif k > i and k > j: res += c[i] * c[j] * c[k]`
	`10`	`+ return res % (10**9 + 7)`

Comments

(0)