这道题我是从样例中看出思路了
2 4
0 0 1 1
看这组数据， 输出的是No， 为什么呢？因为两个1之间没有神龙喝水， 所以一定会有水灾。
然后就启发了我，两次同一个湖的降水之间必须至少有一次神龙喝水， 否则就会有水灾。
如果是第一个湖的话那么就看作在第0次有一次降水。
所以每一次找就用二分来找离前一次降水最近的那一次来喝水。
然后我思路是对的， 但是实现的时候想复杂了很多。
因为这个思路涉及不断地修改一个有序的数列， 我就想用vector， 然后用过就标记， 下一次找的时候用
一个while循环来略去标记过的。事实证明很复杂， 很容易写错。
然后看到博客， 看到用一个set来实现，非常的简洁方便， 核心代码非常的短。
因为set本身就是有序的， 同时自带二分， 而且删除很方便。
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<set>
#define REP(i, a, b) for(int i = (a); i < (b); i++)
using namespace std;const int MAXN = 1123456;
int ans[MAXN], pre[MAXN], rain[MAXN];int main()
{int T, n, m;scanf("%d", &T);while(T--){memset(pre, 0, sizeof(pre));memset(ans, 0, sizeof(ans));set<int> s;scanf("%d%d", &n, &m); REP(i, 0, m) scanf("%d", &rain[i]);bool ok = true;REP(i, 0, m){if(rain[i] == 0){s.insert(i);continue;} ans[i] = -1;auto it = s.lower_bound(pre[rain[i]]);if(it == s.end()) { ok = false; break; }ans[*it] = rain[i];pre[rain[i]] = i;s.erase(*it);}if(!ok) puts("NO");else{puts("YES");bool first = true;REP(i, 0, m)if(ans[i] != -1){if(first) first = false;else printf(" ");printf("%d", ans[i]);}puts("");}} return 0;
}