Commit e664089

committed

1 parent 71ddba0 commit e664089Copy full SHA for e664089

File tree

-10

lines changed

-10

lines changed

Lines changed: 6 additions & 10 deletions

Original file line number	Diff line number	Diff line change
`@@ -7,20 +7,16 @@ fun main() {`
`7`	`7`	`fun sortMatrix(grid: Array<IntArray>): Array<IntArray> {`
`8`	`8`	`return grid.indices.flatMap { r ->`
`9`	`9`	`grid[r].indices.map { r to it }`
`10`		`- }.groupBy { it.first - it.second }.flatMap { (diff, positions) ->`
`11`		`- val d = diff *2+1`
`12`		`-`
`13`		`- val sorted = positions.sortedWith { (r1, c1), (r2, c2) ->`
`14`		`- (grid[r2][c2] - grid[r1][c1]) * d`
	`10`	`+ }.groupBy { (it.first - it.second) *2+1 }.flatMap { (diff, positions) ->`
	`11`	`+ positions.sortedWith { (r1, c1), (r2, c2) ->`
	`12`	`+ (grid[r2][c2] - grid[r1][c1]) * diff`
	`13`	`+ }.let { sorted ->`
	`14`	`+ positions.indices.map { positions[it] to grid[sorted[it].first][sorted[it].second] }`
`15`	`15`	`}`
`16`		`-`
`17`		`- positions.indices.map { positions[it] to sorted[it] }`
`18`	`16`	`}.toMap().let {`
`19`	`17`	`Array(grid.size) { r ->`
`20`	`18`	`IntArray(grid[r].size) { c ->`
`21`		`- it[r to c]?.let { (tr, tc) ->`
`22`		`- grid[tr][tc]`
`23`		`- } ?: 0`
	`19`	`+ it[r to c] ?: 0`
`24`	`20`	`}`
`25`	`21`	`}`
`26`	`22`	`}`

Comments

(0)