*[codility]Number-of-disc-intersections

http://codility.com/demo/take-sample-test/beta2010/

这题以前做的时候是先排序再二分,现在觉得没有必要。首先圆可以看成线段,把线段的进入作为一个事件,出去作为一个事件。注意根据题意,同样的点,进入要在出去之前。那么O(n)扫一遍就可以得到结果。注意的是A[i]+i可能会超出int。

// you can also use includes, for example:
#include <algorithm>
int solution(const vector<int> &A) {
    // write your code in C++98
    vector<pair<long long, int> > intervals;
    for (int i = 0; i < A.size(); i++) {
        intervals.push_back(make_pair(((long long)i)-A[i], -1));
        intervals.push_back(make_pair(((long long)i)+A[i], 1));
    }
    sort(intervals.begin(), intervals.end());
    int ans = 0;
    int height = 0;
    for (int i = 0; i < intervals.size(); i++) {
        pair<long long, int> pp = intervals[i];
        if (pp.second == -1) { // start point
            ans += height;
            height++;
            if (ans > 10000000) return -1;
        }
        else { // end point
            height--;
        }
    }
    return ans;
}

  

posted @ 2013-11-14 19:31  阿牧遥  阅读(365)  评论(0编辑  收藏  举报