2021秋季算法入门班第五章习题:优先队列、并查集

目录

小A与任务

题面

需要完成n个任务,第i个任务需要xix_i的时间,必须在yiy_i时间前完成,支付m枚金币可以在提前mzim* z_i的时间完成。

思路

首先一定是按照完成时间yiy_i进行排序,优先完成即将到时间的任务;题目要求使用金币最少,如果需要使用金币,一定是使用ziz_i最小的方法。

因此使用优先队列对ziz_i进行升序,假设某一时刻无法完成某个任务,考虑前面已经被确认可以完成的任务中,选择ziz_i最小的第i个任务(选择队头元素,有可能队头元素就是这一次任务本身,需要先push)。对于队头元素仍需考虑xix_i的值是否足够填满该段时间,如果需要的时间比队头大,则计算出费用,将队头元素之间弹出即可;若需要的时间比队头的小,则计算出费用,还需要把队头元素塞回队列。

代码

#include <bits/stdc++.h>
using namespace std;
const int N = 2e5+10;
int n;

struct node{
    int z , x , y;
    bool operator<(const node &s)const {
        return z < s.z;
    }
}a[N];

bool cmp(const node a,const node b){
    return a.y < b.y;
}

int main(){
    cin >> n;
    for(int i=1;i<=n;i++){
        cin >> a[i].z >> a[i].x >> a[i].y;
    }
    sort(a+1,a+n+1,cmp);
    double ans = 0;
    int cur = 0;
    priority_queue<node> pq;
    for(int i=1;i<=n;i++){
        cur += a[i].x;
        pq.push(a[i]);
        if(cur > a[i].y){
            int nd = cur - a[i].y;
            cur = a[i].y;
            while(!pq.empty()){
                if(nd == 0) break;
                auto tmp = pq.top(); pq.pop();
                if(nd > tmp.x){
                    ans += 1.0*tmp.x/tmp.z;
                    nd -= tmp.x;
                }else{
                    ans += 1.0*nd/tmp.z;
                    pq.push({tmp.z,tmp.x-nd,tmp.y});
                    nd = 0;
                }
            }
        }
    }
    printf("%.1lf\n",ans);
    return 0;
}

小C的周末

题面

n个人,每个人只想玩一款游戏。某一款游戏想要开始,必须要求所有玩这一款游戏的人连在同一条圈子里。小C有q条线,每条线都会把两个人连起来,注意一点就行,连的同时还会把与两个人相连的所有人连在一起。

思路

首先两个人合并在一起,肯定是并查集,那么就有一个问题,可能连线的两个人a和b不是玩的同一款游戏的,但是其他与a连接的人c可能与b玩的是同一款游戏。

并查集合并两个人的同时,需要将两个人的圈子里所有玩某款游戏的人合并在一起,但是1e51e51e5* 1e5的数组开不出来,可以选择使用map只存储玩某款游戏的人,最后两个人合并的时候小的合并到大的里面节省时间。

代码

#include <bits/stdc++.h>
using namespace std;
const int N = 1e5+10;
int n,m,q;
int wt[N],nd[N],ret[N];
int rt[N];
map<int,int> mp[N]; 

int find(int x){
    return rt[x]==x?x:rt[x]=find(rt[x]);
}

void un(int x,int y){
    int xx = find(x) , yy = find(y);
    if(xx != yy){
        rt[xx] = yy;
    }
}

int main(){
    while(scanf("%d %d %d",&n,&m,&q)!=EOF){
        for(int i=1;i<=m;i++){
            ret[i] = -1;
            nd[i] = 0;
        }
        for(int i=1;i<=n;i++){
            mp[i].clear();
            scanf("%d",&wt[i]);
            rt[i] = i;
            nd[wt[i]]++;
            mp[i][wt[i]]++;
        }
        for(int i=1;i<=q;i++){
            int x,y;
            scanf("%d %d",&x,&y);
            x = find(x) , y = find(y);
            if(mp[x].size() > mp[y].size()){
                swap(x,y);
            }
            rt[x] = y;
            for(auto it=mp[x].begin();it!=mp[x].end();it++){
                mp[y][it->first] += it->second;
                if(mp[y][it->first] == nd[it->first]){
                    ret[it->first] = i;
                }
            }
        }
        for(int i=1;i<=m;i++){
            if(nd[i] == 1){
                puts("0");
            }else{
                cout << ret[i] << endl;
            }
        }
        
    }
    return 0;
}
全部评论

相关推荐

04-21 16:05
已编辑
山西大学 Java
不吃压力👿:我和你简历差不多,好多看到28就不回复了,回复的基本是全栈或低代码
点赞 评论 收藏
分享
上周组里招人,我面了六个候选人,回来跟同事吃饭的时候聊起一个让我挺感慨的现象。前三个候选人,算法题写得都不错。第一道二分查找,五分钟之内给出解法,边界条件也处理得干净。第二道动态规划,状态转移方程写对了,空间复杂度也优化了一版。我翻他们的简历,力扣刷题量都在300以上。后三个呢,就有点参差不齐了。有的边界条件没处理好,有的直接说这道题没刷过能不能换个思路讲讲。其中有一个女生,我印象特别深——她拿到题之后没有马上写,而是先问我:“面试官,我能先跟你确认一下我对题目的理解吗?”然后她把自己的思路讲了一遍,虽然最后代码写得不是最优解,但整个沟通过程非常顺畅。这个女生的代码不是最优的,但当我问她“如果这里是线上环境,你会怎么设计’的时候,她给我讲了一套完整的方案——异常怎么处理、日志怎么打、怎么平滑发布。她对这是之前在实习的时候踩过的坑。”我在想LeetCode到底在筛选什么?我自己的经历可能有点代表性。我当年校招的时候,也是刷了三百多道题才敢去面试。那时候大家都刷,你不刷就过不了笔试关。后来工作了,前三年基本没再打开过力扣。真正干活的时候,没人让你写反转链表,也没人让你手撕红黑树。更多的是:这个接口为什么慢了、那个服务为什么OOM了、线上数据对不上了得排查一下。所以后来我当面试官,慢慢调整了自己的评判标准。算法题我还会出,但目的变了。我出算法题,不是想看你能不能背出最优解。而是想看你拿到一个陌生问题的时候,是怎么思考的。你会先理清题意吗?你会主动问边界条件吗?你想不出来的时候会怎么办?你写出来的代码,变量命名乱不乱、结构清不清楚?这些才是工作中真正用得到的能力。LeetCode是一个工具,不是目的。它帮你熟悉数据结构和常见算法思路,这没问题。但如果你刷了三百道题,却说不清楚自己的项目解决了什么问题、遇到了什么困难、你是怎么解决的,那这三百道题可能真的白刷了。所以还要不要刷LeetCode?要刷,但别只刷题。刷题的时候,多问自己几个为什么:为什么用这个数据结构?为什么这个解法比那个好?如果换个条件,解法还成立吗?把刷题当成锻炼思维的方式,而不是背答案的任务。毕竟面试官想看到的,从来不是一台背题机器,而是一个能解决问题的人。
牛客51274894...:意思是光刷力扣还不够卷
AI时代还有必要刷lee...
点赞 评论 收藏
分享
评论
1
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务