Return k points with the smallest squared Euclidean distance to the origin.
A good solution should
- State the invariant or decision rule that makes the approach correct.
- Explain the time and space complexity in terms of the input sizes and required output.
- Handle empty, minimal, duplicate, and boundary-shaped inputs covered by the contract.
Reasoning prompt
Squared distance avoids square roots. Compare sorting all points with maintaining a bounded heap; output order may be arbitrary.
Use the linked judge for the canonical problem. The examples above are compact TRT checks; write additional cases before submitting, especially for the boundary that tends to break your chosen invariant.