Cod sursa(job #897626)

Utilizator marta_diannaFII Filimon Marta Diana marta_dianna Data 27 februarie 2013 21:29:46
Problema Tricouri Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.22 kb
#include<fstream>
#include<algorithm>
#define PMAX 21
#define KMAX 6
#define NMAX 300010

using namespace std;

ifstream f("tricouri.in");
ofstream g("tricouri.out");

int n, m, val[PMAX][PMAX][KMAX], a[NMAX], sum[KMAX], ind[PMAX], nr[KMAX][PMAX];

int i, p, z, k, mx, T;

void Back(int h)
{
    int i;
    if (h==k+1)
    {
        if (sum[k]%p==0)
            nr[k][p]=max(nr[k][p], sum[k]);
    }
    else
    {
        for (i=0; i<p; ++i)
            if (ind[i]<=val[p][i][0])
            {
                sum[h]=sum[h-1]+val[p][i][ind[i]++];
                Back(h+1);
                sum[h]=0; --ind[i];
            }
    }
}

int main()
{
    f>>n>>m;

    for (i=1; i<=n; ++i) f>>a[i];

    sort(a+1, a+n+1);

    for (i=n; i>0; --i)
    {
        for (p=2; p<=20; ++p)
        {
            z=a[i]%p;

            if (val[p][z][0]<5)
                val[p][z][++val[p][z][0]]=a[i];
        }
    }

    for (k=1; k<6; ++k)
        for (p=2; p<=20; ++p)
        {
            for (i=0; i<p; ++i) ind[i]=1;
            Back(1);
        }

    for (T=1; T<=m; ++T)
    {
        f>>k>>p;

        /*mx=-1;

        Back(1);*/

        if (!nr[k][p]) g<<"-1\n";
        else g<<nr[k][p]<<"\n";
    }

    f.close();
    g.close();
    return 0;
}