在社交媒体网站上有 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条)