4724: [POI2017]Podzielno

4724: [POI2017]Podzielno

Time Limit: 10 Sec   Memory Limit: 128 MB
Submit: 229   Solved: 108
[ Submit][ Status][ Discuss]

Description

B进制数，每个数字i(i=0,1,...,B-1)有a[i]个。你要用这些数字组成一个最大的B进制数X(不能有前导零，不需要

用完所有数字)，使得X是B-1的倍数。q次询问，每次询问X在B进制下的第k位数字是什么(最低位是第0位)。

Input

第一行包含两个正整数B(2<=B<=10^6),q(1<=q<=10^5)。

第二行包含B个正整数a[0],a[1],a[2],...,a[B-1](1<=a[i]<=10^6)。

接下来q行，每行一个整数k(0<=k<=10^18)，表示一个询问。

Output

输出q行，每行一个整数，依次回答每个询问，如果那一位不存在，请输出-1。

Sample Input

3 3
1 1 1
0
1
2

Sample Output

0
2
-1

HINT

Source

鸣谢Claris上传

[ Submit][ Status][ Discuss]
若B进制数码x是(B - 1)的倍数 必定满足，B进制下各个位置的数的和是B - 1的倍数 证明显然，考虑B进制下第k位，必定写成t * B^k的形式，而t * B^k % (B - 1) = t 那么把读入的所有数字乘上权值加起来 如果结果mod B - 1不为0,把结果所在位置减去1即可 因为要数字最大，优先是位数最多 将剩下的数字升序排好，自然就得到最大的B进制数了

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
 
const int maxn = 1E6 + 10;
typedef long long LL;
 
int m;
LL sum,B,A[maxn];
 
inline LL getLL()
{
    char ch = getchar(); LL ret = 0;
    while (ch < '0' || '9' < ch) ch = getchar();
    while ('0' <= ch && ch <= '9')
        ret = ret * 10LL + 1LL * (ch - '0'),ch = getchar();
    return ret;
}
 
int main()
{
    #ifdef DMC
        freopen("DMC.txt","r",stdin);
    #endif
     
    B = getLL(); m = getLL();
    for (int i = 0; i < B; i++)
    {
        A[i] = getLL();
        sum += 1LL * i * A[i];
        sum %= (B - 1);
    }
    if (sum) --A[sum]; A[B] = 1E18 + 233LL;
    for (int i = 1; i < B; i++) A[i] += A[i - 1];
    for (int i = 0; i < B; i++) A[i] -= 1LL;
    while (m--)
    {
        LL x = getLL();
        int pos = lower_bound(A,A + B + 1,x) - A;
        printf("%d\n",pos == B ? -1 : pos);
    }
    return 0;
}