牛客寒假算法基础集训营2：处女座的期末复习（贪心）

链接：https://ac.nowcoder.com/acm/contest/327/J
来源：牛客网
 

题目描述

快要期末考试了，处女座现在有n门课程需要考试，每一门课程需要花ai小时进行复习，考试的起始时间为bi，处女座为了考试可以不吃饭不睡觉，处女座想知道他能否复习完所有的科目（即在每一门考试之前复习完该科目）。每一门课的考试时间都为两小时。

输入描述:

第一行一个整数n

第二行n个整数a1,a2,…,an,表示每门课需要复习的时间

第三行n个整数b1,b2,…,bn,表示每门课考试的时间

1<=n<=105

0<=ai<=109

0<=bi<=109

输出描述:

如果处女座能复习完，输出”YES”，否则输出”NO”

示例1

输入

复制

3
0 1 1
2 6 4

输出

复制

扫描二维码关注公众号，回复： 5676172 查看本文章

YES

说明

在0-1小时复习第2门课，
在1-2小时复习第3门课，
在2-4小时考第1门课，
在4-6小时考第3门课，
在6-8小时考第2门课

备注:

考试时不能复习，保证考试时间不会重叠。

复习可以拆开，只要复习时间够了即可。

思路：创建一个结构体记录一次考试需要的复习时间和开始时间，按考试开始时间排序，依次对复习时间求和，对比在这个时间点之前，复习所需时间是否大于这场考试开始时间。

#include <iostream>

#include <cstring>

#include <algorithm>

#include <cmath>

#define inf 0x3f3f3f3f

using namespace std;

const int MAX=1e5+5;

struct Node
{
    int str;
    int end;
}a[MAX];

bool flag;

bool cmp(Node a, Node b)
{
    return a.end < b.end;
}
int main()
{
    int n;
    cin >> n;
    for(int i = 1; i <= n; i++){
        cin >> a[i].str;
    }
    for(int i = 1; i <= n; i++){
        cin >> a[i].end;
    }
    sort(a + 1, a + n + 1, cmp);
    flag = true;
    int sum = 0;
    for(int i = 1; i <= n; i++){
        sum += a[i].str;
//        cout << sum <<endl;
        if(sum > a[i].end){
            flag = false;
            break;
        }
        sum += 2;
//        cout << "sum=" << sum << "  end=" << a[i].end << endl;
    }
    if(flag){
        cout << "YES" << endl;
    }
    else{
        cout << "NO" << endl;
    }
    return 0;
}