101487E - Enter The Dragon

思路：做的時候兩個地方理解錯了，第一個事我以為龍吸了水，水就幹了，其實龍是在下雨之前吸的，下雨時湖水又滿了，所以湖水永遠不會幹；第二個是以為只要找前面一段沒下雨的天吸水就可以了，其實是上一次滿了之後沒下雨的天吸水都可以。

代碼：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
#define mem(a,b) memset(a,b,sizeof(a))

const int N=1e6+5
 
;
set<int>s;
int a[N];
int ans[N];
int pre_full_pos[N];
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    int t,n,m;
    cin>>t;
    while(t--)
    {
        cin>>n>>m;
        for(int i=0;i<m;i++)cin>>a[i];
        s.clear();
        mem(pre_full_pos,
 
0);
        mem(ans,0);
        bool flag=false;
        for(int i=0;i<m&&!flag;i++)
        {
            if(a[i])
            {
                set<int>::iterator it=s.lower_bound(pre_full_pos[a[i]]);
                if(it!=s.end())
                {
                    ans[
 
*it]=a[i];
                    s.erase(it);
                    pre_full_pos[a[i]]=i;
                }
                else flag=true;
            }
            else s.insert(i); 
        }
        if(flag){
            cout<<"NO"<<endl;
            continue;
        }
        cout<<"YES"<<endl;
        for(int i=0;i<m;i++)if(!a[i])cout<<ans[i]<<‘ ‘;
        cout<<endl; 
    }
    return 0;
}
/*
10
4 10
0 0 3 4 0 0 1 0 4 3 
*/

Codeforces 101487E - Enter The Dragon