每日一题---786. 第 K 个最小的素数分数[力扣][Go]

题目描述

给你一个按递增顺序排序的数组 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.先将所有结果都整出来
2.再通过排序就能找到第K个

但是自己写的代码太乱了，思路和力扣官方给的一样，这里就将力扣官方的解题代码拿来了。

// 官方暴力解法
func kthSmallestPrimeFraction(arr []int, k int) []int {
	n := len(arr)
	// 定义一个结构体来存储除数与被除数
	type pair struct {
		x,y int
	}
	// 定义一个起始长度为0，长度为n*(n-1)/2的数组，因为n个数能形成n*(n-1)/2个结果
	sum := make([]pair,0,n*(n-1)/2)
	// 暴力
	for i, x := range arr {
		for _, y := range arr[i+1:] {
			sum = append(sum, pair{x,y})
		}
	}
	// 自定义排序
	sort.Slice(sum, func(i, j int) bool {
		a,b := sum[i],sum[j]
		return a.x*b.y < a.y*b.x
	})
	return []int{sum[k-1].x,sum[k-1].y}
}

提交结果