UVA-1623 Enter The Dragon （贪心）

题目大意：有n个装满水的湖，m天。每天可能下雨也可能晴天，只要下雨就会把湖填满，若已满，则发洪水。有一台只能在晴天使用的抽水机，每次抽水只能抽一个湖，并且全部抽光。问是否存在一种使得不发洪水的抽水方案。

题目分析：贪心。贪心策略：对于每个下雨天 i ，让在这天之前的并且在第a[i]个湖上一次水满之后的一个晴天抽走第a[i]个湖中水。

代码如下：

# include<iostream>
# include<cstdio>
# include<set>
# include<cstring>
# include<algorithm>
using namespace std;
 
set<int>day;
int ans[1000005],lst[1000005],n,m;
 
void solve()
{
    memset(lst,0,sizeof(lst));
    memset(ans,0,sizeof(ans));
    int a,flag=1;
    day.clear();
    for(int i=0;i<m;++i){
        scanf("%d",&a);
        if(!flag)   continue;
        if(!a)
            day.insert(i);
        else{
            ans[i]=-1;
            set<int>::iterator it=day.lower_bound(lst[a]);
            if(it==day.end())
                flag=0;
            else{
                ans[*it]=a;
                lst[a]=i;
                day.erase(it);
            }
        }
    }
    if(!flag)
        printf("NO\n");
    else{
        printf("YES\n");
        for(int i=0;i<m;++i)
            if(ans[i]>=0)
                printf("%d ",ans[i]);
        printf("\n");
    }
}
 
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        solve();
    }
    return 0;
}