
在社交媒体网站上有 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 发送一条好友请求。另外,用户不会向自己发送好友请求。
返回在该社交媒体网站上产生的好友请求总数。
样例: 方法一:class Solution {
public:
int numFriendRequests(vector& ages) {
int cnt[121] = {0}; //cnt[i]表示年龄为i的人数
for (int i = 0; i < ages.size(); i++)
{
cnt[ages[i]]++;
}
int pre[121] = {0}; //pre[i]表示年龄小于等于i的人数
for (int i = 1; i <= 120; i++)
{
pre[i] = pre[i - 1] + cnt[i];
}
int ans = 0;
for (int i = 15; i <= 120; i++)
{
if (cnt[i])
{
int minage = 0.5 * i + 7;
//对于每个年龄为i的人,他会向pre[i] - pre[minage] - 1个人发送好友请求,其中包括年龄等于他的人,减一是要排除自己
ans += cnt[i] * (pre[i] - pre[minage] - 1);
}
}
return ans;
}
};
看完多个题解后,发现还是这种方法最方便,由于年龄只限定在1到120之间,所以可以直接开两个数组,一个cnt[i]用于记录年龄为i的人数,一个pre[i]用于记录年龄小于等于i的人数。
由题目可知,对于每个年龄为age的人来说,其能够向以下年龄范围内的人发送好友请求:
( 0.5 * age + 7, age】
所以即是pre[i] - pre[minage] - 1,减一是因为自己不能够向自己发送好友请求。
另外当0.5 * age + 7 >= age,即age <= 14 时,此区间将不存在,即年龄小于等于14的人不能发送任何的好友请求,所以不必考虑年龄小于等于14的人。
欢迎分享,转载请注明来源:内存溢出
微信扫一扫
支付宝扫一扫
评论列表(0条)