字节跳动2018校招大数据方向（第一批）

ACM题集：https://blog.csdn.net/weixin_39778570/article/details/83187443

第三题

链接：https://www.nowcoder.com/questionTerminal/f76b7bc64e554edaa53d8e0d84f921c5
来源：牛客网

[编程题]编程题3
热度指数：1104时间限制：C/C++ 1秒，其他语言2秒空间限制：C/C++ 80M，其他语言160M
算法知识视频讲解
产品经理(PM)有很多好的idea，而这些idea需要程序员实现。现在有N个PM，在某个时间会想出一个 idea，每个 idea 有提出时间、所需时间和优先等级。对于一个PM来说，最想实现的idea首先考虑优先等级高的，相同的情况下优先所需时间最小的，还相同的情况下选择最早想出的，没有 PM 会在同一时刻提出两个 idea。

同时有M个程序员，每个程序员空闲的时候就会查看每个PM尚未执行并且最想完成的一个idea,然后从中挑选出所需时间最小的一个idea独立实现，如果所需时间相同则选择PM序号最小的。直到完成了idea才会重复上述操作。如果有多个同时处于空闲状态的程序员，那么他们会依次进行查看idea的操作。

求每个idea实现的时间。

输入第一行三个数N、M、P，分别表示有N个PM，M个程序员，P个idea。随后有P行，每行有4个数字，分别是PM序号、提出时间、优先等级和所需时间。输出P行，分别表示每个idea实现的时间点。


输入描述:
输入第一行三个数N、M、P，分别表示有N个PM，M个程序员，P个idea。随后有P行，每行有4个数字，分别是PM序号、提出时间、优先等级和所需时间。全部数据范围 [1, 3000]。


输出描述:
输出P行，分别表示每个idea实现的时间点。
示例1
输入
2 2 5
1 1 1 2
1 2 1 1
1 3 2 2
2 1 1 2
2 3 5 5
输出
3
4
5
3
9

读题很重要
注意每个时刻能做的任务必须是pm在这个时刻或者之前就想出来了的
维护每个pm想出来的任务列队，然后程序员再进行挑选
如果某个时刻没有任务，那么一直等到下一个有任务被想出来的时刻

#include
#define ll long long
#define fo(i,j,n) for(register int i=j; i<=n; ++i)
using namespace std;
const int N = 5e5+5;
struct Idea{
	int id,pid,time,pri,need;
	bool operator < (const Idea& p)const{
		return time<p.time; 
	}
}idea[N];
struct Read{
	int id,pid,time,pri,need;
	Read(int a, int b, int c, int d, int e):id(a),pid(b),time(c),pri(d),need(e){}
	bool operator < (const Read& p)const{
		// 大在前使用<比较,小在前使用>比较   ===  在重载<号的时候 -- 大根堆 
		if(pri==p.pri){
			if(need==p.need){
				return time> p.time;
			}else return need>p.need;
		}else return pri<p.pri;
	}
}; 
struct Task{
	int id,need, pid;
	Task(int a, int b, int c):id(a),need(b),pid(c){}
	bool operator < (const Task& p)const{
		if(need==p.need){
			return pid>p.pid;
		}else return need>p.need;
	} 
};
int n,m,p, ans[N],peop[N];
// 每个pm的想好了的任务列队 
priority_queue<Read> pm_read[N];
void solve(){
	sort(idea+1,idea+1+p);
	int idx = 1;
	int cnt = 0;
	while(cnt<p){		
		// 程序员最早空闲时间 
		int nowT = peop[1]; 
		for(int i=idx; i<=p; i++){
			if(idea[i].time <= nowT){
				idx = i+1;
				pm_read[idea[i].pid].push(Read(idea[i].id,idea[i].pid,idea[i].time,idea[i].pri,idea[i].need));
			}else break;
		}
		// 就绪任务列队 
		priority_queue<Task> task;
		fo(i,1,n){
			if(pm_read[i].size()){
				Read r = pm_read[i].top();
				task.push(Task(r.id,r.need,r.pid)); 
			}
		}
		if(task.size()){
			Task t = task.top();
			pm_read[t.pid].pop();
			peop[1] += t.need; 
			ans[t.id] = peop[1];
			cnt++;  
			sort(peop+1, peop+1+m);
		}else{
			if(idx<=p){
				fo(i,1,m){
					// 程序员们等待 
					if(peop[i]<idea[idx].time){
						peop[i] = idea[idx].time;
					}else{
						break;
					}
				}	
			}
		} 
	}
	fo(i,1,p){
		printf("%d\n",ans[i]);
	}
} 
int main(){
	scanf("%d%d%d",&n,&m,&p);
	fo(i,1,p){
		idea[i].id = i;
		scanf("%d%d%d%d",&idea[i].pid,&idea[i].time,&idea[i].pri,&idea[i].need);
	}
	fo(i,1,m)peop[i]=0;
	solve();
	return 0;
}