Commit 7c65e0a

committed

[ACCEPTED][Problem 2799]Count Complete Subarrays in an Array

1 parent 55852ee commit 7c65e0aCopy full SHA for 7c65e0a

File tree

+20

-13

lines changed

+20

-13

lines changed

Lines changed: 20 additions & 13 deletions

Original file line number	Diff line number	Diff line change
`@@ -3,24 +3,31 @@ package medium`
`3`	`3`	`object CountCompleteSubArraysInAnArray2799 {`
`4`	`4`	`fun countCompleteSubarrays(nums: IntArray): Int {`
`5`	`5`	`val n = nums.size`
`6`		`- val k = nums.distinct().size`
	`6`	`+ val distinct = nums.distinct().size`
`7`	`7`	`var result = 0`
`8`		`- var start = 0`
`9`		`- var end = 0`
	`8`	`+ var left = 0`
	`9`	`+ var right = 0`
	`10`	`+ val distinctMap = HashMap<Int, Int>()`
`10`	`11`
`11`		`- while (start < n) {`
`12`		`- while (end < n) {`
`13`		`- val subArray = nums.copyOfRange(start, end +1)`
`14`		`- val subArrDistinct = subArray.distinct().size`
`15`		`- if (subArrDistinct == k)`
`16`		`- result++`
`17`		`- end++`
	`12`	`+ while (left < n) {`
	`13`	`+ if (left >0) {`
	`14`	`+ val key = nums[left -1]`
	`15`	`+ distinctMap[key] = distinctMap.getOrDefault(key, 0) -1`
	`16`	`+ if (distinctMap.getOrDefault(key, 0) == 0) {`
	`17`	`+ distinctMap.remove(key)`
	`18`	`+ }`
`18`	`19`	`}`
`19`		`- start++`
`20`		`- end = start`
	`20`	`+ while (right < n && distinctMap.size < distinct) {`
	`21`	`+ val currentKey = nums[right]`
	`22`	`+ distinctMap[currentKey] = distinctMap.getOrDefault(currentKey, 0) + 1`
	`23`	`+ right++`
	`24`	`+ }`
	`25`	`+ if (distinctMap.size == distinct) {`
	`26`	`+ result += n - right + 1`
	`27`	`+ }`
	`28`	`+ left++`
`21`	`29`	`}`
`22`	`30`
`23`	`31`	`return result`
`24`		`-`
`25`	`32`	`}`
`26`	`33`	`}`

Comments

(0)