算法在推力/ cudpp中找到第k個最小元素

Question

我正在尋找推力/ cudapp中第k個最小元素算法的實現。我谷歌搜索，但似乎沒有找到它。有沒有人知道是否存在這樣的算法？

我看到有重新排序，但沒有說第k個最小。

Answer 1

目前推力並未提供選擇算法（即STL中的std::nth_element），儘管它在我們的雷達上，並且有good evidence可以在GPU上快速完成選擇。您現在唯一的辦法是使用thrust::sort或thrust::sort_by_key（或其stable_變體）對數據進行排序，然後選擇適當的元素。排序基本類型（例如int，float，char，double）的推力與非常快的基數排序的代碼實現的，所以絕對錶現還是會比較好，雖然效率不高的specialized selection method。