4.2 perguntas US Missão caneta 2 (árvores Fenwick)

Aqui Insert Picture Descrição

resolução:

Manter uma soma (colo e para baixo), o loop interno, cada parte pode colocar cálculos círculo completo (divisão) de módulo

A situação não pode ser um círculo completo (soma> m), tem, obviamente, uma soma prefixo comum> m, de modo que metade dos prefixos (prefixos e uso dinâmico manutenção árvore Fenwick), a primeira posição é maior do que a do Suprimido prefixo m fora.

Para mais detalhes, olhe para o código:

/*
 *  Author : Jk_Chen
 *    Date : 2020-04-02-20.14.15
 */
#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define rep(i,a,b) for(int i=(int)(a);i<=(int)(b);i++)
#define per(i,a,b) for(int i=(int)(a);i>=(int)(b);i--)
#define mmm(a,b) memset(a,b,sizeof(a))
#define pb push_back
#define pill pair<int, int>
#define fi first
#define se second
void test(){cerr<<"\n";}
template<typename T,typename... Args>void test(T x,Args... args){cerr<<x<<" ";test(args...);}
const LL mod=1e9+7;
const int maxn=1e5+9;
const int inf=0x3f3f3f3f;
LL rd(){ LL ans=0; char last=' ',ch=getchar();
    while(!(ch>='0' && ch<='9'))last=ch,ch=getchar();
    while(ch>='0' && ch<='9')ans=ans*10+ch-'0',ch=getchar();
    if(last=='-')ans=-ans; return ans;
}
#define rd rd()
/*_________________________________________________________begin*/

int n;
LL a[maxn];

LL tr[maxn];
void update(int pos,LL val){
    while(pos<=n){
        tr[pos]+=val;
        pos+=pos&-pos;
    }
}
LL query(int pos){
    LL res=0;
    while(pos>0){
        res+=tr[pos];
        pos-=pos&-pos;
    }
    return res;
}


int main(){
    n=rd;
    LL m=rd;
    LL sum=0,res=n;

    rep(i,1,n){
        a[i]=rd;
        sum+=a[i];
        update(i,a[i]);
    }
    LL ans=0;
    while(res>0&&m){
        LL lun=m/sum;
        m=m%sum;
        ans+=lun*res;
        int l=1,r=n,A=-1;
        while(l<=r){
            int mid=(l+r)>>1;
            if(query(mid)>m)A=mid,r=mid-1;
            else l=mid+1;
        }
        update(A,-a[A]);
        res--;
        sum-=a[A];
        continue;
    }
    printf("%lld\n",ans);
    return 0;
}

/*_________________________________________________________end*/
Publicado 773 artigos originais · ganhou elogios 345 · Exibições 200.000 +

Acho que você gosta

Origin blog.csdn.net/jk_chen_acmer/article/details/105279440
Recomendado
Clasificación