一、题目描述
题目:适龄的朋友
难度:中等
描述:在社交媒体网站上有 n 个用户。给你一个整数数组 ages ,其中 ages[i] 是第 i 个用户的年龄。
如果下述任意一个条件为真,那么用户 x 将不会向用户 y(x != y)发送好友请求:
- age[y] <= 0.5 * age[x] + 7
- age[y] > age[x]
- age[y] > 100 && age[x] < 100
否则,x 将会向 y 发送一条好友请求。
注意,如果 x 向 y 发送一条好友请求,y 不必也向 x 发送一条好友请求。另外,用户不会向自己发送好友请求。
返回在该社交媒体网站上产生的好友请求总数。
示例1
输入:ages = [16,16]
输出:2
解释:2 人互发好友请求。
示例2
输入:ages = [20,30,100,110,120]
输出:3
解释:产生的好友请求为 110 -> 100 ,120 -> 110 ,120 -> 100 。
二、题目解析
本题的思想在于计算给定每个年龄的用户满足交友条件的用户和,首先分析条件:
- 第一个条件给定了最小交友年龄,对于正常的年龄直接计算此值作为下限即可,但是此处又个细节在于当年龄小于15岁的时候因为该条件会导致永远不能交友,因此解题的时候需要排除该年龄段的数据。
- 对于2、3条件,博主思前想后认为这两个条件是重复的,满足条件2的时候就已经满足条件3了,因此只需将条件2作为交友年龄的上限即可。
分析好了条件,不难看出当数据有序的时候计算就会变的容易许多,因此第一步我们需要将数据整体进行排序操作。
排序完成后,因为分析题目的时候我们发现此题具有上限和下限作为约束,因此可以考虑使用滑窗的方式进行筛选,方式如下:
移动左右指针的时候,我们只需要不断判断同时满足上述条件1、2时的情况进行计数即可。
特殊情况
当出现相同年龄的用户时,他们能够进行相互交友,此时所有相同年龄用户的交友数量应该是相同的。
例如:
用户年龄:35、37、37
其中每一个37岁的用户都能够和一个35岁和一个37岁的用户交友,因此这种情况的数量时相同的,考虑多个相同年龄用户存在的情况,我们应该算到最后一个该年龄的用户作为他们能够交友的数量,如下图所示:
此时我们重复使用最大数量作为每个35岁用户能够交友的数量即可,即每个35岁用户能够交到1个30岁的朋友和2个35岁的朋友(交友可以相互但不是必须相互,题中给的是发送交友请求)
三、解题代码
解法
按照解析中的方法进行代码编写如下:
class Solution: def numFriendRequests(self, ages: List[int]) -> int: ages.sort() left, right, res = 0, 1, 0 n = len(ages) for i in range(n): if ages[i] < 15: continue else: while ages[left] <= 0.5 * ages[i] + 7: left += 1 while right < n and ages[right] <= ages[i]: right += 1 res += right - left - 1 return res