[codility]Number-of-disc-intersections

来源:互联网 发布:tensorflow 图片分类 编辑:程序博客网 时间:2024/05/22 11:59
// you can also use includes, for example:#include <algorithm>struct IntervalSeg{    long long left, right;    IntervalSeg(long long _left = 0, long long _right = 0):left(_left),right(_right){}    bool operator<(const IntervalSeg& orh) const    {        if(left == orh.left) return right < orh.right;        else return left < orh.left;    }};int getFirstBiggerIdx(const vector<IntervalSeg>& intervalSeg, long long target){    int left = 0;    int right = intervalSeg.size()-1;    while(left <= right)    {        int mid = left+(right-left)/2;        if(intervalSeg[mid].left > target) right = mid-1;        else left = mid+1;    }    return left;}int solution(const vector<int> &A) {    // write your code in C++98    //...convert original problem to counting overlap number of given interval segments    vector<IntervalSeg> intervalSeg(A.size());    for(int i = 0; i < A.size(); ++i)    {        intervalSeg[i].left = intervalSeg[i].right = i;        intervalSeg[i].left -= A[i];        intervalSeg[i].right += A[i];    }    //...first interval segments first    sort(intervalSeg.begin(), intervalSeg.end());    //...enumerate each interval segment and using binary search to calculate how much interval segments overlap it    long long result = 0;    for(int i = 0; i < intervalSeg.size(); ++i)    {        long long curRight = intervalSeg[i].right;        int firstBiggerIdx = getFirstBiggerIdx(intervalSeg, curRight);        result += firstBiggerIdx-i-1;    }    //...return result    long long MAX_LIMIT = 10000000;    if(result > MAX_LIMIT) return -1;    else return (int)result;}

原创粉丝点击