UVA - 1623 - Enter The Dragon

    

题意：给定n个湖（标号1~n，起始时湖里均满水），给定m天天气预报，每天天气预报情况为一个数字a[i]（范围0~n），当a[i]为0时代表晴天不下雨，否则就代表向哪个标号的湖中下暴雨。在不下雨的天气时你可以请神龙喝干任意一个湖里的水，下雨时则不能。若再往某个喝干的湖里下暴雨，则此湖会被重新填满。求是否能避免发生水灾，若能则给出在晴朗天气时神龙的喝水的顺序。

贪心：对于每个下雨的天气，在它之前的天内，找最后一次被填满的日期之后的晴朗天气，让神龙喝光这个湖内的水，若无法找到则无法避免水灾。

#include<cstdio>
#include<cstring>
#include<cctype>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<sstream>
#include<iterator>
#include<algorithm>
#include<string>
#include<vector>
#include<set>
#include<map>
#include<deque>
#include<queue>
#include<stack>
#include<list>
typedef long long ll;
typedef unsigned long long llu;
const int MAXN = 100 + 10;
const int MAXT = 1000000 + 10;
const int INF = 0x7f7f7f7f;
const double pi = acos(-1.0);
const double EPS = 1e-6;
using namespace std;

int n, m, a[MAXT], c[MAXT], t[MAXT], last[MAXT], cnum;
set<int> k;

//c存神龙喝水的顺序（实际先存晴朗天气的日期，再结合t数组进行输出）
//t实际仅对晴朗天气有用，负责存这一天喝哪个湖里的水
//last存对于某一个湖来说，这个湖最后被填满的日期
//k存当前可以用的晴朗天数的日期，结合lower_bound，找某个湖最后一次被填满日期之后的第一个可供给神龙喝此湖水的日期

bool judge(){
    memset(t, 0, sizeof t);
    memset(last, 0, sizeof last);
    k.clear();
    cnum = 0;
    for(int i = 0; i < m; ++i)
        if(a[i]){
            set<int>::iterator lur = k.lower_bound(last[a[i]]);
            if(lur == k.end())  return false;
            t[*lur] = a[i];
            last[a[i]] = i;
            k.erase(lur);
        }
        else{
            c[cnum++] = i;
            k.insert(i);
        }
    return true;
}

int main(){
    int T;
    scanf("%d", &T);
    while(T--){
        scanf("%d%d", &n, &m);
        for(int i = 0; i < m; ++i)  scanf("%d", &a[i]);
        if(judge()){
            printf("YES\n");
            for(int i = 0; i < cnum; ++i){
                if(i)  printf(" ");
                printf("%d", t[c[i]]);
            }
            printf("\n");
        }
        else  printf("NO\n");
    }
    return 0;
}