P1392 取数

传送门:取数

如若你看完题解后,仍有问题,欢迎评论

首先说一下 我首先想到的思路 ( 20%通过率 ):通过dfs , 将所有的情况放入priority_queue中(greater<int>),维持堆中的数据为k个 -> TLE

代码:

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
int n, m, k;
const int N = 810;
int a[N][N];
priority_queue<int> heap;
void dfs(int i  , int sum)
{
    if (i == n + 1) {
        if (heap.size() == k && sum < heap.top()) {
            heap.pop(); heap.push(sum);
        }
        else if (heap.size() < k)heap.push(sum);
        return;
    }
    for (int k = 1; k <= m; k++)
    {
        dfs(i + 1, sum + a[i][k]);
    }
}
int main()
{
    scanf("%d%d%d", &n, &m, &k);
    for (int i = 1; i <= n; i++)
        for (int j = 1; j <= m; j++)
            scanf("%d", &a[i][j]);
    dfs(1, 0);
    
    vector<int> res;
    while (heap.size())
    {
        res.push_back(heap.top()); heap.pop();
    }
    reverse(res.begin(), res.end());
    for (auto e : res)
        printf("%d ", e);
    return 0;
}

错误代码的时间复杂度分析: n * m * klongk (建堆 + 遍历整个矩阵)> 10 ^ 8 因此TLE

优化方法:

时间复杂度分析:n * ( mlongm ( 排序 ) + klongk ( 调和级数 ) ) == nmlongm + nklongk = 10 ^ 6 (约等于 ) < 10 ^ 8 可以AC

AC Code:

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
const int N = 810;
int a[N] , b[N];
int n , m , k;
int x,y;
priority_queue<int> heap;
int main()
{
    scanf("%d%d%d",&n,&m,&k);
    heap.push(0);
    for( int i = 1 ; i <= n; i++)
    {
        x = 0; y = 0;
        while( heap.size()) b[++y] = heap.top(),heap.pop();
        for( int j= 1 ; j<= m;j++)scanf("%d",&a[j]);
        sort( a + 1 , a + m + 1 );
        for( int l = 1 ; l <= m;l++)
        {
            for( int r = y ; r ; r--)
            {
                if( x < k )heap.push( a[l] + b[r] ),x++;
                else if( heap.top() < a[l] + b[r] )break;
                else heap.pop(),heap.push(a[l] + b[r] );
            }
        }
    }
    vector<int> res;
    while( heap.size())
    {
        res.push_back(heap.top());
        heap.pop();
    }
    reverse( res.begin(),res.end());
    for( auto e : res  )
    {
        printf("%d ",e );
    }
    return 0;
}
相关推荐
时光追逐者1 分钟前
C# 哈希查找算法实操
算法·c#·哈希算法
Jasmine_llq26 分钟前
《P3825 [NOI2017] 游戏》
算法·游戏·枚举法·2-sat 算法·tarjan 算法·邻接表存储
Miraitowa_cheems29 分钟前
LeetCode算法日记 - Day 38: 二叉树的锯齿形层序遍历、二叉树最大宽度
java·linux·运维·算法·leetcode·链表·职场和发展
wangzy198230 分钟前
图形基础算法:如何将点与带曲线边的多边形位置关系算法做稳定
算法
艾醒1 小时前
探索大语言模型(LLM):Ollama快速安装部署及使用(含Linux环境下离线安装)
人工智能·深度学习·算法
艾醒1 小时前
探索大语言模型(LLM):Open-WebUI的安装
人工智能·算法·全栈
猫天意2 小时前
【CVPR2023】奔跑而非行走:追求更高FLOPS以实现更快神经网络
人工智能·深度学习·神经网络·算法·机器学习·卷积神经网络
宁檬精2 小时前
算法练习——55.跳跃游戏
数据结构·算法·游戏
王璐WL2 小时前
【C语言入门级教学】内存函数
c语言·开发语言·算法
啃啃大瓜2 小时前
python常量变量运算符
开发语言·python·算法