问CUDA推力:如何实现支持“模板”的“划分”？
EN

Stack Overflow用户

提问于 2012-08-21 11:07:27

回答 1查看 792关注 0票数 3

假设有一个整型数组：

A[]={2, 2, 9, 8, 5, 7, 0, 6}

和一张模板：

B[]={1, 0, 0, 1, 1, 1, 0, 1}

我的问题是，我们如何根据B[]重新排列A[]，以便如果Bi==1，Bj==0，那么Ai将被保证在新数组中位于Aj之前，它应该如下所示：

C[]={2, 8, 5, 7, 6, 2, 9, 0}

PS:我发现"partition"函数几乎就是答案，除了它只支持谓词。有什么变通方法吗？

任何提示都非常感谢！

回答已采纳

发布于 2012-08-31 11:52:11

这可以使用thrust::stable_sort_by_key()来实现。

票数 1

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/12047961

复制

相似问题

问CUDA推力:如何实现支持“模板”的“划分”？EN