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;
}
相关推荐
KoiHeng39 分钟前
部分排序算法的Java模拟实现(复习向,非0基础)
java·算法·排序算法
艾莉丝努力练剑8 小时前
【数据结构与算法】数据结构初阶:详解顺序表和链表(四)——单链表(下)
c语言·开发语言·数据结构·学习·算法·链表
yngsqq9 小时前
移动碰撞法 ——套料排版算法——CAD c#
算法
秋说10 小时前
【PTA数据结构 | C语言版】根据层序序列重构二叉树
c语言·数据结构·算法
秋说11 小时前
【PTA数据结构 | C语言版】前序遍历二叉树
c语言·数据结构·算法
会唱歌的小黄李12 小时前
【算法】贪心算法:最大数C++
c++·算法·贪心算法
NuyoahC12 小时前
笔试——Day8
c++·算法·笔试
墨染点香12 小时前
LeetCode Hot100 【1.两数之和、2.两数相加、3.无重复字符的最长子串】
算法·leetcode·职场和发展
秋说13 小时前
【PTA数据结构 | C语言版】二叉树层序序列化
c语言·数据结构·算法
地平线开发者13 小时前
开发者说|Aux-Think:为什么测试时推理反而让机器人「误入歧途」?
算法·自动驾驶