Cod sursa(job #2097605)

Utilizator georgerapeanuRapeanu George georgerapeanu Data 31 decembrie 2017 22:16:30
Problema Ghiozdan Scor 90
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.98 kb
#include <fstream>
#include <algorithm>
#include <deque>
using namespace std;
ifstream f("ghiozdan.in");
ofstream g("ghiozdan.out");
int N,W;
int fr[205];
int dp[2][75005];
int ant[2][75005];
int D[75005],stq,drq;
void computedp(int i,int weight,int mid,int W){
    int l = (i & 1);
    int nonl = 1 - l;
    for(int j = 0;j <= W;j++){
            dp[l][j] = 1 << 28;
    }
    if(!fr[weight]){
        for(int j = 0;j <= W;j++){
            dp[l][j] = dp[nonl][j];
            ant[l][j] = (i <= mid ? j : ant[nonl][j]);
        }
        return ;
    }
    for(int r = 0;r < weight;r++){
        stq = 1;
        drq = 0;
        for(int j = r;j <= W;j += weight){
            while(stq <= drq && j - D[stq] > fr[weight] * weight){
                stq++;
            }
            while(stq <= drq && dp[nonl][D[drq]] - (D[drq] / weight) >= dp[nonl][j] - (j / weight)){
                drq--;
            }
            D[++drq] = j;
            dp[l][j] = dp[nonl][D[stq]] - (D[stq] / weight) + (j / weight);
            ant[l][j] = (i <= mid ? j : ant[nonl][D[stq]]);
        }
    }
}
void solve(int st,int dr,int W){
    if(st == dr || !W){
        for(int i = st;i <= W;i += st){
            g << st << "\n";
        }
        return ;
    }
    int mid = (st + dr) / 2;
    for(int j = 0;j <= W;j++){
            dp[(st & 1) ^ 1][j] = 1 << 28;
    }
    dp[(st & 1) ^ 1][0] = 0;
    for(int i = st;i <= dr;i++){
        computedp(i,i,mid,W);
    }
    int tmp = ant[dr & 1][W];
    solve(st,mid,tmp);
    solve(mid + 1,dr,W - tmp);
}
int main()
{
    f >> N >> W;
    for(int i = 1;i <= N;i++){
        int x;
        f >> x;
        fr[x]++;
    }
    for(int j = 1;j <= W;j++){
            dp[0][j] = 1 << 28;
    }
    for(int i = 1;i <= 200;i++){
        computedp(i,i,0,W);
    }
    while(dp[0][W] == 1 << 28){
        W--;
    }
    g << W << " " << dp[0][W] << "\n";
    solve(1,200,W);
    f.close();
    g.close();
    return 0;
}