CSES - Datatähti 2025 alku - Results
Submission details
Task:Tikut
Sender:adex720
Submission time:2024-11-01 20:19:21 +0200
Language:C++ (C++20)
Status:COMPILE ERROR

Compiler report

input/code.cpp: In function 'int laske(std::priority_queue<Tikku, std::vector<Tikku>, Compare>, int, int, int)':
input/code.cpp:147:13: warning: unused variable 'uusiSuurempi' [-Wunused-variable]
  147 |         int uusiSuurempi = (pisin.pituus + pisin.jakoja) / (pisin.jakoja + 1);
      |             ^~~~~~~~~~~~
input/code.cpp: In function 'int main()':
input/code.cpp:300:5: error: '_sleep' was not declared in this scope; did you mean 'usleep'?
  300 |     _sleep(500);
      |     ^~~~~~
      |     usleep

Code

#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;
    }
};

vector<int> luoJono(priority_queue<Tikku, vector<Tikku>, Compare> q, int maksimiMaara, int pieninMahdollinen, int uusiPienempi, int pienin, int pisin)
{

    vector<int> tulos;
    vector<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) // maksimi toistot = maksimiMaara
        {
            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_back(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 vector<int>();
    }

    sort(tulos.begin(), tulos.end());

    for (int i = min(maksimiMaara, (int)(tulos.size())) - 1; i >= 0; i--)
    {
        jono.push_back(tulos[i]);
    }

    return jono;
}

int laske(priority_queue<Tikku, vector<Tikku>, Compare> q, int m, int maksimiLisays, int pienin)
{
    int m2 = m;

    deque<vector<int>> vaihtoehtoiset;
    vector<int> indeksit;

    while (m > 0)
    {
        auto pisin = q.top();

        q.pop();

        int uusiSuurempi = (pisin.pituus + pisin.jakoja) / (pisin.jakoja + 1);
        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;
            vector<int> jono = luoJono(priority_queue<Tikku, vector<Tikku>, Compare>(q), maksimiMaara, (pisin.suurinPituus + uusiPienempi - 2) >> 1, uusiPienempi, pienin, pisin.suurinPituus);

            if (!jono.empty())
            {
                vaihtoehtoiset.push_back(jono);
                indeksit.push_back(jono.size() - 1);
            }
        }

        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;
                int indexIndex = 0;
                for (auto it = vaihtoehtoiset.begin(); it != vaihtoehtoiset.end();)
                {
                    int index = indeksit[indexIndex];
                    int v = (*it)[index];
                    if (v < paras)
                    {
                        paras = v;
                    }
                    else if (v > tulos)
                    {
                        it = vaihtoehtoiset.erase(it);
                        indeksit.erase(indeksit.begin() + indexIndex);
                        continue;
                    }

                    if (index == 0)
                    {
                        it = vaihtoehtoiset.erase(it);
                        indeksit.erase(indeksit.begin() + indexIndex);
                    }
                    else
                    {
                        it++;
                        indeksit[indexIndex]--;
                        indexIndex++;
                    }
                }
                cout << paras << " ";
            }
            break;
        }

        m -= pisin.maara;

        for (int i = 0; i < pisin.maara - 1; i++)
        {
            int tulos = (pituusEnnen - pienin);
            int paras = tulos;

            int indexIndex = 0;
            for (auto it = vaihtoehtoiset.begin(); it != vaihtoehtoiset.end();)
            {
                int index = indeksit[indexIndex];
                int v = (*it)[index];
                if (v < paras)
                {
                    paras = v;
                }
                else if (v > tulos)
                {

                    it = vaihtoehtoiset.erase(it);
                    indeksit.erase(indeksit.begin() + indexIndex);
                    continue;
                }

                if (index == 0)
                {
                    it = vaihtoehtoiset.erase(it);
                    indeksit.erase(indeksit.begin() + indexIndex);
                }
                else
                {
                    it++;
                    indeksit[indexIndex]--;
                    indexIndex++;
                }
            }

            cout << paras << " ";
        }

        int tulos = (q.top().suurinPituus) - pienin;
        int paras = tulos;

        int indexIndex = 0;
        for (auto it = vaihtoehtoiset.begin(); it != vaihtoehtoiset.end();)
        {
            int index = indeksit[indexIndex];
            int v = (*it)[index];
            if (v < paras)
            {
                paras = v;
            }
            else if (v > tulos)
            {

                it = vaihtoehtoiset.erase(it);
                indeksit.erase(indeksit.begin() + indexIndex);
                continue;
            }

            if (index == 0)
            {
                it = vaihtoehtoiset.erase(it);
                indeksit.erase(indeksit.begin() + indexIndex);
            }
            else
            {
                it++;
                indeksit[indexIndex]--;
                indexIndex++;
            }
        }

        cout << paras << " ";
        continue;
    }

    return m2;
}

int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);

    int n, m;
    cin >> n >> m;

    _sleep(500);

    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));

    laske(q, m, suurin, pienin);
}