We draw N discs on a plane. The discs are numbered from 0 to N − 1. An array A of N non-negative integers, specifying the radiuses of the discs, is given. The J-th disc is drawn with its center at (J, 0) and radius A[J].
We say that the J-th disc and K-th disc intersect if J ≠ K and the J-th and K-th discs have at least one common point (assuming that the discs contain their borders).
The figure below shows discs drawn for N = 6 and A as follows:
A[0] = 1 A[1] = 5 A[2] = 2 A[3] = 1 A[4] = 4 A[5] = 0There are eleven (unordered) pairs of discs that intersect, namely:
- discs 1 and 4 intersect, and both intersect with all the other discs;
- disc 2 also intersects with discs 0 and 3.
Write a function:
object Solution { def solution(a: Array[Int]): Int }
that, given an array A describing N discs as explained above, returns the number of (unordered) pairs of intersecting discs. The function should return −1 if the number of intersecting pairs exceeds 10,000,000.
Given array A shown above, the function should return 11, as explained above.
Write an efficient algorithm for the following assumptions:
- N is an integer within the range [0..100,000];
- each element of array A is an integer within the range [0..2,147,483,647].
import scala.collection.JavaConversions._
// you can use println for debugging purposes, e.g.
// println("this is a debug message")
object Solution {
def solution(A: Array[Int]): Int = {
case class DiskBorder(location: Long, isOpen: Boolean)
val borders = Array.fill[DiskBorder](A.length * 2)(null)
@scala.annotation.tailrec
def fill(idx: Int): Unit = {
if (idx < A.length) {
val radius = A(idx).toLong
val open = DiskBorder(idx.toLong - radius, true)
borders(2 * idx) = open
val close = DiskBorder(idx.toLong + radius, false)
borders(2 * idx + 1) = close
fill(idx + 1)
}
}
fill(0)
val f: DiskBorder => (Int, Int) = {
case DiskBorder(location, isOpen) =>
val openFirst = if (isOpen) 0 else 1
(location, openFirst)
}
scala.util.Sorting.stableSort(borders, f)
@scala.annotation.tailrec
def go(idx: Int, currentlyOpen: Int, accum: Int): Int = {
if (accum > 10000000) -1
else if (idx >= borders.length) accum
else {
if (borders(idx).isOpen) {
go(idx + 1, currentlyOpen + 1, accum + currentlyOpen)
} else {
go(idx + 1, currentlyOpen - 1, accum)
}
}
}
go(0, 0, 0)
}
}
Solution.scala:28: error: type mismatch; found : Long required: Int (location, openFirst) ^ one error found
import scala.collection.JavaConversions._
// you can use println for debugging purposes, e.g.
// println("this is a debug message")
object Solution {
def solution(A: Array[Int]): Int = {
case class DiskBorder(location: Long, isOpen: Boolean)
val borders = Array.fill[DiskBorder](A.length * 2)(null)
@scala.annotation.tailrec
def fill(idx: Int): Unit = {
if (idx < A.length) {
val radius = A(idx).toLong
val open = DiskBorder(idx.toLong - radius, true)
borders(2 * idx) = open
val close = DiskBorder(idx.toLong + radius, false)
borders(2 * idx + 1) = close
fill(idx + 1)
}
}
fill(0)
val f: DiskBorder => (Long, Int) = {
case DiskBorder(location, isOpen) =>
val openFirst = if (isOpen) 0 else 1
(location, openFirst)
}
scala.util.Sorting.stableSort(borders, f)
@scala.annotation.tailrec
def go(idx: Int, currentlyOpen: Int, accum: Int): Int = {
if (accum > 10000000) -1
else if (idx >= borders.length) accum
else {
if (borders(idx).isOpen) {
go(idx + 1, currentlyOpen + 1, accum + currentlyOpen)
} else {
go(idx + 1, currentlyOpen - 1, accum)
}
}
}
go(0, 0, 0)
}
}
import scala.collection.JavaConversions._
// you can use println for debugging purposes, e.g.
// println("this is a debug message")
object Solution {
def solution(A: Array[Int]): Int = {
case class DiskBorder(location: Long, isOpen: Boolean)
val borders = Array.fill[DiskBorder](A.length * 2)(null)
@scala.annotation.tailrec
def fill(idx: Int): Unit = {
if (idx < A.length) {
val radius = A(idx).toLong
val open = DiskBorder(idx.toLong - radius, true)
borders(2 * idx) = open
val close = DiskBorder(idx.toLong + radius, false)
borders(2 * idx + 1) = close
fill(idx + 1)
}
}
fill(0)
val f: DiskBorder => (Long, Int) = {
case DiskBorder(location, isOpen) =>
val openFirst = if (isOpen) 0 else 1
(location, openFirst)
}
scala.util.Sorting.stableSort(borders, f)
@scala.annotation.tailrec
def go(idx: Int, currentlyOpen: Int, accum: Int): Int = {
if (accum > 10000000) -1
else if (idx >= borders.length) accum
else {
if (borders(idx).isOpen) {
go(idx + 1, currentlyOpen + 1, accum + currentlyOpen)
} else {
go(idx + 1, currentlyOpen - 1, accum)
}
}
}
go(0, 0, 0)
}
}
import scala.collection.JavaConversions._
// you can use println for debugging purposes, e.g.
// println("this is a debug message")
object Solution {
def solution(A: Array[Int]): Int = {
case class DiskBorder(location: Long, isOpen: Boolean)
val borders = Array.fill[DiskBorder](A.length * 2)(null)
@scala.annotation.tailrec
def fill(idx: Int): Unit = {
if (idx < A.length) {
val radius = A(idx).toLong
val open = DiskBorder(idx.toLong - radius, true)
borders(2 * idx) = open
val close = DiskBorder(idx.toLong + radius, false)
borders(2 * idx + 1) = close
fill(idx + 1)
}
}
fill(0)
val f: DiskBorder => (Long, Int) = {
case DiskBorder(location, isOpen) =>
val openFirst = if (isOpen) 0 else 1
(location, openFirst)
}
scala.util.Sorting.stableSort(borders, f)
@scala.annotation.tailrec
def go(idx: Int, currentlyOpen: Int, accum: Int): Int = {
if (accum > 10000000) -1
else if (idx >= borders.length) accum
else {
if (borders(idx).isOpen) {
go(idx + 1, currentlyOpen + 1, accum + currentlyOpen)
} else {
go(idx + 1, currentlyOpen - 1, accum)
}
}
}
go(0, 0, 0)
}
}
The solution obtained perfect score.