【bzoj1495】中国好区间

来源:互联网 发布:天刀男号捏脸数据 编辑:程序博客网 时间:2024/06/11 20:54

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

题解
a[i] > =T则a[i]=1
a[i] < T则a[i]=0
问题转化为求区间和大于等于k的个数

代码

#include<bits/stdc++.h>#define mod 1000000007#define inf 1000000005#define N 100005#define pa pair<int,int>typedef long long ll;using namespace std;inline int read(){    int x=0,f=1;char ch=getchar();    while (ch<'0'||ch>'9'){if (ch=='-')f=-1;ch=getchar();}    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}    return x*f;}ll T,a,b,c,p,ans;ll sum[10000005],n,k;int main(){    cin>>n>>k>>T>>a>>b>>c>>p;    for (int i=1;i<=n;i++)    {        a=(a*b+c)%p;        if (a>=T) sum[i]=sum[i-1]+1;else sum[i]=sum[i-1];    }    ll l=0,r=1;    while (sum[r]-sum[0]<k) r++;    while (r<=n)    {        while (sum[r]-sum[l]>=k) l++;        ans+=l;        r++;    }    cout<<ans;    return 0;}
原创粉丝点击