Don't waste your time on this problem. It is garbage. The editorial suggest that generate independent samples from the estimated distribution. Obviously it can only give something close to the expected solution by chance. Some of the top solutions simple return a list of values depending on first input line (there is only two test cases). Some other solution uses the interpolation function of the scipy module. They even produces negative values. So it is not clear how can they achived max points.