51nod1495 中国好区间 思维

题目大意：

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

思路：可以求出所有的ai,dt[i]记录前i个数中大于等于t的个数，显然若（a[i]>=t)则dt[i] = dt[i-1] + 1;否则dt[i] = dt[i-1];接下来计算没个数对答案的贡献，因为要求区间第k大数大于等于t，所以每个区间至少包含k个大于等于t的数，所以对于没个数a[i] ，只需要dt[i]>=k时，找到前面比t大的数有dt[i]-k+1的第一个大于等于t的数出现的a[j]的位置就是该数贡献的答案。跑一边就可以了，O(n).

#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <fstream>
#include <algorithm>
#include <cmath>
#include <queue>
#include <stack>
#include <vector>
#include <map>
#include <set>
#include <iomanip>

using namespace std;
#define maxn 10000005
#define MOD 1000000007
#define mem(a , b) memset(a , b , sizeof(a))
#define LL long long
#define INF 1000000000
int a[maxn];
int vis[maxn] , dt[maxn];
int n , k , t , b , c , p;
LL ans = 0;

void init()
{
    LL ss = 0;
    for(int i = 1 ; i <= n ; i ++)
    {
        //a[i] = (mult_mod(a[i-1] , b , p) % p + c) % p;
        ss = a[i-1];
        a[i] = (ss * b + c) % p;
        dt[i] = dt[i-1];
        if(a[i] >= t) dt[i] += 1;
        if(dt[i] && !vis[dt[i]]) vis[dt[i]] = i;
    }
}

int main()
{
    while(scanf("%d %d %d %d %d %d %d" , &n , &k , &t , &a[0] , &b , &c , &p) != EOF)
    {
        dt[0] = 0;
        mem(vis , 0);
        ans = 0;
        init();
        int i = k;
        while(dt[i] < k) i ++;
        for( ; i <= n ; i ++)  ans += (vis[dt[i] - k + 1]);
        printf("%lld\n" , ans);
    }
    return 0;
}