题目
题目描述
给定
n
n
n 个正整数的序列
a
1
,
a
2
,
a
3
,
…
,
a
n
a_1,a_2,a_3,\dots,a_n
a1,a2,a3,…,an ,对该序列可执行下面的操作:
- 选择一个大于 k k k 的正整数 a i a_i ai ,将 a i a_i ai 的值减去 1 1 1 ;然后选择 a i − 1 a_{i-1} ai−1 和 a i + 1 a_{i+1} ai+1 中的一个,将值加上 1 1 1 。
共给出 m m m 个正整数 k k k ,对于每次给定的正整数 k k k ,经过以上操作一定次数后,求出最长的一个连续子序列,使得这个子序列的每个数都不小于给定的 k k k 。
输入格式
第一行两个正整数
n
n
n 和
m
m
m 。
第二行 n n n 个正整数,第 i i i 个正整数表示 a i a_i ai 。
第三行 m m m 个正整数,第 i i i 个正整数表示第 i i i 次给定的 k k k 。
输出格式
共一行,输出
m
m
m 个正整数,第
i
i
i 个数表示对于第
i
i
i 次给定的
k
k
k ,经过一定次数操作后,最长连续子序列的长度。
数据范围与约定
对于
30
%
30\%
30% 的数据,
n
<
30
n<30
n<30 。
对于 50 % 50\% 50% 的数据, n ≤ 2000 n\le 2000 n≤2000 。
对于 100 % 100\% 100% 的数据, n ≤ 1 0 6 , m ≤ 50 , k ≤ 1 0 9 , a i ≤ 1 0 9 n\le 10^6, m\le 50, k\le 10^9, a_i\le 10^9 n≤106,m≤50,k≤109,ai≤109 。
思路
对于一个最长的连续子序列 [ l , r ] \left[l,r\right] [l,r] ,由定义有 a l − 1 < k , a r + 1 < k a_{l-1}<k,a_{r+1}<k al−1<k,ar+1<k 。
因此, ∑ i = l r a i \sum_{i=l}^{r}a_i ∑i=lrai 不会变大。别人进不来嘛!
由于现在的 ∑ i = l r a i ≥ k ( r − l + 1 ) \sum_{i=l}^{r}a_i\ge k(r-l+1) ∑i=lrai≥k(r−l+1),所以原来的 ∑ i = l r a i ≥ k ( r − l + 1 ) \sum_{i=l}^{r}a_i\ge k(r-l+1) ∑i=lrai≥k(r−l+1) 。而且这是充分的,内部进行调整,一定可以全部获得至少 k k k 。
令 s i = ∑ j = 1 i a j s_i=\sum_{j=1}^{i}a_j si=∑j=1iaj ,那么 ( l , r ] (l,r] (l,r] 可以成为答案,当且仅当 ( s r − k r ) − ( s l − k l ) ≥ 0 (s_r-kr)-(s_l-kl)\ge 0 (sr−kr)−(sl−kl)≥0 。令 f ( x ) = s x − k x f(x)=s_x-kx f(x)=sx−kx ,我们要求的就是最远的正序对。
这可以用 单调栈。先求出一个递减单调栈(因为“靠左” + + +“小” = = =优秀),然后从右往左枚举右端点(此时可以弹栈顶——右端点更小,只有左端点也更小才有可能更优)。
代码
#include <cstdio>
#include <iostream>
#include <vector>
using namespace std;
inline int readint(){
int a = 0, f = 1; char c = getchar();
for(; c<'0' or c>'9'; c=getchar())
if(c == '-') f = -1;
for(; '0'<=c and c<='9'; c=getchar())
a = (a<<3)+(a<<1)+(c^48);
return a*f;
}
struct Edge{
int to, nxt, data;
Edge(){ to = nxt = data = 0; }
Edge(int T,int N,int D):to(T),nxt(N){
data = D;
}
};
const int MaxN = 1000000;
int n, m; long long s[MaxN+1];
void init(){
n = readint(), m = readint();
for(int i=1; i<=n; ++i)
s[i] = readint()+s[i-1];
}
vector<int> v;
# define val(r) (s[(r)]-1ll*(r)*x)
void work(int x){
v.clear(), v.push_back(0);
int ans = 0;
for(int i=1; i<=n; ++i)
if(not v.empty() and val(v.back()) > val(i))
v.push_back(i);
for(int i=n; i>=1; --i)
while(not v.empty() and val(v.back()) <= val(i))
ans = max(ans,i-v.back()), v.pop_back();
printf("%d ",ans);
}
# undef val
void solve(){
while(m --)
work(readint());
}
int main(){
init();
solve();
return 0;
}