在社交媒体网站上有 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 发送一条好友请求。另外,用户不会向自己发送好友请求。
返回在该社交媒体网站上产生的好友请求总数。
很容易发现条件3蕴含在条件2中,所以不成问题,然后写了个铸币算法,看看样例范围就知道TLE了……我是废物,这个时间复杂度显然是无法接受的
class Solution {
public int numFriendRequests(int[] ages) {
int ans=0;
Arrays.sort(ages);
for(int i=0;i<ages.length;i++){
for(int j=i+1;j<ages.length;j++){
boolean itoj=true,jtoi=true;
if(ages[j]<=0.5 * ages[i] + 7){
itoj=false;
}
if(ages[i]<ages[j]){
itoj=false;
}
if(ages[i]<=0.5 * ages[j] + 7){
jtoi=false;
}
if(ages[i]>ages[j]){
jtoi=false;
}
if(itoj){
ans++;
}
if(jtoi){
ans++;
}
}
}
return ans;
}
}
再看看题目0.5×ages[x]+7<ages[y]≤ages[x],因此age>14的才有交朋友的权力(年龄歧视!)
age[y]的范围为(0.5×ages[x]+7,ages[x]),左右都是递增,为了利用这个性质我们将ages排序再遍历,如果边界不符合就更新边界即可,不符合就将指针移动一位,这样left-right区间内的年龄都符合要求,去掉y自己,剩下的都是可以交朋友的人(注意y和x交朋友x未必同意,还得回头检查一遍x)
class Solution {
public int numFriendRequests(int[] ages) {
int n = ages.length;
Arrays.sort(ages);
int left = 0, right = 0, ans = 0;
for (int age : ages) {
if (age < 15) {
continue;
}
while (ages[left] <= 0.5 * age + 7) {
++left;
}
while (right + 1 < n && ages[right + 1] <= age) {
++right;
}
ans += right - left;
}
return ans;
}
}