bzoj4724 [POI2017]Podzielno 二分

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的倍数！可以发现$\begin{align}27\rightarrow 2+7=9\\ 36\rightarrow 3+6=9\end{align}$
诸如此类
因此我们做出一个猜想，一个十进制数加上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;
}