题目链接
https://nuoyanli.com/contest/1/problem/3D
题面
题意
给定 n n n组 a i , b i a_i,b_i ai,bi, q q q次询问,每次给 k k k,问至少多少到第几个 i i i才有 k ≤ m a x 1 m ( a i − b i ) k \leq max_1^m(a_i-b_i) k≤max1m(ai−bi)。找出这个 m m m。
思路
因为是求和,所以前缀和处理一下,由于
n
n
n和
q
q
q都在
1
0
6
10^6
106所以暴力找肯定超时(数据没给满导致很多人水过Orz),所以考虑二分,由于变化
(
a
i
−
b
i
)
(a_i-b_i)
(ai−bi)可能为负数,所以前缀和不具单调性,但是不难发现如果
(
a
i
−
b
i
)
≤
0
(a_i-b_i )\leq 0
(ai−bi)≤0这个
i
i
i一定不会是答案,因为
∑
1
i
(
a
i
−
b
i
)
≤
∑
1
i
−
1
(
a
i
−
1
−
b
i
−
1
)
\sum_1^i(a_i-b_i)\leq\sum_1^{i-1}(a_{i-1}-b_{i-1})
∑1i(ai−bi)≤∑1i−1(ai−1−bi−1)又要天数尽可能小(
m
m
m),显然
i
−
1
i-1
i−1更优,所以我排除这些变化为负的重新组成一个序列,这个序列具有单调性,所以二分可行。
ps:其实就是维护一个严格单调递增的序列(启发于单调栈思想)。
注意
l
o
n
g
l
o
n
g
!
long\ long!
long long!
参考代码
#include<bits/stdc++.h>
using namespace std;
const int N = 1e6 + 7;
typedef long long LL;
LL a[N], b[N], sum[N];
LL que[N], id[N],n, q,k;
int main() {
scanf("%d%d",&n,&q);
for (int i = 1; i <= n; i++) {
scanf("%lld",&a[i]);
}
for (int i = 1; i <= n; i++) {
scanf("%lld",&b[i]);
}
for (int i = 1; i <= n; i++) {
sum[i] = sum[i - 1] + a[i] - b[i];
}
int cnt = 0;
que[++cnt] = 0;
id[cnt] = 0;
for (int i = 1; i <= n; i++) {
if (sum[i] > que[cnt])
que[++cnt] = sum[i], id[cnt] = i;
}
while (q--) {
scanf("%lld",&k);
int pos = lower_bound(que + 1, que + 1 + cnt, k) - que;
printf("%d\n",id[pos]);
}
return 0;
}