A-DDL的恐惧
题目描述
ZJM 有 n 个作业,每个作业都有自己的 DDL,如果 ZJM 没有在 DDL 前做完这个作业,那么老师会扣掉这个作业的全部平时分。
所以 ZJM 想知道如何安排做作业的顺序,才能尽可能少扣一点分。
请你帮帮他吧!
Input
输入包含T个测试用例。输入的第一行是单个整数T,为测试用例的数量。
每个测试用例以一个正整数N开头(1<=N<=1000),表示作业的数量。
然后两行。第一行包含N个整数,表示DDL,下一行包含N个整数,表示扣的分。
Output
对于每个测试用例,您应该输出最小的总降低分数,每个测试用例一行。
Sample Input
3
3
3 3 3
10 5 1
3
1 3 1
6 2 3
7
1 4 6 4 2 4 3
3 2 1 7 6 5 4
Sample Output
0
3
5
Hint
上方有三组样例。
对于第一组样例,有三个作业它们的DDL均为第三天,ZJM每天做一个正好在DDL前全部做完,所以没有扣分,输出0。
对于第二组样例,有三个作业,它们的DDL分别为第一天,第三天、第一天。ZJM在第一天做了第一个作业,第二天做了第二个作业,共扣了3分,输出3。
题解
- 用结构体job来代表一个作业,包含DDL和成绩grade。
- 将所有作业的信息输入。声明一个vector,存放DDL大于等于(在循环时)第i天的作业(即在第i天完成不会扣分的作业)。从第n天开始向前循环至第1天,用一个大根堆来储存如果在今天做完就不会扣分的作业。每向前循环一天,遍历job数组找到DDL比这一天小的放入堆 (堆是大根堆,分数多的为大),然后取出堆顶元素,即分数最多的作业,表示今天完成了这个作业。循环完成后,堆为空则所有作业都已做完;堆不为空则堆中没有被取出的元素就是没有完成的作业,把这些分数相加输出。
代码
#include <iostream>
#include <algorithm>
#include <string.h>
#include <vector>
using namespace std;
struct job{
int ddl;
int grade;
bool operator<(const job& j){
if(grade==j.grade){
return ddl<j.ddl;
}else{
return grade<j.grade;
}
}
};
int main(int argc, char** argv) {
int t;
cin>>t;
for(int ii=0;ii<t;ii++){
int n;
cin>>n;
job jo[n];//n份作业
for(int i=0;i<n;i++){
cin>>jo[i].ddl;
}
for(int i=0;i<n;i++){
cin>>jo[i].grade;
}
vector<job> v;
for(int i=n;i>0;i--){//i是天数,从后往前遍历,n~1
for(int j=0;j<n;j++){
//遍历数组找到比这一天小的ddl放入堆
if(jo[j].ddl==i){
v.push_back(jo[j]);
push_heap(v.begin(),v.end());
}
}
if(!v.empty()){
pop_heap(v.begin(),v.end());//重建堆
v.pop_back();//去掉最后一个
}
}
if(v.empty()){
printf("0\n");
} else{
int sz=v.size();
int ans=0;
for(int i=0;i<sz;i++){
ans=ans+v[i].grade;
}
printf("%d\n",ans);
}
}
return 0;
}