51nod1495 中国好区间

1495 中国好区间 

基准时间限制：0.7 秒 空间限制：131072 KB 分值: 80  难度：5级算法题

 收藏

 关注

阿尔法在玩一个游戏，阿尔法给出了一个长度为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示例

题解：这题的思路确实挺神奇，因为我们要求一个区间第k大的数大于等于T，所以我们直接将每个数字大于等于T的编号为1，小于的编号为0，这样我们只要查询总和大于等于k的区间的个数即可。

代码：

#include<bits/stdc++.h>
using namespace std;
long long i,n,m,T,a[1],b1,c,p,sum,b[10000001],l;
long long ans;
int main(){
	scanf("%d%d%d%d%d%d%d",&n,&m,&T,&a[0],&b1,&c,&p);
	for(i=1;i<=n;i++){
		a[0]=(a[0]*b1+c)%p;
		//printf("%d ",a[0]);
		if(a[0]>=T)b[i]=1;
	}
	//printf("\n");
	l=1;
	for(i=1;i<=n;i++){
		//printf("%d %d\n",i,sum);
		if(b[i])sum++;
		if(sum>=m){
			while(sum>=m&&l<=i){
				if(b[l])sum--;
				l++; 
			}
			//printf("%d\n",l);
		}
		ans+=l-1;
		//printf("%lld\n",ans);
	}
	printf("%lld",ans);
}