LeetCode in Kotlin

3025. Find the Number of Ways to Place People I

Medium

You are given a 2D array points of size n x 2 representing integer coordinates of some points on a 2D-plane, where points[i] = [xi, yi].

We define the right direction as positive x-axis (increasing x-coordinate) and the left direction as negative x-axis (decreasing x-coordinate). Similarly, we define the up direction as positive y-axis (increasing y-coordinate) and the down direction as negative y-axis (decreasing y-coordinate)

You have to place n people, including Alice and Bob, at these points such that there is exactly one person at every point. Alice wants to be alone with Bob, so Alice will build a rectangular fence with Alice’s position as the upper left corner and Bob’s position as the lower right corner of the fence (Note that the fence might not enclose any area, i.e. it can be a line). If any person other than Alice and Bob is either inside the fence or on the fence, Alice will be sad.

Return the number of pairs of points where you can place Alice and Bob, such that Alice does not become sad on building the fence.

Note that Alice can only build a fence with Alice’s position as the upper left corner, and Bob’s position as the lower right corner. For example, Alice cannot build either of the fences in the picture below with four corners (1, 1), (1, 3), (3, 1), and (3, 3), because:

Example 1:

Input: points = [[1,1],[2,2],[3,3]]

Output: 0

Explanation: There is no way to place Alice and Bob such that Alice can build a fence with Alice’s position as the upper left corner and Bob’s position as the lower right corner. Hence we return 0.

Example 2:

Input: points = [[6,2],[4,4],[2,6]]

Output: 2

Explanation: There are two ways to place Alice and Bob such that Alice will not be sad:

You cannot place Alice at (2, 6) and Bob at (6, 2) because the person at (4, 4) will be inside the fence.

Example 3:

Input: points = [[3,1],[1,3],[1,1]]

Output: 2

Explanation: There are two ways to place Alice and Bob such that Alice will not be sad:

You cannot place Alice at (1, 3) and Bob at (3, 1) because the person at (1, 1) will be on the fence.

Note that it does not matter if the fence encloses any area, the first and second fences in the image are valid.

Constraints:

Solution

class Solution {
    fun numberOfPairs(points: Array<IntArray>): Int {
        points.sortWith { a: IntArray, b: IntArray -> if (a[0] == b[0]) b[1] - a[1] else a[0] - b[0] }
        var cnt = 0
        for (i in points.indices) {
            var bot = Int.MIN_VALUE
            var top = points[i][1]
            for (j in i + 1 until points.size) {
                val y1 = points[j][1]
                if (y1 <= top && y1 > bot) {
                    cnt++
                    bot = y1
                    if (y1 == top) {
                        top--
                    }
                }
            }
        }
        return cnt
    }
}