来自学长的快乐AK题——Day8 荒地追猎

荒地追猎——数论

Description

数据规模

思路

设x_i表示第i位的话就是
$$(\sum _{i=0}{x}_i\ast B^i)\%(B-1)=0$$
化简一下得：
$$\sum _{i=0}(x_i\ast B^i\%(B-1))=0$$
拆开单独的一个来看：
$$x_i\ast B^i\%(B-1)=(x_i\%(B-1))\ast (B^i\%(B-1))=(x_i\%(B-1))\ast 1$$
求和后：
$$\sum _{i=0}(x_i\%(B-1)\ast 1)=(\sum _{i=0}{x}_i)\%(B-1)=0$$

所以其实就是各位数字的和为B-1的倍数就好了。

然后再回头看题目发现有限制a_i>=1。这样如果用上所有数字的和对B-1取模为t的话，若t不为0，我们就让a_t减去一个1就好了。

因为要让结果最大，所以更大的数肯定放在更高的位上，更小的数肯定放在更低的位上。因此对于从小到大的数，其所放的位也是从低到高。
然后对于一个询问，求a数组的前缀和然后二分。

代码

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long ll;
const ll N=1e6;
ll b,q,a[N],s[N],sum;
int main()
{
	scanf("%lld%lld",&b,&q);
	for(ll i=0;i<b;i++)
	scanf("%lld",&a[i]),sum=(sum+a[i]*i)%(b-1);
	if(sum) a[sum]--;
	for(ll i=0;i<b;i++)
	s[i+1]=s[i]+a[i];
	for(ll i=1,x;i<=q;i++)
	{
		scanf("%lld",&x);
		ll p=lower_bound(s+1,s+b+1,x+1)-s-1;
		if(p>=b) p=-1;
		printf("%lld\n",p);
	}
	return 0;
}