P8755 [蓝桥杯 2021 省 AB2] 负载均衡（堆，贪心）

今天开始准备更新一下蓝桥杯的相关题解，为了准备明年的蓝桥杯

[蓝桥杯 2021 省 AB2] 负载均衡

题目描述

有 $n$ 台计算机，第 $i$ 台计算机的运算能力为 $v_i$ 。

有一系列的任务被指派到各个计算机上，第 $i$ 个任务在 $a_i$ 时刻分配，指定计算机编号为 $b_i$, 耗时为 $c_i$ 且算力消耗为 $d_i$。如果此任务成功分配，将立刻开始运行, 期间持续占用 $b_i$ 号计算机 $d_i$ 的算力, 持续 $c_i$ 秒。

对于每次任务分配，如果计算机剩余的运算能力不足则输出 $-1$，并取消这次分配，否则输出分配完这个任务后这台计算机的剩余运算能力。

输入格式

输入的第一行包含两个整数 $n,m$，分别表示计算机数目和要分配的任务数。

第二行包含 $n$ 个整数 $v_1,v_2,\cdots v_n$，分别表示每个计算机的运算能力。

接下来 $m$ 行每行 $4$ 个整数 $a_i,b_i,c_i,d_i$，意义如上所述。数据保证 $a_i$ 严格递增，即 $a_i<a_{i+1}$ 。

输出格式

输出 $m$ 行，每行包含一个数，对应每次任务分配的结果。

样例 #1

样例输入 #1

2 6
5 5
1 1 5 3
2 2 2 6
3 1 2 3
4 1 6 1
5 1 3 3
6 1 3 4

样例输出 #1

2
-1
-1
1
-1
0

提示

【样例说明】

时刻 $1$，第 $1$ 个任务被分配到第 $1$ 台计算机，耗时为 $5$，这个任务时刻 $6$ 会结束, 占用计算机 $1$ 的算力 $3$。

时刻 $2$，第 $2$ 个任务需要的算力不足，所以分配失败了。

时刻 $3$，第 $1$ 个计算机仍然正在计算第 $1$ 个任务，剩余算力不足 $3$，所以失败。

时刻 $4$，第 $1$ 个计算机仍然正在计算第 $1$ 个任务，但剩余算力足够，分配后剩余算力 $1$。

时刻 $5$，第 $1$ 个计算机仍然正在计算第 $1$，$4$ 个任务，剩余算力不足 $4$，失败。

时刻 $6$，第 $1$ 个计算机仍然正在计算第 $4$ 个任务，剩余算力足够，且恰好用完。

【评测用例规模与约定】

对于 $20\%$ 的评测用例, $n,m\le 200$。

对于 $40\%$ 的评测用例, $n,m\le 2000$。

对于所有评测用例, $1\le n,m\le 2\times 10^5,1\le a_i,c_i,d_i,v_i\le 10^9,1\le b_i\le n$。

蓝桥杯 2021 第二轮省赛 A 组 H 题（B 组 I 题）。

分析

其实这题在看y总讲之前不会，开始准备硬模拟的，但是发现有问题，于是看了y总的堆的做法，思路是对每个编号的计算机建一个$pair$的小根堆，堆的$first$元素是存放任务的$ddl$，$second$元素是存放这个任务需要的算力，如果当我们发现堆顶元素的$first$值$\le$当前的时间点即$a_i$，就说明当前的任务已经完成，就把算力加回来，并且把堆顶弹出，然后如果当前剩余的算力比完成这个任务需要的算力高，就把当前剩余算力减去完成这个任务的算力，并且把这个任务的$ddl$和算力值放入堆中。

代码

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <string> 
#include <queue>
#include <vector>
#include <map>
#include <set>
#include <stack> 
#include <cmath>
#define ll long long
#define AC return
#define Please 0
using namespace std;
const int N=200020;
typedef pair<int,int>PII;
inline int read(){//快读 
	int x=0,f=1;char ch=getchar();
	while(ch<'0' || ch>'9'){
		if(ch=='-') f=-1;
		ch=getchar();
	}
	while(ch>='0' && ch<='9'){
		x=x*10+ch-'0'; 
		ch=getchar();
	}
	AC x*f;
}
int v[N],n,m;
priority_queue<PII,vector<PII>,greater<PII>> heap[N];
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++){
		scanf("%d",&v[i]);
	}
	for(int i=1;i<=m;i++){
		int a,b,c,d;
		scanf("%d%d%d%d",&a,&b,&c,&d);
		while(heap[b].size() && heap[b].top().first<=a){
			v[b]+=heap[b].top().second;
			heap[b].pop();
		}
		if(v[b]<d){
			printf("-1\n");
		}
		else{
			v[b]-=d;
			heap[b].push({a+c,d});
			printf("%d\n",v[b]);
		}
	}
	return (0-0) //<3;
}

这题算是对堆的一个应用。