825 适龄的朋友(字典)

1. 问题描述:

人们会互相发送好友请求,现在给定一个包含有他们年龄的数组,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.

2. 思路分析:

① 第一个比较容易想到的是暴力破解,但是提交上去超时了,第二个想到的方法是对数组中对应年龄进行计数,可以使用python中的字典进行计数,统计出当前年龄的人数,使用一个循环累加当前年龄的总人数方便后面计算发送好友请求的人数,然后遍历0-121这个年龄段能够发送的请求,公式如下:

j = i // 2 + 7
res += (dic[i] - dic[j] - 1) * (dic[i] - dic[i - 1])

j表示在0-i这个范围之内是否存在年龄使得i > j,dic[i] - dic[j]表示在i这个年龄段能够发送的好友请求(减去不符合题目的人数),-1表示自己不能够给自己发送请求,(dic[i] - dic[i - 1])表示当前年龄i的人数,我们只需要将结果相乘起来即可

3. 代码如下:

import collections
from typing import List


class Solution:
    def numFriendRequests(self, ages: List[int]) -> int:
        # 先统计1-120这个年龄段的人数是多少
        # collections.defaultdict(int)会初始化字典的值为0
        dic = collections.defaultdict(int)
        for i in ages:
            dic[i] += 1
        # 累加到当前年龄段的人数
        for i in range(1, 121):
            dic[i] += dic[i - 1]
        res = 0
        for i in range(121):
            j = i // 2 + 7
            if j >= i: continue
            res += (dic[i] - dic[j] - 1) * (dic[i] - dic[i - 1])
        return res

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值