1623-Enter the Dragon【贪心】

题目大意

传送门

某市有n个湖，每个湖的水都是满的。
天气预报会显示未来的m天内，往哪个湖中下雨。
如果湖水中本来就有水，现在又往里面下雨，就会引起水灾。
为了避免水灾，市长请来了一条神龙，神龙可以在不下雨的天里喝干某一个湖中的水。
神龙应该怎么喝水才能避免水灾。
n<=10^6
m<=10^6

样例

input

4
2 4
0 0 1 1
2 4
0 1 0 2
2 3
0 1 2
2 4
0 0 0 1

output

NO
YES
1 2
NO
YES
0 1 0

解释

对于样例
2 4
0 1 0 2
龙王可以在第一天喝干第1个湖，那么第二天往第1个湖中下雨的时候，就不会引起水灾。
同理。

思路

每一个非下雨天都能中和一个下雨天，我本来想的是使用数组来存储下雨天的相关信息，后来就不知道怎么处理了。

看了解答是使用set来存储下雨天，这样就可以使用set自带的lower_bound来查找池塘i距离上一个满水的日期full[i]的最近的一天来吸水。

然后动态来维护这个set。
当set中查找不到池塘i距离上一个满水日期full[i]的后续最近的一天的时候，就是false。

代码

#include<cstdio>
#include<map>
#include<queue>
#include<cstring>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>
#include<stdlib.h>
#include <math.h>
#include <stack>
#include <set>
using namespace std;
#define  LL long long

#define MAXN 1000000 + 5
int CaseNum;
int n,m;

int num;

set<int> d;//存储第几天不降雨
int ans[MAXN];//记录非下雨天的操作
int full[MAXN];//记录第i个湖的上一次满水的时间

int main()
{
    int CaseNum;
    scanf("%d",&CaseNum);
    while(CaseNum--)
    {
        bool flag = true;
        memset(full,0,sizeof(full));
        memset(ans,0,sizeof(ans));
        d.clear();
        scanf("%d%d",&n,&m);
        for(int i=0; i<m; i++)
        {
            scanf("%d",&num);
            if(!flag) continue;
            if(num == 0) d.insert(i);
            else
            {
                ans[i] = -1; //最后的结果不输出下雨天的操作
                set<int>::iterator it = d.lower_bound(full[num]);
                if(it == d.end()) flag = false;
                else
                {
                    full[num] = i;
                    ans[*it] = num;
                    d.erase(*it);
                }
            }
        }
        if(!flag) printf("NO\n");
        else
        {
            bool first=true;
            printf("YES");
            for(int i=0; i<m; ++i)
                if(ans[i]>=0)
                    printf("%c%d",first?(first=false,'\n'):' ',ans[i]);
            printf("\n");
        }


    }



}
/*
4
2 4
0 0 1 1
2 4
0 1 0 2
2 3
0 1 2
2 4
0 0 0 1

NO
YES
1 2
NO
YES
0 1 0
*/

Hit