【BZOJ】4724: [POI2017]Podzielno

传送门:bzoj4724

---

题解

列个式子(这里懒得贴了)，我们就会发现：
$a*B^k\equiv a \pmod {(B-1)}$
于是就变成各数位之和要成倍数。
为了保证结果最大，若此时所有数之和模B-1仍有余，那么因为保证了每一位上个数大于等于1，我们直接去掉对应位上的数就好了(保证位数最多且最大)。
然后就很简单了。

---

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cctype>
using namespace std;
typedef long long ll;
const int N=1e6+10,M=1e5+10;
int n,m,ans[N],cnt;
ll sum,tot,a[N];

struct Q{
    ll pos;int num;
    bool operator <(const Q& u)const{
       return u.pos<pos;
    }
}q[M];

int main(){
    scanf("%d%d",&n,&m);
    for(int i=0;i<n;++i){
      scanf("%lld",&a[i]);tot+=a[i];sum=(sum+1ll*i*a[i]%(n-1))%(n-1);
      }
    if(sum!=0){a[sum]--;tot--;}
    for(int i=1;i<=m;++i){scanf("%lld",&q[i].pos);q[i].pos++;q[i].num=i;}
    sort(q+1,q+m+1);
    cnt=1;
    while(q[cnt].pos>tot && cnt<=m) 
      {ans[q[cnt].num]=-1;cnt++;}
    for(int i=n-1;i>=0;i--){
        while(q[cnt].pos>(tot-a[i]) && cnt<=m) 
        {ans[q[cnt].num]=i;cnt++;}
        tot-=a[i];
    }
    for(int i=1;i<=m;i++) printf("%d\n",ans[i]);
}