c ++代码缓慢且超出时间限制

时间:2016-12-15 09:48:21

标签: c++ algorithm sorting

段的起点和终点在向量开始结束中给出。矢量点给出了一个点列表。任务是找到包含每个点的段数。

在我的解决方案中, start end point 的每个元素分别被赋予标签l,p和r并存储在vector 对。然后, Pairs 向量首先按第一个元素排序,然后按第二个元素排序。最后,我迭代对向量并增加一个变量 coverage 如果它是一个起始点,如果它是一个终点则递减它,如果它是一个点,则将coverage的值分配给答案。登记/>
该算法似乎是正确的,时间复杂度为O(nlog(n))但超过了时间限制。代码的哪一部分很慢?
代码:

vector<int> fast_count_segments(vector<int>& starts, vector<int>& ends, vector<int>& points) {
    vector<int> cnt(points.size());
    const int left_label = 1, point_label = 2, right_label = 3;
    std::map<int, std::set<int>> orig_point_map;
    vector<pair<int,int>> pairs(2*starts.size()+points.size());
    int k = 0;
    for (auto& i : starts)
    {
        pairs[k++] = std::make_pair(i, left_label);
    }
    for (auto& i : ends)
    {
        pairs[k++] = std::make_pair(i, right_label);
    }
    for (auto i = 0;i < points.size();i++)
    {
        int point = points[i];
        pairs[k++] = std::make_pair(point, point_label);
        orig_point_map[point].emplace(i);
    }

    std::sort(pairs.begin(), pairs.end());
    int coverage = 0;
    for (auto& x : pairs) {
        if (x.second == 1) {
            coverage++;
        }
        else if (x.second == 3) {
            coverage--;
        }
        else {
            std::set<int> indices = orig_point_map[x.first];
            for(auto i : indices) {
                cnt[i] = coverage;
            }
        }
    }
    return cnt;
}

1 个答案:

答案 0 :(得分:0)

虽然复杂性很好,但您可以创建许多可以避免的副本。 您可以使用修改后的版本:

std::vector<int> fast_count_segments(std::vector<int> starts,
                                     std::vector<int> ends,
                                     const std::vector<int>& points) {
    std::vector<int> cnt(points.size());
    std::vector<std::pair<int, int>> pairs(points.size());

    for (auto i = 0u; i != points.size(); ++i) {
        pairs[i] = std::make_pair(points[i], i);
    }
    std::sort(starts.begin(), starts.end());
    std::sort(ends.begin(), ends.end());
    std::sort(pairs.begin(), pairs.end());
    int coverage = 0;
    auto it_start = starts.begin();
    auto it_end = ends.begin();
    for (auto& x : pairs) {

        while (it_start != starts.end() && *it_start <= x.first) {
            ++it_start;
            ++coverage;
        }
        while (it_end != ends.end() && *it_end < x.first) {
            ++it_end;
            --coverage;
        }
        cnt[x.second] = coverage;     
    }
    return cnt;
}