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

专栏《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

猜你喜欢

转载自blog.csdn.net/qq_35164554/article/details/122173853