Cod sursa(job #2527667)

Utilizator NicolaalexandraNicola Alexandra Mihaela Nicolaalexandra Data 20 ianuarie 2020 19:22:00
Problema NumMst Scor 45
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.78 kb
#include <bits/stdc++.h>
#define DIM 5000
#define DIMM 10000010
using namespace std;

ifstream fin ("nummst.in");
ofstream fout ("nummst.out");
int ciur[DIMM];
int sol[DIM][700];
double dp[DIM][700];
vector <int> ans,p;
int n,i,j,k,d,nr;
int main (){

    fin>>n;

    for (i=2;i<=n/i;i++)
        if (n%i == 0)
            break;

    d = n/i;
    n = i;

    p.push_back(0);
    for (i=2;i<n;i++){
        if (!ciur[i]){
            for (j=i+i;j<=n;j+=i)
                ciur[j] = 1;
            p.push_back(i);
        }
    }
    k = p.size()-1;
    /// dp[i][j] - cmmmc ul maxim daca il scriu pe i ca suma de puteri de nr prime
    /// cu factori primi din primii j

    //dp[0][0] = log(1);

    for (j=1;j<=k;j++){

        /// nu iau nimic de la factorul asta
        for (i=1;i<=n;i++)
            dp[i][j] = dp[i][j-1];

        int nr = p[j];
        while (nr <= n){
            double val = log(nr);
            for (i=nr;i<=n;i++){

                if (dp[i-nr][j-1] + val > dp[i][j]){
                    dp[i][j] = dp[i-nr][j-1] + val;
                    sol[i][j] = nr; /// tin minte din ce obtin starea i,j
                }
            }

            nr *= p[j];
        }
    }

    /*double maxi = 0;
    int poz = 0;
    for (i=1;i<=k;i++){
        if (dp[n][i] > maxi){
            maxi = dp[n][i];
            poz = i;
        }
    }*/

    int x = n, poz = k;
    while (poz > 0){
        if (sol[x][poz]){
            ans.push_back(sol[x][poz]*d);
            x -= sol[x][poz];
        }
        poz--;
    }
    while (x > 0){ /// trb sa ma asigur ca suma da n
        ans.push_back(d);
        x--;
    }

    sort (ans.begin(),ans.end());
    for (auto it:ans)
        fout<<it<<" ";

    return 0;
}