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