UVA 1623 Enter The Dragon

题意：

　　一只龙，在每个不下雨的日子都可以喝干一个湖里的水，当湖满时，再向这个湖里下雨就会溢出。给出下雨的顺序，求龙喝水的序列。

分析：

　　记录每个湖上次满水的日子，和不下雨的日子。下雨时，查找当前湖上次灌满的日子之后有没有不下雨的日子，让龙在离上次灌满最近的一天喝光那个湖的水。

代码：

　　

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#include <set>
using namespace std;
int n,m;
const int maxn=1000010;
set<int>day;
int ans[maxn];
int full[maxn];
int pos;
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        pos=0;
        scanf("%d%d",&n,&m);
        int i,j;
        int flag=1;
        memset(ans,0,sizeof(ans));
        memset(full,0,sizeof(full));
        day.clear();
        for(i=0;i<m;i++)
        {
            int k;
            scanf("%d",&k);
            if(flag==0)
                continue;
            if(k==0)
                day.insert(i);
            else
            {
                ans[i]=-1;
                set<int>::iterator it=day.lower_bound(full[k]);
                if(it==day.end())
                    flag=0;
                else
                {
                    ans[*it]=k;
                    full[k]=i;
                    day.erase(*it);
                }
            }
        }
        if(flag==0)
            printf("NO
");
        else
        {
            int flag1=1;
             printf("YES
");
             for(i=0;i<m;i++)
             {
                 if(ans[i]>=0)
                 {
                     if(flag1==1)
                        printf("%d",ans[i]);
                     else
                        printf(" %d",ans[i]);
                     flag1++;
                 }
             }
             printf("
");
        }
    }
}