二分。
题意:
给我们
m
m
m个地面高度和
n
n
n个炮弹高度,如果一个炮弹高度不大于当前的地面高度
a
[
i
]
a[i]
a[i],那么该炮弹就会停止,同时
a
[
i
−
1
]
+
1
a[i - 1] + 1
a[i−1]+1。如果当前的炮弹停在了第一个地面位置或者在最后一个地面位置都没停的话,视为无效炮弹。让我们输出最后每个地面位置的高度。
题解:
我们要把握一个性质,那就是,炮弹只会被第一个比它高的地面
i
i
i拦截下来,后面比
a
[
i
]
a[i]
a[i]矮的地面都不可能拦截到导弹,这样,我们可以在输入地面高度时维护一个非严格上升的序列。之后,对每个炮弹进行二分,找到第一个不低于它的地面高度
i
i
i,
a
[
i
−
1
]
+
1
a[i - 1] + 1
a[i−1]+1,并维护非严格上升序列当前位置的高度。
线段树也可以做。
实现细节见代码:
#include <bits/stdc++.h>
using namespace std;
const int MAXN = 5e4 + 10;
int m, n;
int a[MAXN], cur[MAXN];
void solve(int b) {
int l = 0, r = m;
while (l < r) {
int mid = l + r >> 1;
if (cur[mid] >= b) {
r = mid;
}
else {
l = mid + 1;
}
}
a[l - 1]++;
cur[l - 1] = max(cur[l - 1], a[l - 1]);
return;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0), cout.tie(0);
cin >> m >> n;
int maxx = -1;
for (int i = 0; i < m; i++) {
cin >> a[i];
cur[i] = max(cur[i - 1], a[i]);
}
for (int i = 0; i < n; i++) {
int b;
cin >> b;
if (b <= cur[0] || b > cur[m - 1]) continue; // 特判无效炮弹
solve(b);
}
for (int i = 0; i < m; i++) {
cout << a[i] << endl;
}
return 0;
}