栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Java

2022-06-16:给定一个数组arr,含有n个数字,都是非负数, 给定一个正数k, 返回所有子序列中,累加和最小的前k个子序列累加和。 假设K不大,怎么算最快? 来自亚马逊。

Java 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

2022-06-16:给定一个数组arr,含有n个数字,都是非负数, 给定一个正数k, 返回所有子序列中,累加和最小的前k个子序列累加和。 假设K不大,怎么算最快? 来自亚马逊。

2022-06-16:给定一个数组arr,含有n个数字,都是非负数,
给定一个正数k,
返回所有子序列中,累加和最小的前k个子序列累加和。
假设K不大,怎么算最快?
来自亚马逊。

答案2022-06-16:

排序,小根堆。

代码用rust编写。代码如下:

fn main() {
    let mut nums: Vec = vec![6, 19, 3, 8, 29];
    let ans = top_min_sum2(&mut nums, 3);
    println!("ans = {:?}", ans);
}

fn top_min_sum2(arr: &mut Vec, k: i32) -> Vec {
    arr.sort();
    // (最右的下标,集合的累加和)
    let mut heap: Vec> = vec![];
    heap.push(vec![0, arr[0]]);
    let mut ans: Vec = vec![];
    for _ in 0..k {
        ans.push(0);
    }
    // ans[0] = 0
    // 0 1 2  k-1
    // k个!
    for i in 1..k {
        heap.sort_by(|a, b| b[1].cmp(&a[1]));
        let  cur = heap.pop().unwrap();
        // (7, 100)
        // 左 :8, 100 - arr[7] + arr[8]
        // 右 :8, 100 + arr[8]
        let  last = cur[0];
        let  sum = cur[1];
        ans[i as usize] = sum;
        if last + 1 < arr.len() as i32 {
            heap.push(vec![
                last + 1,
                sum - arr[last as usize] + arr[(last + 1) as usize],
            ]);
            heap.push(vec![last + 1, sum + arr[(last + 1) as usize]]);
        }
    }
    return ans;
}

执行结果如下:


左神java代码

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/978888.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号