ZOJ4029 浙江省赛

之前想到了先枚举logp(a[i])=k，然后找到对应区间，之后发现向下取整并无法直接用前缀和去维护，于是就继续二分k-2k,2k-3k这种区间，然而显然是过不了的，TLEGG,不过最后挂机也只能写写暴力了。

结果发现k只有31，所以sum[k][i]来预处理a[i]/k向下取整，的前缀和,30*n的复杂度，那么枚举出k后二分找到对应区间就能直接求出这段区间的value之和了。

注意，减法部分要+mod)%mod，WA了好久。。。这种小细节又错了，菜哭.jpg

#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxl 500010
#define mod 1000000000
#define eps 1e-8

using namespace std;

int n,m;
int a[maxl];
long long sum[32][maxl];
long long ans;

inline void prework()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	sort(a+1,a+1+n);
	for(int k=1;k<=31;k++)
		for(int i=1;i<=n;i++)
		{
			sum[k][i]=a[i]/k;
			sum[k][i]=(sum[k][i]+sum[k][i-1])%mod;
		} 
}

inline void mainwork()
{
	long long curans,mi,p,x;
	int l,r;ans=0;
	for(int cas=1;cas<=m;cas++)
	{
		curans=0;mi=1;
		scanf("%d",&p);
		x=p;l=1;
		while(l<=n)
		{
			while(x<a[l])
			{
				x*=p;
				mi++;
			}
			r=upper_bound(a+l,a+n+1,x)-a;
			curans=(curans+(sum[mi][r-1]-sum[mi][l-1]+mod)%mod)%mod;
			l=r;
		}
		ans=(cas*curans+ans)%mod;
	}
}

inline void print()
{
	printf("%lld\n",ans);
}

int main()
{
	int t;
	scanf("%d",&t);
	for(int i=1;i<=t;i++)
	{
		prework();
		mainwork();
		print();
	}
	return 0;
}