#2748

Number of Beautiful Pairs

Easy
ArrayHash TableMathCountingNumber TheoryHash MapArray
LeetCode ↗

Approaches

Brute ForceOptimal
Complexity Comparison
Brute ForceOptimal Solution
Time
O(n²)
O(n + k²)
Space
O(1)
O(k)
💡

Intuition

Time O(n + k²)Space O(k)

We can optimize the solution by precomputing the first and last digits for all numbers and then using a frequency array to count occurrences. This allows us to quickly check coprimality without nested loops.

⚙️

Algorithm

5 steps
  1. 1Step 1: Create an array to count occurrences of first digits (1-9) and last digits (0-9).
  2. 2Step 2: Iterate through the nums array and fill the frequency arrays for first and last digits.
  3. 3Step 3: For each first digit, check against all last digits using the gcd function.
  4. 4Step 4: Multiply the counts of coprime pairs and accumulate the result.
  5. 5Step 5: Return the total count of beautiful pairs.
solution.py16 lines
1from math import gcd
2
3def countBeautifulPairs(nums):
4    first_count = [0] * 10
5    last_count = [0] * 10
6    for num in nums:
7        first_digit = int(str(num)[0])
8        last_digit = num % 10
9        first_count[first_digit] += 1
10        last_count[last_digit] += 1
11    count = 0
12    for i in range(1, 10):
13        for j in range(10):
14            if gcd(i, j) == 1:
15                count += first_count[i] * last_count[j]
16    return count

Complexity note: The time complexity is O(n + k²) where n is the number of elements in nums and k is the number of unique digits (which is constant, 10). The space complexity is O(k) due to the frequency arrays.

  • 1Understanding coprimality and how to efficiently check it using gcd.
  • 2Recognizing patterns in the problem to optimize the solution.

Solutions and explanations are original Tejav content. Problem titles © LeetCode — use the LeetCode button above for the full problem statement.