阿尔法在玩一个游戏,阿尔法给出了一个长度为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。
Input
读入一行,7个数字,表示n(n<=10000000),k(k<=n),T,a[0],b,c,p。 所有数字均为正整数且小于等于10^9。
Output
输出一行表示好区间的个数。
Input示例
5 2 100 10 124 7 300
Output示例
7
#include <iostream>
#include <cstring>
using namespace std;
typedef long long ll;
const int MAXN = 1e7 + 5;
int n, k, T, b, c, p;
ll a[MAXN];
ll solve()
{
int count = 0;
ll right = 1;
ll left = 1;
ll result = 0;
while (right <= n)
{
if (a[right] >= T)
{
count++;
}
while (count > k)
{
if (a[left] >= T)
{
count--;
}
left++;
}
if (count == k)
{
while (a[left] < T)
{
left++;
}
result += left;
}
right++;
}
return result;
}
int main()
{
cin >> n >>k >> T >> a[0] >> b >> c >> p;
for (int i = 1; i <= n; i++)
{
a[i] = (a[i - 1] * b + c) % p;
}
ll result = solve();
cout << result << endl;
return 0;
}