ZOJ 4029 Now Loading!!! (思维)

题目链接：http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=4029

思路：朴素算法复杂度O(n*m)，考虑如何减少遍历，无非是n或m不能一个个去遍历。把n分成若干个区间，每个区间log的值都一样。然后预处理一下前缀和就能出结果了。

#include <bits/stdc++.h>
typedef long long ll;
using namespace std;
const int MOD=1e9;
const int N=5e5+5;
int n,m,a[N],p[N];
ll res;
int dp[31][N];
void init()
{
    res=0;
}
void read()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)    scanf("%d",&a[i]);
    for(int i=1;i<=m;i++)   scanf("%d",&p[i]);
}
void precalc()
{
    sort(a+1,a+1+n);
    for(int i=1;i<=30;i++)
    {
        dp[i][0]=0;
        for(int j=1;j<=n;j++)
        {
            dp[i][j]=(dp[i][j-1]+a[j]/i)%MOD;
        }
    }
}
void calc()
{
    for(int i=1;i<=m;i++)
    {
        ll temp=1,tsum=0;
        int now=1;
        for(int j=1;j<=30;j++)
        {
            temp*=p[i];
            int index=upper_bound(a+now,a+n+1,temp)-a;
            if(index!=now)
            {
                tsum=(tsum+(dp[j][index-1]-dp[j][now-1])+MOD)%MOD;
                now=index;
            }
            if(index==n+1)    break;
        }
        res=(res+i*tsum)%MOD;
    }
}
void print()
{
    printf("%lld\n",res);
}
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        init();
        read();
        precalc();
        calc();
        print();
    }
    return 0;
}