Cod sursa(job #1945600)

Utilizator Daria09Florea Daria Daria09 Data 29 martie 2017 16:32:40
Problema Caramizi Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.71 kb
#include <iostream>
#include <fstream>
#define DIM 1000010
#include <algorithm>
using namespace std;
ifstream f("caramizi.in");
ofstream g("caramizi.out");
int n,m,a[DIM],MAX;
long long s[DIM],sol[DIM];
void read()
{
    f>>n>>m;
    for(int i=1;i<=n;i++)
        f>>a[i];
}
void solve()
{
    sort(a+1,a+n+1);
    for(int i=1;i<=n;i++)s[i]=a[i]+s[i-1];
    MAX=1000007;
    int k=1;
    for(int i=1;i<=MAX;i++)
    {
        while(i>=a[k]&&k<=n)++k;
        long long rez=1LL*s[k-1]+1LL*(n-k+1)*i;
        sol[i]=max(sol[i-1],rez-rez%i);
    }
    int d;
    for(m=m;m>0;--m)
    {
        f>>d;
        g<<sol[d]<<'\n';
    }
}
int main()
{
    read();
    solve();
    return 0;
}