Commit bd214b5

author

Sandy

authored

Merge pull request #231 from openset/develop

Update: Maximize Distance to Closest Person

2 parents 2296948 + 345e6dd commit bd214b5Copy full SHA for bd214b5

File tree

+28

-20

lines changed

+28

-20

lines changed

Lines changed: 12 additions & 19 deletions

Original file line number	Diff line number	Diff line change
`@@ -1,28 +1,21 @@`
`1`	`1`	`package maximize_distance_to_closest_person`
`2`	`2`
`3`	`3`	`func maxDistToClosest(seats []int) int {`
`4`		`- ans, prev, future, l:= 0, -1, 0, len(seats)`
	`4`	`+ ans, ec:= 0, 0`
`5`	`5`	`for i, v := range seats {`
`6`		`- if v == 1 {`
`7`		`- prev = i`
`8`		`- } else {`
`9`		`- for future < l && seats[future] == 0 \|\| future < i {`
`10`		`- future++`
`11`		`- }`
`12`		`- left, right := l, l`
`13`		`- if prev != -1 {`
`14`		`- left = i - prev`
`15`		`- }`
`16`		`- if future != l {`
`17`		`- right = future - i`
`18`		`- }`
`19`		`- if right < left {`
`20`		`- left = right`
`21`		`- }`
`22`		`- if left > ans {`
`23`		`- ans = left`
	`6`	`+ if v == 0 {`
	`7`	`+ ec++`
	`8`	`+ } else if ec > 0 {`
	`9`	`+ if i == ec { // 001`
	`10`	`+ ans = ec`
	`11`	`+ } else if (ec+1)/2 > ans { // 101`
	`12`	`+ ans = (ec + 1) / 2`
`24`	`13`	`}`
	`14`	`+ ec = 0`
`25`	`15`	`}`
`26`	`16`	`}`
	`17`	`+ if ec > ans { // 100`
	`18`	`+ ans = ec`
	`19`	`+ }`
`27`	`20`	`return ans`
`28`	`21`	`}`

Lines changed: 16 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -17,11 +17,26 @@ func TestMaxDistToClosest(t *testing.T) {`
`17`	`17`	`input: []int{1, 0, 0, 0},`
`18`	`18`	`expected: 3,`
`19`	`19`	`},`
`20`		`-`
`21`	`20`	`{`
`22`	`21`	`input: []int{0, 0, 0, 1},`
`23`	`22`	`expected: 3,`
`24`	`23`	`},`
	`24`	`+ {`
	`25`	`+ input: []int{0, 0, 0, 0, 0, 1, 0},`
	`26`	`+ expected: 5,`
	`27`	`+ },`
	`28`	`+ {`
	`29`	`+ input: []int{0, 1, 0, 0, 0, 1, 0},`
	`30`	`+ expected: 2,`
	`31`	`+ },`
	`32`	`+ {`
	`33`	`+ input: []int{0, 1, 0, 0, 0, 0, 0},`
	`34`	`+ expected: 5,`
	`35`	`+ },`
	`36`	`+ {`
	`37`	`+ input: []int{1, 0, 1},`
	`38`	`+ expected: 1,`
	`39`	`+ },`
`25`	`40`	`}`
`26`	`41`	`for _, tc := range tests {`
`27`	`42`	`output := maxDistToClosest(tc.input)`

Comments

(0)