bzoj 5088: HDU 6000 Wash 堆+贪心

题意

你现在要洗L件衣服。你有n台洗衣机和m台烘干机。由于你的机器非常的小，因此你每次只能洗涤(烘干)一件衣服。第i台洗衣机洗一件衣服需要wi分钟，第i台烘干机烘干一件衣服需要di分钟。请问把所有衣服洗干净并烘干，最少需要多少时间？假设衣服在机器间转移不需要时间，并且洗完的衣服可以过一会再烘干。
L<=10^6,N<=10^5,M<=10^5

分析

我的想法是，对于洗衣机，肯定是每次找结束时间最靠前的那个来洗。对于烘干机，我们仍然可以将其看作是洗衣机，只不过是倒过来的洗衣机。设第一批洗衣机的结束时间分别为a[1],a[2],…,a[L]，也就是第i件衣服在ai时刻取出，第二批洗衣机的结束时间为b[1],b[2],…,b[L]。
那么答案就是max(a[i]+b[L-i+1])。
实现的话拿个堆随便维护一下就好了。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;

typedef long long LL;

const int N=100005;

int k,n,m,t[N];
LL a[N*10],b[N*10];
priority_queue<pair<LL,int> > que;

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();}
    return x*f;
}

void solve(LL *a,int n)
{
    while (!que.empty()) que.pop();
    for (int i=1;i<=n;i++) que.push(make_pair(-t[i],i));
    for (int i=1;i<=k;i++)
    {
        pair<LL,int> u=que.top();que.pop();
        a[i]=-u.first;
        u.first-=t[u.second];que.push(u);
    }
}

int main()
{
    k=read();n=read();m=read();
    for (int i=1;i<=n;i++) t[i]=read();
    solve(a,n);
    for (int i=1;i<=m;i++) t[i]=read();
    solve(b,m);
    LL ans=0;
    for (int i=1;i<=k;i++) ans=max(ans,a[i]+b[k-i+1]);
    cout<<ans;
    return 0;
}