2018ACM浙江省赛 ZOJ 4029 Now Loading!!!（前缀和+二分）

题目描述：

DreamGrid has $n$ integers $a_1,a_2,\dots ,a_n$. DreamGrid also has $m$ queries, and each time he would like to know the value of $$\sum _{1\le i\le n}\lfloor \frac{a_i}{\lceil {\log }_p{a}_i\rceil }\rfloor$$ for a given number $p$, where ⌊ x ⌋ = max ⁡ { y ∈ Z ∣ y ≤ x } \lfloor x \rfloor = \max\{y \in \mathbb{Z} \mid y \le x\} ⌊x⌋=max{y∈Z∣y≤x}, ⌈ x ⌉ = min ⁡ { y ∈ Z ∣ y ≥ x } \lceil x \rceil = \min\{y \in \mathbb{Z} \mid y \ge x\} ⌈x⌉=min{y∈Z∣y≥x}.

Input

There are multiple test cases. The first line of input is an integer $T$ indicating the number of test cases. For each test case:

The first line contains two integers $n$ and $m$ ($1\le n,m\le 5\times 10^5$) – the number of integers and the number of queries.

The second line contains $n$ integers $a_1,a_2,\dots ,a_n$ ($2\le a_i\le 10^9$).

The third line contains $m$ integers $p_1,p_2,\dots ,p_m$ ($2\le p_i\le 10^9$).

It is guaranteed that neither the sum of all $n$ nor the sum of all $m$ exceeds $2\times 10^6$.

Output

For each test case, output an integer $(\sum _{i=1}^{m}i\cdot z_i)\mathrm{m}\mathrm{o}\mathrm{d}10^9$, where $z_i$ is the answer for the $i$-th query.

Sample Input

2
3 2
100 1000 10000
100 10
4 5
2323 223 12312 3
1232 324 2 3 5

Sample Output

11366
45619

翻译：

输入n和m，数组的长度为n，第二行的n个数表示a数组，接下来输入m个p，
对于每一个p，遍历a数组，求得$Z_i$

$$\sum _{1\le i\le n}\lfloor \frac{a_i}{\lceil {\log }_p{a}_i\rceil }\rfloor$$
可以得到m个这样的数$Z_i$

输出的结果： $(\sum _{i=1}^{m}i\cdot z_i)\mathrm{m}\mathrm{o}\mathrm{d}10^9$

分析：

分母的取值范围：
每一个$a_i$的范围为$[2,10^9]$,每一个p的范围为$[2,10^9]$，
对于分母：$$\lceil {\log }_p{a}_i\rceil$$
int的范围为2³¹-1(即2147483647)

分母的范围可以最小为1，31可以为上限

枚举每一个分母的值，对：
$$\frac{a_i}{\lceil {\log }_p{a}_i\rceil }$$ 进行前缀和

           for(int k=1; k<=30; k++)///枚举分母
            for(int i=1; i<=n; i++)
                sum[i][k]=sum[i-1][k]+a[i]/k;

输入每一个p，求的一个$Z_i$:

p和每一个$a_i$的大小密不可分，并且有向上取整，对a数组从小到大排序，可以用二分解决

初始化两个值 s1=p,s2=1和index=1;
从index=1枚举分母，通过二分找到对应的区间，记录结果，更新s1和s2

        for(int i=1; i<=m; i++)
        {
            LL ans=0,p;
            cin>>p;
            LL s1=p,s2=1;
            int index=1;
            while(true)
            {
                int pos1=upper_bound(a+1,a+1+n,s1)-(a+1);///upper找到第一个大于s1的下标，减去a+1,相当于又减去1，例如 2 3 4 5 upper_bound(3)返回的是位置3，但pos1=2
                int pos2=upper_bound(a+1,a+1+n,s2)-(a+1);
                ans+=sum[pos1][index]-sum[pos2][index];///[pos2,pos1]这个区间的分母都为index
                index++;
                s2=s1;
                s1*=p;///分母为1找完了，再找分母为2，分母为3，，
                if(pos1==n)
                    break;
            }
            res=(res+ans*i)%mod;
        }

完整代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;
const int mod=1e9;
const int N=5*1e5+10;
int t,n,m;
LL a[N],sum[N][35];
int main()
{
    cin>>t;
    while (t--)
    {
        cin>>n>>m;
        for (int i=1; i<=n; i++)
            cin>>a[i];
        sort(a+1,a+n+1);
        for(int k=1; k<=30; k++)///枚举分母
            for(int i=1; i<=n; i++)
                sum[i][k]=sum[i-1][k]+a[i]/k;
        LL res=0;
        for(int i=1; i<=m; i++)
        {
            LL ans=0,p;
            cin>>p;
            LL s1=p,s2=1;
            int index=1;
            while(true)
            {
                int pos1=upper_bound(a+1,a+1+n,s1)-(a+1);
                int pos2=upper_bound(a+1,a+1+n,s2)-(a+1);
                ans+=sum[pos1][index]-sum[pos2][index];
                index++;
                s2=s1;
                s1*=p;
                if(pos1==n)
                    break;
            }
            res=(res+ans*i)%mod;
        }
        cout<<res<<endl;
    }
    return 0;
}