Python｜Leetcode《825》｜适龄的朋友

Posted 2022-01-18 二哥不像程序员

专栏《LeetCode｜一刷到底》
打卡每天leetcode精选每日一题（尽量不断更！）

点击关注不迷路!!!

一、题目描述

• 题目：适龄的朋友

• 难度：中等

• 描述：在社交媒体网站上有 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 。

二、题目解析

本题的思想在于计算给定每个年龄的用户满足交友条件的用户和，首先分析条件：

1. 第一个条件给定了最小交友年龄，对于正常的年龄直接计算此值作为下限即可，但是此处又个细节在于当年龄小于15岁的时候因为该条件会导致永远不能交友，因此解题的时候需要排除该年龄段的数据。
2. 对于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