ZOJ 4029 Now Loading!!!（数论+二分）

Description

给出$a_1,...,a_n$和$p_1,...,p_m$，求$\sum\limits_{i=1}^mi\cdot \sum\limits_{j=1}^n\lfloor\frac{a_j}{\lceil log_{p_i}a_j\rceil}\rfloor$

Input

第一行一整数$T$表示用例组数，每组用例首先输入两个整数$n,m$，之后输入$n$个整数$a_1,...,a_n$和$m$个整数$p_1,...,p_m$$(1\le n,m\le 5\cdot 10^5,2\le a_i,p_i\le 10^9)$

Output

输出一个整数表示结果，答案模$10^9$

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

Solution

考虑每个$a_j$对答案的贡献，由于$\lceil log_{p_i}a_j \rceil$的取值必然介于$[1,31]$之间，在把$p_i$从小到大排完序后，这$m$个数会被分成不超过$31$段，其中每段的$\lceil log_{p_i}a_j \rceil$取值相同，只要维护一个$p_i$下标的前缀和就可以$O(31)$的计算$a_j$对答案的贡献，分段需要二分$\lceil log_{p_i}a_j \rceil$的取值

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=500005;
int T,n,m,a[maxn];
ll sum[maxn];
#define val first
#define id second 
#define mod 1000000000
P p[maxn];
ll Pow(ll a,int b)
{
    ll ans=1;
    while(b)
    {
        if(b&1)ans=ans*a;
        a=a*a;
        b>>=1;
    }
    return ans;
}
int find(int x,int j,int r)//p^j>=x的最小j 
{
    int l=1,ans=-1,mid;
    while(l<=r)
    {
        mid=(l+r)/2;
        if(Pow(p[mid].val,j)>=x)ans=mid,r=mid-1;
        else l=mid+1;
    }
    return ans;
}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        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].val),p[i].id=i;
        sort(p+1,p+m+1);
        sum[0]=0;
        for(int i=1;i<=m;i++)sum[i]=p[i].id+sum[i-1];
        ll ans=0;
        for(int i=1;i<=n;i++)
        {
            int num=0;
            ll temp=1;
            while(1)
            {
                if((ll)temp*p[m].val<a[i])temp*=p[m].val;
                else break; 
            }
            int pre=m;
            for(int j=num;j<=31;j++)
            {
                int pos=find(a[i],j,pre);
                if(pos!=-1)
                {
                    ans+=(ll)(sum[pre]-sum[pos-1])*(a[i]/j)%mod;
                    if(ans>=mod)ans-=mod;
                    pre=pos-1;
                }
                if(pre==0)break;
            }
        }
        printf("%lld\n",ans);
    }
    return 0;
}