[itint5]区间相交

http://www.itint5.com/oj/#14

要记录原来的索引,所以用了额外的空间,新生成一个结构。如果要省空间,可以用指针来排序,最后拿指针减去索引0的位置就是index,见:http://www.itint5.com/discuss/172/%E4%B8%BA%E5%95%A5%E6%88%91%E8%BF%99%E9%A2%98%E5%9C%A8%E5%A4%A7%E6%95%B0%E6%8D%AE%E6%97%B6%E4%BC%9Asegmentation-fault

这里没有用指针,也有很多错误,足可见面试时不要自取其辱,寻找合理的方法即可。

一要注意,sort和comp函数,因为是高级的,所以返回bool,并且参数是const引用。二要注意,排完序后,并不是前一个和后一个比较就行了,要记录之前的end最大的那个。

struct IntervalPair {
    int start;
    int end;
    int idx;
};

bool comp(const IntervalPair &pa, const IntervalPair &pb) {
    if (pa.start != pb.start) {
        return pa.start < pb.start;
    } else {
        return pb.end < pb.end;
    }
}

void intersected(vector<Interval> &intervals, vector<bool> &isIntersected) {
    int size = intervals.size();
    if (size <= 1) return;
    vector<IntervalPair> vec(size);
    for (int i = 0; i < size; i++) {
        vec[i].start = intervals[i].start;
        vec[i].end = intervals[i].end;
        vec[i].idx = i;
    }
    sort(vec.begin(), vec.end(), comp);
    int max_end = vec[0].end;
    int max_end_idx = 0;
    for (int i = 0; i < size - 1; i++) {
        if (vec[i].end > max_end) {
            max_end = vec[i].end;
            max_end_idx = i;
        }
        if (max_end >= vec[i+1].start) {
            isIntersected[vec[max_end_idx].idx] = true;
            isIntersected[vec[i+1].idx] = true;
        }
    }
}

  

posted @ 2014-01-16 17:44  阿牧遥  阅读(262)  评论(0编辑  收藏  举报