Tikut

input/code.cpp: In function 'std::vector<int> luoJono(std::priority_queue<Tikku, std::vector<Tikku>, Compare>, int, int, int, int, int, int)':
input/code.cpp:68:10: warning: unused variable 'valmis' [-Wunused-variable]
   68 |     bool valmis = false;
      |          ^~~~~~
input/code.cpp: In function 'int main()':
input/code.cpp:296:15: warning: unused variable 'a' [-Wunused-variable]
  296 |     int n, m, a;
      |               ^
input/code.cpp:319:9: warning: unused variable 'ereja' [-Wunused-variable]
  319 |     int ereja = setti.size();
      |         ^~~~~
input/code.cpp:17:22: warning: 'pituus' may be used uninitialized in this function [-Wmaybe-uninitialized]
   17 |         suurinPituus = laskeSuurinPituus();
      |         ~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~~
input/code.cpp:327:9: note: 'pituus' was declared here
  327 |     int pituus;
      |         ^~~~~~

#include <bits/stdc++.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 maksimiLisays, int maksimiMaara, int pieninMahdollinen, int uusiPienempi, int pienin, int pisin)
{
 
    vector<int> tulos;
    vector<int> jono;
 
    int nollia = 0;
    bool valmis = false;
    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.pituus;
        int toiseksiSuurin = q.top().suurinPituus;
 
        if (!(uusiPienempi >= pienin || (pisin.maara == 1 && pienin - uusiPienempi <= pisin.pituus - toiseksiSuurin)))
        {
            int ero = (pienin - uusiPienempi) - (pisin.pituus - toiseksiSuurin);
            int maksimiMaara = ero <= 0 ? min(m, pisin.maara - 1) : m;
            vector<int> jono = luoJono(priority_queue<Tikku, vector<Tikku>, Compare>(q), max(uusiSuurempi, toiseksiSuurin) - uusiPienempi, 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);
            }
 
            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, a;
    cin >> n >> m;
 
    int pituudet[n];
    set<int> setti;
    for (int i = 0; i < n; i++)
    {
        cin >> pituudet[i];
        setti.insert(pituudet[i]);
    }
 
    if (n == 1)
    {
        for (int i = 2; i <= m + 1; i++)
        {
            cout << (n % i == 0 ? 0 : 1) << " ";
        }
 
        return 0;
    }
 
    sort(pituudet, pituudet + n);
 
    int ereja = setti.size();
    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);
}

Test 1

Group: 1, 3, 4, 5, 6

Verdict: WRONG ANSWER

Test 2

Group: 1, 4, 5, 6

Verdict: WRONG ANSWER

Test 3

Group: 1, 4, 5, 6

Verdict: WRONG ANSWER

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: WRONG ANSWER

Test 13

Group: 4, 5, 6

Verdict: RUNTIME ERROR

Test 14

Group: 4, 5, 6

Verdict: RUNTIME ERROR

Test 15

Group: 4, 5, 6

Verdict: WRONG ANSWER

Test 16

Group: 5, 6

Verdict: ACCEPTED

Test 17

Group: 5, 6

Verdict: RUNTIME ERROR

Test 18

Group: 5, 6

Verdict: RUNTIME ERROR

Test 19

Group: 5, 6

Verdict: RUNTIME ERROR

Test 20

Group: 6

Verdict: TIME LIMIT EXCEEDED

Test 21

Group: 6

Verdict: TIME LIMIT EXCEEDED

Test 22

Group: 6

Verdict: WRONG ANSWER