*[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;

}

  

你可能感兴趣的:(intersect)