Tikut

input/code.cpp: In function 'int main()':
input/code.cpp:18:22: warning: 'pituus' may be used uninitialized in this function [-Wmaybe-uninitialized]
   18 |         suurinPituus = laskeSuurinPituus();
      |         ~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~~
input/code.cpp:296:9: note: 'pituus' was declared here
  296 |     int pituus;
      |         ^~~~~~

#include <bits/stdc++.h>
#include <stdlib.h>
using namespace std;
 
class Tikku
{
public:
    int pituus;
    int jakoja;
    int maara;
    int suurinPituus;
 
    Tikku(int a, int b)
    {
        pituus = a;
        jakoja = 1;
        maara = b;
        suurinPituus = laskeSuurinPituus();
    }
 
    int laskeSuurinPituus()
    {
        return (pituus + jakoja - 1) / (jakoja);
    }
 
    int seuraavaSuurinPituus()
    {
        return (pituus + jakoja) / (jakoja + 1);
    }
 
    int laskePieninPituus()
    {
        return (pituus) / (jakoja);
    }
 
    int seuraavaPieninPituus()
    {
        return (pituus) / (jakoja + 1);
    }
 
    void jaa()
    {
        jakoja++;
        suurinPituus = laskeSuurinPituus();
    }
};
 
class Compare
{
public:
    bool operator()(Tikku a, Tikku b)
    {
        int v = a.suurinPituus - b.suurinPituus;
        if (v != 0)
            return v < 0;
 
        v = a.pituus - b.pituus;
        return v < 0;
    }
};
 
queue<int> luoJono(priority_queue<Tikku, vector<Tikku>, Compare> q, int maksimiMaara, int pieninMahdollinen, int uusiPienempi, int pienin, int pisin)
{
 
    vector<int> tulos;
    queue<int> jono;
 
    int nollia = 0;
    while (!q.empty())
    {
        Tikku t = q.top();
        q.pop();
 
        int alkup = t.pituus;
        int maar = t.maara;
        int jakom = t.jakoja;
 
        int lisaa = 0;
 
        if (t.suurinPituus < pieninMahdollinen)
            break;
 
        while (true) // jakoja enintään lopuksi = (pituus ennen jakoja - pienin sallittu pituus) / (pienin sallittu pituus - 1)
        {
            lisaa++;
            int uusiPienin = alkup / (jakom + lisaa);
            if (uusiPienin <= uusiPienempi)
                break;
 
            if (uusiPienin >= pienin)
            {
                nollia += maar;
 
                if (nollia >= maksimiMaara)
                {
                    for (int i = 0; i < maksimiMaara; i++)
                    {
                        jono.push(pisin - pienin);
                    }
 
                    return jono;
                }
 
                for (int i = 0; i < maar; i++)
                {
                    tulos.push_back(pisin - pienin);
                }
 
                continue;
            }
 
            for (int i = 0; i < maar; i++)
            {
                tulos.push_back(pisin - uusiPienin);
            }
        }
    }
 
    if (tulos.empty())
    {
        return queue<int>();
    }
 
    sort(tulos.begin(), tulos.end());
 
    int loppu = min(maksimiMaara, (int)(tulos.size()));
    for (int i = 0; i < loppu; i++)
    {
        jono.push(tulos[i]);
    }
 
    return jono;
}
 
int laske(priority_queue<Tikku, vector<Tikku>, Compare> q, int m, int maksimiLisays, int pienin)
{
    int m2 = m;
 
    queue<int> vaihtoehtoiset;
 
    while (m > 0)
    {
        auto pisin = q.top();
 
        q.pop();
 
        int uusiPienempi = pisin.pituus / (pisin.jakoja + 1);
        int pituusEnnen = pisin.suurinPituus;
        int toiseksiSuurin = q.top().suurinPituus;
 
        if (!(uusiPienempi >= pienin || (pisin.maara == 1 && pienin - uusiPienempi <= pituusEnnen - toiseksiSuurin)))
        {
            int ero = (pienin - uusiPienempi) - (pituusEnnen - toiseksiSuurin);
            int maksimiMaara = ero <= 0 ? min(m, pisin.maara - 1) : m;
            queue<int> jono = luoJono(q, maksimiMaara, (pisin.suurinPituus + uusiPienempi) >> 1, uusiPienempi, pienin, pisin.suurinPituus);
 
            if (!jono.empty())
            {
                queue<int> uusi;
 
                while (!jono.empty() && !vaihtoehtoiset.empty())
                {
                    uusi.push(min(jono.front(), vaihtoehtoiset.front()));
                    jono.pop();
                    vaihtoehtoiset.pop();
                }
                while (!jono.empty())
                {
                    uusi.push(jono.front());
                    jono.pop();
                }
                while (!vaihtoehtoiset.empty())
                {
                    uusi.push(vaihtoehtoiset.front());
                    vaihtoehtoiset.pop();
                }
                vaihtoehtoiset = uusi;
            }
        }
 
        pisin.jaa();
        q.push(pisin);
 
        if (uusiPienempi < pienin)
        {
            pienin = uusiPienempi;
        }
 
        if (m < pisin.maara)
        {
            for (; m > 0; m--)
            {
                int tulos = pituusEnnen - pienin;
                int paras = tulos;
 
                if (!vaihtoehtoiset.empty())
                {
                    int v = vaihtoehtoiset.front();
                    vaihtoehtoiset.pop();
                    if (v < paras)
                    {
                        paras = v;
                    }
                    else if (v > tulos)
                    {
                        vaihtoehtoiset = queue<int>();
                    }
                }
 
                cout << paras << " ";
            }
            break;
        }
 
        m -= pisin.maara;
 
        for (int i = 0; i < pisin.maara - 1; i++)
        {
            int tulos = (pituusEnnen - pienin);
            int paras = tulos;
 
            if (!vaihtoehtoiset.empty())
            {
                int v = vaihtoehtoiset.front();
                vaihtoehtoiset.pop();
                if (v < paras)
                {
                    paras = v;
                }
                else if (v > tulos)
                {
                    vaihtoehtoiset = queue<int>();
                }
            }
 
            cout << paras << " ";
        }
 
        int tulos = (q.top().suurinPituus) - pienin;
        int paras = tulos;
 
        if (!vaihtoehtoiset.empty())
        {
            int v = vaihtoehtoiset.front();
            vaihtoehtoiset.pop();
            if (v < paras)
            {
                paras = v;
            }
            else if (v > tulos)
            {
                vaihtoehtoiset = queue<int>();
            }
        }
 
        cout << paras << " ";
        continue;
    }
 
    return m2;
}
 
int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
 
    int n, m;
    cin >> n >> m;
 
    int pituudet[n];
    for (int i = 0; i < n; i++)
    {
        cin >> pituudet[i];
    }
 
    if (n == 1)
    {
        for (int i = 2; i <= m + 1; i++)
        {
            cout << (pituudet[0] % i == 0 ? 0 : 1) << " ";
        }
 
        return 0;
    }
 
    sort(pituudet, pituudet + n);
 
    int pienin = pituudet[0];
    int suurin = pituudet[n - 1];
 
    // käsittele ereja=1 tai 2
 
    priority_queue<Tikku, vector<Tikku>, Compare> q;
 
    int pituus;
    int maara = 1;
    int edellinen = pituudet[0];
 
    for (int i = 1; i < n; i++)
    {
        pituus = pituudet[i];
        if (pituus == edellinen)
        {
            maara++;
            continue;
        }
 
        q.push(Tikku(edellinen, maara));
        edellinen = pituus;
        maara = 1;
    }
 
    q.push(Tikku(pituus, maara));
 
    m = min(m, 900);
    laske(q, m, suurin, pienin);
}

Test 1

Group: 1, 3, 4, 5, 6

Verdict: ACCEPTED

Test 2

Group: 1, 4, 5, 6

Verdict: ACCEPTED

Test 3

Group: 1, 4, 5, 6

Verdict: ACCEPTED

Test 4

Group: 1, 4, 5, 6

Verdict: ACCEPTED

Test 5

Group: 2, 5, 6

Verdict: WRONG ANSWER

Test 6

Group: 2, 5, 6

Verdict: WRONG ANSWER

Test 7

Group: 3, 5, 6

Verdict: ACCEPTED

Test 8

Group: 3, 5, 6

Verdict: ACCEPTED

Test 9

Group: 3, 5, 6

Verdict: ACCEPTED

Test 10

Group: 3, 5, 6

Verdict: ACCEPTED

Test 11

Group: 3, 5, 6

Verdict: ACCEPTED

Test 12

Group: 4, 5, 6

Verdict: ACCEPTED

Test 13

Group: 4, 5, 6

Verdict: ACCEPTED

Test 14

Group: 4, 5, 6

Verdict: ACCEPTED

Test 15

Group: 4, 5, 6

Verdict: ACCEPTED

Test 16

Group: 5, 6

Verdict: WRONG ANSWER

Test 17

Group: 5, 6

Verdict: WRONG ANSWER

Test 18

Group: 5, 6

Verdict: WRONG ANSWER

Test 19

Group: 5, 6

Verdict: WRONG ANSWER

Test 20

Group: 6

Verdict: TIME LIMIT EXCEEDED

Test 21

Group: 6

Verdict: WRONG ANSWER

Test 22

Group: 6

Verdict: WRONG ANSWER