std::sort 向量的向量

Question

我实际上是在尝试根据描述符中包含的向量对描述符向量进行排序

float           xi, yi;     // Descriptor location
vector<double>  fv;         // The feature vector

（这里是名为 fv 的向量）我想要做的是在 Descriptor 的向量中搜索（为了更清楚起见，我将其命名为 Vec_Desc）方差最大的维度为此，我使用 :

double Tree::_get_mean(vector<Descriptor>& vec, int indice)
{
    double sum = 0;
    int size = vec.size();
    for (int i = 0; i < size; i++) {
        sum += vec[i].fv[indice];
    }
    sum /= size;
    return sum;
}

double Tree::_get_variance(vector<Descriptor>& vec, int indice)
{
int size = vec.size();
double var = 0;
double mean = _get_mean(vec, indice);
    for (int i = 0; i < size; i++) {
        var += ((vec[i].fv[indice] - mean) *  (vec[i].fv[indice] - mean)) / size;
    }
    return var;
}

int Tree::choose_dimension(vector<Descriptor>& vec)
{
    double var = _get_variance(vec, 0);
    int indice = 0;
    for (int i = 1; i < vec[0].fv.size(); i++) {
         if (var > _get_variance(vec, i)) {
            var = _get_variance(vec, i);
            indice = i;
        }
    }
}

然后我想根据我发现的 dim 对 vec_desc 进行排序。试图这样做：

class _compare {
    int step;
public: 
     _compare(int s) : step(s) {}

    bool operator()(const vector<double>& p1, const vector<double>& p2) {
        return p1[step] < p2[step];
    }
};

void Tree::_sort_vector(vector<Descriptor>& vec, int i)
{
    std::sort(vec.begin(), vec.end(), _compare(i));
}

void Tree::sort_vector(vector<Descriptor>& vec)
{
    _sort_vector(vec, choose_dimension(vec));
}

但是有了这个，我将使用他自己的值而不是 fv 中包含的值对 vec_desc 进行排序... 我该怎么做？

Answer 1

您的比较器不正确 - 因为您正在排序 vector<Descriptor> 它必须比较 Descriptor 而不是 vector<double>:

的两个实例

class _compare {
    int step;
public: 
     _compare(int s) : step(s) {}

    bool operator()(const Descriptor& d1, const Descriptor& d2) const {
        return d1.fv[step] < d2.fv[step];
    }
};

std::sort 向量的向量

std::sort on vector of vector

c++

sorting

vector

std