阿尔法在玩一个游戏,阿尔法给出了一个长度为n的序列,他认为,一段好的区间,它的长度是>=k的,且该区间的第k大的那个数,一定大于等于T。那么问题来了,阿尔法想知道有多少好的区间。
由于阿尔法的序列长度实在是太大了,无法在规定时间内读入。
他想了一个绝妙的方法。
读入a[0],b,c,p,则a[i]=(a[i-1]*b+c)mod p。
样例解释:
a1~a5分别为47,135,247,35,147
对应的7个区间分别为[1,3],[2,3],[1,4],[2,4],[1,5],[2,5],[3,5]
对于重复的数字1,2,2 第一大是2,第二大也是2,第三大是1。
输入
读入一行,7个数字,表示n(n<=10000000),k(k<=n),T,a[0],b,c,p。
所有数字均为正整数且小于等于10^9。
输出
输出一行表示好区间的个数。
输入样例
5 2 100 10 124 7 300
输出样例
7
标记头尾,先使数组第一次达到有k个数>=T,想象一下毛毛虫
然后头缩进,尾后拉,不断更新达到要求的数,答案累加
#include<bits/stdc++.h>
using namespace std;
#define ll long long
ll dp[10000010];
int main()
{
ios::sync_with_stdio(false);
ll n,k,t,b,c,p,tmp,l,r,ans=0;
cin>>n>>k>>t>>tmp>>b>>c>>p;
for(int i=1;i<=n;i++)
{
tmp=(tmp*b+c)%p;
dp[i]=dp[i-1];
if(tmp>=t) dp[i]++;
}
l=1;r=k;
while(1)
{
while(r<=n&&dp[r]-dp[l-1]<k)
r++;
if(r>n) break;
ans+=n-r+1;
l++;
}
cout<<ans;
return 0;
}