ZOJ - 4029 Now Loading!!! 前缀和+二分

题意:m次计算,对于给出的式子求和

题解:因为分母取了log 所以分母的范围就在1-30 之间 ,所以数组a先排序,预处理求一下 除每个数的前缀和,对于每一个p,二分查找对应的区间即可,复杂度 n*log(n)*log(n),数组不要开太大 会报ML

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
#define INF 0x3f3f3f3f
typedef long long ll;
const int N=5e5+10;
const ll mod=1e9;
int a[N],p;
ll sum[31][N];
int n,m;
int main()
{
    int T;
    int nn=1;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)scanf("%d",&a[i]);
        sort(a+1,a+1+n);
        for(int i=1;i<=30;i++)
            for(int j=1;j<=n;j++)
                sum[i][j]=(sum[i][j-1]+a[j]/i)%mod;
        ll res=0;
        ll p1,cnt,p2,k,ans;
        for(int i=1;i<=m;i++)
        {
            scanf("%lld",&p);
            p1=0,cnt=1,p2,k=1,ans=0;
            while(p1<n)
            {
                p2=upper_bound(a+1,a+1+n,cnt*p)-a;
                ans=((ans+sum[k][p2-1]-sum[k][p1])%mod+mod)%mod;
                p1=p2-1;
                cnt*=p;
                k++;
            }
            res=(res+ans*i%mod)%mod;
        }
        printf("%lld\n",(res%mod+mod)%mod);
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/mmk27_word/article/details/84382142
now