牛客I名家之壁
思路:
先看数据范围,只能用
O
(
n
)
O(n)
O(n)的写法去写,那么就用单调队列去维护最大最小值。
那么接下来就是怎么解决寻找区间
[
l
,
r
]
[l,r]
[l,r]的问题了。我们不妨固定
l
l
l,右移
r
r
r,可以发现,最大值不可能减小,最小值也不可能增大,也就是最大值和最小值的差值是非递减的。也就是如果有一点
r
r
r满足
max
l
≤
i
≤
r
a
[
i
]
−
min
l
≤
i
≤
r
a
[
i
]
>
m
\begin{aligned}\max\limits_{l\le i\le r}a[i]-\min\limits_{l\le i\le r}a[i]>m\end{aligned}
l≤i≤rmaxa[i]−l≤i≤rmina[i]>m,也就是
[
l
,
r
]
[l,r]
[l,r]满足条件的话,那么
[
l
,
r
+
1
]
,
[
l
,
r
+
2
]
,
⋯
,
[
l
,
n
]
[l,r+1],[l,r+2],\cdots,[l,n]
[l,r+1],[l,r+2],⋯,[l,n]都满足条件。那么如果固定
l
l
l,到
r
r
r满足时,
a
n
s
+
=
n
−
r
+
1
ans+=n-r+1
ans+=n−r+1。这时候再不断左移
l
l
l即可。
代码:
#include<bits/stdc++.h>
#define pii pair<int,int>
#define ll long long
#define cl(x,y) memset(x,y,sizeof(x))
#define ct cerr<<"Time elapsed:"<<1.0*clock()/CLOCKS_PER_SEC<<"s.\n";
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define all(x) x.begin(),x.end()
#define lson x<<1,l,mid
#define rson x<<1|1,mid+1,r
#define INF 1e18
const int N=1e6+10;
const int mod=1e9;
const int inf=0x3f3f3f3f;
const double eps=1e-8;
const double pi=acos(-1);
using namespace std;
struct edge
{
int v,id;
};
signed main()
{
ios::sync_with_stdio(false);
cin.tie(0);cout.tie(0);
int n,m,i,j=1,a,b,c;
ll ans=0;
cin>>n>>m>>a>>b>>c;
deque<edge> q1,q2;
for(i=1;i<=n;i++)
{
a=(ll)(a*b+c)%mod;
edge pre={a,i};
while(!q1.empty() && a>=q1.back().v)
q1.pop_back();
q1.push_back(pre);
while(!q2.empty() && a<=q2.back().v)
q2.pop_back();
q2.push_back(pre);
while(q1.front().v-q2.front().v>m)
{
ans+=n-i+1;
if(q1.front().id==j)
q1.pop_front();
if(q2.front().id==j)
q2.pop_front();
j++;
}
}
cout<<ans<<endl;
return 0;
}
}