Question

786. 第 K 个最小的质数分数

English Version

题目描述

给你一个按递增顺序排序的数组 arr 和一个整数 k 。数组 arr 由 1 和若干 质数 组成，且其中所有整数互不相同。

对于每对满足 0 <= i < j < arr.length 的 i 和 j ，可以得到分数 arr[i] / arr[j] 。

那么第 k 个最小的分数是多少呢?  以长度为 2 的整数数组返回你的答案, 这里 answer[0] == arr[i] 且 answer[1] == arr[j] 。

 

示例 1：

输入：arr = [1,2,3,5], k = 3
输出：[2,5]
解释：已构造好的分数,排序后如下所示: 
1/5, 1/3, 2/5, 1/2, 3/5, 2/3
很明显第三个最小的分数是 2/5

示例 2：

输入：arr = [1,7], k = 1
输出：[1,7]

 

提示：

• 2 <= arr.length <= 1000
• 1 <= arr[i] <= 3 * 104
• arr[0] == 1
• arr[i] 是一个 质数 ，i > 0
• arr 中的所有数字 互不相同 ，且按 严格递增 排序
• 1 <= k <= arr.length * (arr.length - 1) / 2

 

进阶：你可以设计并实现时间复杂度小于 O(n2) 的算法解决此问题吗？

解法

方法一

class Solution:
  def kthSmallestPrimeFraction(self, arr: List[int], k: int) -> List[int]:
    h = [(1 / y, 0, j + 1) for j, y in enumerate(arr[1:])]
    heapify(h)
    for _ in range(k - 1):
      _, i, j = heappop(h)
      if i + 1 < j:
        heappush(h, (arr[i + 1] / arr[j], i + 1, j))
    return [arr[h[0][1]], arr[h[0][2]]]

class Solution {
  public int[] kthSmallestPrimeFraction(int[] arr, int k) {
    int n = arr.length;
    Queue<Frac> pq = new PriorityQueue<>();
    for (int i = 1; i < n; i++) {
      pq.offer(new Frac(1, arr[i], 0, i));
    }
    for (int i = 1; i < k; i++) {
      Frac f = pq.poll();
      if (f.i + 1 < f.j) {
        pq.offer(new Frac(arr[f.i + 1], arr[f.j], f.i + 1, f.j));
      }
    }
    Frac f = pq.peek();
    return new int[] {f.x, f.y};
  }

  static class Frac implements Comparable {
    int x, y, i, j;

    public Frac(int x, int y, int i, int j) {
      this.x = x;
      this.y = y;
      this.i = i;
      this.j = j;
    }

    @Override
    public int compareTo(Object o) {
      return x * ((Frac) o).y - ((Frac) o).x * y;
    }
  }
}

class Solution {
public:
  vector<int> kthSmallestPrimeFraction(vector<int>& arr, int k) {
    using pii = pair<int, int>;
    int n = arr.size();
    auto cmp = [&](const pii& a, const pii& b) {
      return arr[a.first] * arr[b.second] > arr[b.first] * arr[a.second];
    };
    priority_queue<pii, vector<pii>, decltype(cmp)> pq(cmp);
    for (int i = 1; i < n; ++i) {
      pq.push({0, i});
    }
    for (int i = 1; i < k; ++i) {
      pii f = pq.top();
      pq.pop();
      if (f.first + 1 < f.second) {
        pq.push({f.first + 1, f.second});
      }
    }
    return {arr[pq.top().first], arr[pq.top().second]};
  }
};

type frac struct{ x, y, i, j int }
type hp []frac

func (a hp) Len() int       { return len(a) }
func (a hp) Swap(i, j int)    { a[i], a[j] = a[j], a[i] }
func (a hp) Less(i, j int) bool { return a[i].x*a[j].y < a[j].x*a[i].y }
func (a *hp) Push(x any)    { *a = append(*a, x.(frac)) }
func (a *hp) Pop() any      { l := len(*a); tmp := (*a)[l-1]; *a = (*a)[:l-1]; return tmp }

func kthSmallestPrimeFraction(arr []int, k int) []int {
  n := len(arr)
  h := make(hp, 0, n-1)
  for i := 1; i < n; i++ {
    h = append(h, frac{1, arr[i], 0, i})
  }
  heap.Init(&h)
  for i := 1; i < k; i++ {
    f := heap.Pop(&h).(frac)
    if f.i+1 < f.j {
      heap.Push(&h, frac{arr[f.i+1], arr[f.j], f.i + 1, f.j})
    }
  }
  return []int{h[0].x, h[0].y}
}