*[codility]Number-of-disc-intersections

时间:2023-03-08 21:46:43

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;
}