rt，算了一下复杂度是 $\mathcal{O}(N\sqrt{D\times S})$ 的，应该不会 TLE，但是唯一的那篇 KD Tree 题解 TLE 了