Top K of Two Sorted Array'sum

题目:有两个序列A和B,A=(a1,a2,...,ak),B=(b1,b2,...,bk),A和B都按升序排列,对于1<=i,j<=k,求k个最小的(ai+bj),要求算法尽量高效 .
思路:
设定两个下标i,j分别指向A,B的某个地方,若当前(i-1)*j>=k或(j-1)*i>=k说明,剩下的组合是最小的i*j,而且可以根据A[i],B[j]两个元素的大小分别移动相应的下标,直到(i-1)*j<k或(j-1)*i<k,此时剩下的组合数为i*j,遍历数组求得前k个最小和,返回给用户。

results matching ""

    No results matching ""