[TC14088]LimitedMemorySeries1

[TC14088]LimitedMemorySeries1

题目大意：

给定长度为\(n(n\le5\times10^6)\)的数组\(X\)，询问不超过\(q(q\le100)\)次，每次询问第\(k_i\)大的数是多少。
数组\(X\)生成方式如下：

X[0] = x0
for i = 1 to n-1:
        X[i] = (X[i-1] * a + b) % (10^9+7)

内存大小1MB，时间1S。

思路：

内存这么紧肯定不能开一个长度为\(n\)的数组，因此按照值域分块。先对每一个块开桶，统计每一块有多少数，确定答案在哪一块中。然后对于块中的每一个数统计个数。

时间复杂度\(\mathcal O(nq)\)。

源代码：

#include<vector>
#include<cstring>
class LimitedMemorySeries1 {
    private:
        using int64=long long;
        static constexpr int B=31623,mod=1e9+7;
        int cnt[B];
    public:
        int64 getSum(const int &n,const int &x0,const int &a,const int &b,std::vector<int> query) {
            int64 ans=0;
            for(auto k:query) {
                memset(cnt,0,sizeof cnt);
                for(register int i=0,x=x0;i<n;i++) {
                    cnt[x/B]++;
                    x=((int64)x*a%mod+b)%mod;
                }
                int p=0,q=0;
                for(;p<B;p++) {
                    if(k-cnt[p]<=-1) break;
                    k-=cnt[p];
                }
                memset(cnt,0,sizeof cnt);
                for(register int i=0,x=x0;i<n;i++) {
                    if(x/B==p) cnt[x%B]++;
                    x=((int64)x*a%mod+b)%mod;
                }
                for(;q<B;q++) {
                    if(k-cnt[q]<=-1) break;
                    k-=cnt[q];
                }
                ans+=(int64)p*B+q;
            }
            return ans;
        }
};

转载于:https://www.cnblogs.com/skylee03/p/9708754.html