A13404. Résumé Review

Uh oh! Applications to tech companies are due soon, and you've been procrastinating by doing contests instead! (Let's pretend for now that it is actually possible to get a job in these uncertain times.)

You have completed many programming projects. In fact, there are exactly $n$ types of programming projects, and you have completed $a_i$ projects of type $i$ . Your résumé has limited space, but you want to carefully choose them in such a way that maximizes your chances of getting hired.

You want to include several projects of the same type to emphasize your expertise, but you also don't want to include so many that the low-quality projects start slipping in. Specifically, you determine the following quantity to be a good indicator of your chances of getting hired:

$$$$ f(b_1,\ldots,b_n)=\sum\limits_{i=1}^n b_i(a_i-b_i^2). $$ </p> <p>Here, $b\_i$ denotes the number of projects of type $i$ you include in your résumé. Of course, you cannot include more projects than you have completed, so you require $0\\le b\_i \\le a\_i$ for all $i$ .</p> <p>Your résumé only has enough room for $k$ projects, and you will absolutely not be hired if your résumé has empty space, so you require $\\sum\\limits\_{i=1}^n b\_i=k$ .</p> <p>Find values for $b\_1,\\ldots, b\_n$ that maximize the value of $f(b\_1,\\ldots,b\_n)$$$ while satisfying the above two constraints.

The first line contains two integers $n$ and $k$ ( $1\le n\le 10^5$ , $1\le k\le \sum\limits_{i=1}^n a_i$ ) — the number of types of programming projects and the résumé size, respectively.

The next line contains $n$ integers $a_1,\ldots,a_n$ ( $1\le a_i\le 10^9$ ) — $a_i$ is equal to the number of completed projects of type $i$ .

In a single line, output $n$ integers $b_1,\ldots, b_n$ that achieve the maximum value of $f(b_1,\ldots,b_n)$ , while satisfying the requirements $0\le b_i\le a_i$ and $\sum\limits_{i=1}^n b_i=k$ . If there are multiple solutions, output any.

Note that you do not have to output the value $f(b_1,\ldots,b_n)$ .

For the first test, the optimal answer is $f=-269$ . Note that a larger $f$ value is possible if we ignored the constraint $\sum\limits_{i=1}^n b_i=k$ .

For the second test, the optimal answer is $f=9$ .