[数据结构与算法]LeetCode 825. 适龄的朋友--多层二分

825. 适龄的朋友

人们会互相发送好友请求，现在给定一个包含有他们年龄的数组，ages[i] 表示第 i 个人的年龄。

当满足以下任一条件时，A 不能给 B（A、B不为同一人）发送好友请求：

age[B] <= 0.5 * age[A] + 7
age[B] > age[A]
age[B] > 100 && age[A] < 100

否则，A 可以给 B 发送好友请求。

注意如果 A 向 B 发出了请求，不等于 B 也一定会向 A 发出请求。而且，人们不会给自己发送好友请求。

求总共会发出多少份好友请求?

示例 1：

输入：[16,16]
输出：2
解释：二人可以互发好友申请。

示例 2：

输入：[16,17,18]
输出：2
解释：好友请求可产生于 17 -> 16, 18 -> 17.

示例 3：

输入：[20,30,100,110,120]
输出：3
解释：好友请求可产生于 110 -> 100, 120 -> 110, 120 -> 100.

提示：

1 <= ages.length <= 20000
1 <= ages[i] <= 120

题解

比较简单吧，先排序，然后就是针对每个人的年龄不断二分去找符合条件的区间年龄。

AC代码

class Solution {
public:
    int numFriendRequests(vector<int>& ages) 
    {
        sort(ages.begin(),ages.end());
        int res=0;
        for(int i=0;i<ages.size();i++)
        {
            int d=ages[i]/2+7;
            int l=0,r=ages.size()-1,fin=-1;
            while(l<=r)
            {
                int mid=(l+r)/2;
                if(ages[mid]>d)
                {
                    fin=mid;
                    r=mid-1;
                }
                else l=mid+1;
            }
            if(fin==-1)continue;
            l=fin,r=ages.size()-1;
            int fin2=-1;
            while(l<=r)
            {
                int mid=(l+r)/2;
                if(ages[mid]<=ages[i])
                {
                    fin2=mid;
                    l=mid+1;
                }
                else r=mid-1;
            }
            if(fin2==-1)continue;
            //这里特别比较
            if(ages[i]<100)//那么ages[B]只能小于等于100
            {
                l=fin,r=fin2;
                int fin3=-1;
                while(l<=r)
                {
                    int mid=(l+r)/2;
                    if(ages[mid]<=100)
                    {
                        fin3=mid;
                        l=mid+1;
                    }
                    else r=mid-1;
                }
                if(fin3==-1)continue;
                if(fin<=i&&i<=fin3)
                res+=(fin3-fin);
                else
                res+=(fin3-fin+1);
            }
            else//如果ages[A]>=100，这里age[B]就无所谓了
            {
                if(fin<=i&&i<=fin2)
                res+=(fin2-fin);
                else
                res+=(fin2-fin+1);
            }
        }
        return res;
    }
};