Description
B进制数,每个数字i(i=0,1,…,B-1)有a[i]个。你要用这些数字组成一个最大的B进制数X(不能有前导零,不需要
用完所有数字),使得X是B-1的倍数。q次询问,每次询问X在B进制下的第k位数字是什么(最低位是第0位)。
第一行包含两个正整数B(2<=B<=10^6),q(1<=q<=10^5)。
第二行包含B个正整数a[0],a[1],a[2],…,aB-1。
接下来q行,每行一个整数k(0<=k<=10^18),表示一个询问。
输出q行,每行一个整数,依次回答每个询问,如果那一位不存在,请输出-1
Solution
回想起一个小时候玩过的游戏,就是给出一些两位的数字,把个位和十位相加的结果带入看运气
机智的我当然发现了猫腻:这些给定数字都是9的倍数!可以发现
27→2+7=936→3+6=9(188)(189)
(188)
27
→
2
+
7
=
9
(189)
36
→
3
+
6
=
9
诸如此类
因此我们做出一个猜想,一个十进制数加上9的倍数后每一位之和模9同余!
我们再大胆一点就是一个B进制数加上B-1的倍数后每一位之和模B-1同余!(强行引入
考虑证明这个东西。
当一个数的某一位+1时,若进位,则这一位要减去B-1,下一位要+1,则总的贡献是+1.
当一个数的某一位-1时,若退位,则这一位要加上B-1,下一位要-1,则总的贡献是-1.
由于所有的a不为0,找到模数减掉即可
Code
#include <stdio.h>
#include <string.h>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
typedef long long LL;
const int N=1000005;
LL a[N];
LL read() {
LL x=0,v=1; char ch=getchar();
for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):(v),ch=getchar());
for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
return x*v;
}
int main(void) {
int B,q,sum=0; scanf("%d%d",&B,&q);
rep(i,0,B-1) {
a[i]=read();
sum=(sum+1LL*i*a[i]%(B-1))%(B-1);
}
if (sum) a[sum]--;
rep(i,1,B-1) a[i]+=a[i-1];
for (;q--;) {
LL x=read()+1;
int tl=0,tr=B-1;
for (;tl<=tr;) {
int mid=(tl+tr)>>1;
if (a[mid]<x) tl=mid+1;
else tr=mid-1;
}
if (tr==B-1) puts("-1");
else printf("%d\n", tr+1);
}
return 0;
}