Cod sursa(job #18103)

Utilizator dominoMircea Pasoi domino Data 18 februarie 2007 05:06:24
Problema Tricouri Scor 60
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.29 kb
#include <stdio.h>
#include <algorithm>
#include <vector>

using namespace std;

#define MAX_N 300005
#define MAX_P 25
#define FIN "tricouri.in"
#define FOUT "tricouri.out"

int N, M, K, P, A[MAX_N], pos[MAX_P][MAX_P], t, Res;
vector<int> V[MAX_P][MAX_P];

void bkt(int lv, int mod)
{
    int i;

    if (lv == K-1)
    {
        i = (P-mod)%P;
        if (pos[P][i] == V[P][i].size()) return;
        t -= V[P][i][pos[P][i]];
        if (Res < t) Res = t;
        t += V[P][i][pos[P][i]];
        return;
    }

    for (i = 0; i < P; i++)
    {
        if (pos[P][i] == V[P][i].size()) continue;
        t -= V[P][i][pos[P][i]++];
        bkt(lv+1, (mod+i)%P);
        t += V[P][i][--pos[P][i]];
    }
}

int main(void)
{
    int i, j;

    freopen(FIN, "r", stdin);
    freopen(FOUT, "w", stdout);

    scanf("%d %d", &N, &M);
    for (i = 0; i < N; i++)
        scanf("%d", A+i);

    for (i = 2; i <= 20; i++)
        for (j = 0; j < N; j++)
            V[i][A[j]%i].push_back(-A[j]);

    for (i = 2; i <= 20; i++)
        for (j = 0; j < i; j++)
            sort(V[i][j].begin(), V[i][j].end());

    for (; M; M--)
    {
        scanf("%d %d", &K, &P);
        Res = -1; t = 0;
        bkt(0, 0);
        printf("%d\n", Res);
    }

    return 0;
}