返回指针而不是vector :: iterator并将vector :: iterator转换为指针

Question

我有一个函数，其返回类型为“ vector :: iterator”，如下所示：

vector<unsigned int>::iterator foo(unsigned int arg, vector<unsigned int> arr, vector<unsigned int> M)
{    if (arg == 0)
        return arr.begin();
    else if (arg == 2)
        return arr.begin() + arr[arg - 1];
    else
        return arr.begin();
}

问题是，如何将其返回值转换为指针而不是std :: vector迭代器？

我应该将其转换为在CUDA中使用，因为我们不能在CUDA中使用vector或其迭代器

这是我的方式：

__host__ __device__ unsigned int* foo(unsigned int arg, unsigned int* arr, unsigned int* M)
{  
    unsigned int* res, tmp;
    if (arg == 0)
        tmp = arr[0];
    else if (arg == 2)
        tmp = arr[0] + arr[arg - 1];
    else
        tmp = arr[0];
        res = &tmp;
    return res; 
}

用于内核工作，例如：

__global__ kernel void kernel . . .
.
.
unsigned int* d_h = foo( , ,)
.
.

但这是错误的并且不起作用，我该怎么办？ 谢谢，

Answer 1

您可以执行&(*it) ，它将提供指向迭代器引用的元素的指针。 如果您不知道operator&是否已被恶意用户重载，请在通用代码中使用std::addressof(*it) std::vector<T> 。

Answer 2

如果进行转换意味着不使用向量而是使用数组来重写代码，则建议对代码进行以下调整：

__host__ __device__ unsigned int* foo(unsigned int arg, unsigned int* arr, unsigned int* M)
{  
    unsigned int *res, *tmp;  // attention * is per item !! 
    if (arg == 0)
        tmp = arr;  // return a pointer to the begining
    else if (arg == 2)
        tmp = arr + arr[arg - 1];  // you need to make pointer arithmetic like in the original code iterator arithmetics
    else
        tmp = arr;  // keep it simple 
    return tmp;    // note that you don't need res. 
}

您代码中的主要问题是声明unsigned int *a,b; 创建一个指针a到一个unsigned int和unsigned int变量b ，但不是两个指针。

第二个问题是行res = &tmp; （ 顺便说一下，它的缩进具有误导性。幸运的是， gcc 6将来会警告您 ）。 这导致您返回一个临时变量的地址。 但是，从函数返回后，临时变量将被销毁，因此返回的指针将指向无处（透视图为UB）。

Answer 3

原则上，您可以使用&*i获取指向迭代器元素的指针，从而获取指向底层矢量数据的指针，但这在这里无济于事。

您的第一个函数已损坏，因为它将迭代器返回到向量中，该向量的寿命在函数返回时结束。

要解决此问题，您可以通过参考获取向量，

vector<unsigned int>::iterator foo(unsigned int arg, vector<unsigned int>& arr)
{
    return arg == 2 ? arr.begin() + arr[1] : arr.begin();
}

但是索引比迭代更健壮，并且在const-ness下表现良好：

int foo(unsigned int arg, const vector<unsigned int>& arr)
{
    return arg == 2 ? arr[1] : 0;
}

vector<unsigned int>::iterator it = myvector.begin() + foo(arg, myvector);
vector<unsigned int>::const_iterator cit = myvector.cbegin() + foo(arg, myvector);

（迭代器用于迭代 -迭代器是瞬时且短暂的。
您不应该将它们用作一种广义的指针。）

当CUDA代码应将指针返回到arr时，它会返回同样错误的指向自动局部变量的指针：

__host__ __device__ unsigned int* foo(unsigned int arg, unsigned int* arr, unsigned int* M)
{
    return arg == 2 ? arr + arr[1] : arr;  
}