ZOJ - 4029 Now Loading!!! 字首和+二分
阿新 • • 發佈:2018-11-29
題意: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; }