算法学习 -- 多路归并

思想 :

抽象出来一个例子 : 合并k个长度相等升序列表 :

抽象成一张表也就是 :

做法 : 用一个小根堆来维护 , 首先将每个序列的第一个元素放入队列中 , 然后模拟,每次取出队头,作为结果序列的下一个元素 , 然后向堆中存入该元素序列中的下一个元素 ,直到所有元素取完为止;

例题 :

1262 - 鱼塘钓鱼

1262. 鱼塘钓鱼 - AcWing题库

c++代码 :

复制代码
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 110 ;
typedef pair<int, int> PII;
int a[N] , b[N] , c[N] ;
int n , T ;

int main(){
    cin >> n ; 
    for(int i=1;i<=n;i++) cin >> a[i] ;
    for(int j=1;j<=n;j++) cin >> b[j] ;
    for(int k=2;k<=n;k++) cin >> c[k] , c[k] += c[k-1] ;
    cin >> T ;
    int ans = 0 ;
    // 只在前k个池塘钓鱼
    for(int i=1;i<=n;i++){
        int t = T - c[i] ; // 本次钓鱼专用时间 
        priority_queue<PII> que ;
        for(int j=1;j<=i;j++) que.push({a[j],j}) ;
        int res = 0 ;
        while(!que.empty() && t > 0){
            auto it = que.top() ; que.pop() ;
            int x = it.first , idx = it.second ; // 获取元素和下标
            res += x ;
            int xx = x - b[idx] ;
            if(xx > 0) que.push({xx , idx}) ;
            t -- ;
        }
        ans = max(ans , res) ;
    }
    cout << ans << endl ;
}

62 . 丑数

62. 丑数 - AcWing题库

复制代码
class Solution {
public:
    int getUglyNumber(int n) {
        if(n <= 1) return n;
        vector<int> f(1,1);
        int i = 0, j = 0, k = 0;
        long long t = 0;
        while(--n)
        {
            t = min(f[i] * 2, min (f[j] * 3, f[k] * 5));
            if(t == f[i] * 2) i++;
            if(t == f[j] * 3) j++;
            if(t == f[k] * 5) k++;
            f.push_back(t);
        }
        return f.back();
    }
};

参考 : 多路归并算法从理论到应用(易懂)-CSDN博客

相关推荐
静心问道3 小时前
XLSR-Wav2Vec2:用于语音识别的无监督跨语言表示学习
人工智能·学习·语音识别
懒惰的bit9 天前
STM32F103C8T6 学习笔记摘要(四)
笔记·stm32·学习
Jay_5159 天前
C++ STL 模板详解:由浅入深掌握标准模板库
c++·学习·stl
冰茶_9 天前
ASP.NET Core API文档与测试实战指南
后端·学习·http·ui·c#·asp.net
丶Darling.9 天前
深度学习与神经网络 | 邱锡鹏 | 第五章学习笔记 卷积神经网络
深度学习·神经网络·学习
cwtlw9 天前
Excel学习03
笔记·学习·其他·excel
牛大了20239 天前
【LLM学习】2-简短学习BERT、GPT主流大模型
gpt·学习·bert
Ting-yu9 天前
零基础学习RabbitMQ(1)--概述
分布式·学习·rabbitmq
丶Darling.9 天前
深度学习与神经网络 | 邱锡鹏 | 第七章学习笔记 网络优化与正则化
深度学习·神经网络·学习
blackA_9 天前
JavaWeb学习——day8(MVC模式与session、cookie)
学习·mvc