这道题我是从样例中看出思路了
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;
}