目录
题目描述:
在社交媒体网站上有 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<int>& 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的人。
方法二
class Solution {
public:
//利用题目所给的三个条件,判断x是否能够向y发送好友请求
bool judge(int agex, int agey)
{
if (agey <= 0.5 * agex + 7) return false;
if (agey > agex) return false;
if (agey > 100 && agex < 100) return false;
return true;
}
int numFriendRequests(vector<int>& ages) {
//sumages[i]表示年龄为i的人数
vector<int> sumages(121, 0);
for (int age : ages) sumages[age]++;
int ans = 0;
//根据条件2可知,用户只有可能向年龄小于等于自己的人发送好友请求,因此年龄从大往小遍历
for (int agex = 120; agex > 0; agex--)
{
if (!sumages[agex]) continue; //如果年龄为agex的人不存在,跳过
for (int agey = agex; agey > 0; agey--)
{
if (!sumages[agey]) continue; //如果年龄为agey的人不存在,跳过
//如果agex和agey满足条件,就发送好友请求
if (judge(agex, agey))
{
//当agex == agey时,将会产生sumages[agex] * (sumages[agex] - 1)条好友请求(排列数A(2,2))
//当agex != agey时,将会产生sumages[agex] * sumages[agey]条好友请求
ans += agex == agey ? sumages[agex] * (sumages[agex] - 1) : sumages[agex] * sumages[agey];
}
}
}
return ans;
}
};
以前写过的那种前缀和的方法,在边界上实在令人头疼,真不想再看一遍了,一年过后自己写了这种比较简单的方法,时间复杂度不超过10^4