LeetCode刷题日记2021-12-27/825. 适龄的朋友-排序&双指针

最新推荐文章于 2024-03-10 20:47:57 发布

Bu9_Maker

最新推荐文章于 2024-03-10 20:47:57 发布

阅读量376

点赞数

分类专栏： Leetcode每日一题文章标签： leetcode 算法 python

本文链接：https://blog.csdn.net/qq_39225746/article/details/122165565

版权

Leetcode每日一题专栏收录该内容

184 篇文章 0 订阅

订阅专栏

本文介绍了如何解决一个关于在社交媒体上根据用户年龄判断好友请求的问题。给定一个年龄数组，用户不会向满足特定条件的其他用户发送好友请求。通过排序年龄并使用双指针方法，可以计算出总的请求次数。代码实现中，首先对年龄数组进行排序，然后遍历数组，根据题目条件判断可能的好友请求，并更新计数。最后返回好友请求的总数。

摘要由CSDN通过智能技术生成

题目描述

在社交媒体网站上有 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 = [16,17,18]
输出：2
解释：产生的好友请求为 17 -> 16 ，18 -> 17 。

示例 3：

输入：ages = [20,30,100,110,120]
输出：3
解释：产生的好友请求为 110 -> 100 ，120 -> 110 ，120 -> 100 。

提示：

n == ages.length
1 <= n <= 2 * 104
1 <= ages[i] <= 120

题解思路

题目给出三个条件下 x不会向y发送好友请求

age[y] <= 0.5 * age[x] + 7
age[y] > age[x]
age[y] > 100 && age[x] < 100

我们将数组从小到大排序之后按照题目给的条件来判断x是否会向y发送好友请求

因为我们是从小到大排序所以当x<15的时候则y不存在

设定左右指针来规定x给y 发送的好友请求的范围按照条件依次遍历并且滑动指针

题解代码

class Solution:
    def numFriendRequests(self, ages: List[int]) -> int:
        ages.sort()
        n=len(ages)

        left,right,ans=0,0,0

        for age in ages:
            if age<15:
                continue

            while ages[left]<=0.5*age+7:
                left+=1
            while right+1<n and ages[right+1]<=age:
                right+=1
            ans+=right-left
        return ans