找朋友
描述
QQ推荐好友时,人们总会根据一些要求去互相发送好友请求,现在给定一个大小为n的包含有他们年龄的数组a[],a[i]表示第i个人的年龄。
当满足以下任一条件时,i不能给j(i、j不为同一人)发送好友请求:
a[j] <= 0.5 * a[i] + 5
a[j] > a[i]
a[j] >= 100 && a[i] <= 100否则,i可以给j发送好友请求。
注意如果i向j发出了请求,不等于j也一定会向i发出请求。而且,人们不会给自己发送好友请求。
求总共会发出多少份好友请求?
输入
第一行输入数字n,表示年龄数组a[]中元素个数。
第二行输入n个数字,空格分隔。分别表示a[0],a[1],a[2]…a[n-1]。
输出
输出总共发送好友请求数。
输入样例 1
3
16 17 18
输出样例 1
3
提示
1 <= n <= 200000
1 <= a[i] <= 120
能倒是能出答案,就是超时了
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string.h>
using namespace std;
//我没法优化了,我还能想到的就是图
//服了,居然还可以这么想,人傻了
const int maxn = 2*(1e5);
int a[maxn];
int main(){
int n;
int i = 0,j = 0;
int temp,result;
int cnt = 0;
scanf("%d",&n);
memset(a,0,sizeof(a));
for(i = 0;i < n;++i){
scanf("%d",&a[i]); //为啥这里必须加&啊?
}
sort(a,a+n);
for(i = 0;i < n;++i){//挨个遍历
temp = a[i] * 0.5 + 5;
int pos = lower_bound(a, a+n, a[i]) - a;//返回 第一个大于的下标
for(j = 0;j < pos;++j){//按正常逻辑来说,已经按顺序排列,a[j]>a[i]就不能发送,查找就限定在这里
if(a[j] >= 100 && a[i] <= 100) continue;
else{
if(a[j] > temp && i != j && a[j] <= a[i]){
++cnt;
}
}
}
}
printf("%d",cnt);
return 0;
}
居然还可以这么写。
基本思路都是,按照已知顺序的范围来确定人数。
但是就是实现的时候,我太暴力了,当时看到这个第一眼就想的是用图,可是我现在还写不来图。
思路的细节有很多不一样。
#include<iostream>//正确答案//完全没用c++函数,改下头文件就是c
#include<cstdio>
#include<algorithm>
using namespace std;
int a[121];
int main(){
long long n;
int i = 0,j = 0;
long long cnt = 0;
int temp;
int compare;
scanf("%lld",&n);
memset(a,0,sizeof(a));
for(i = 0;i < n;++i){
scanf("%d",&temp);
++a[temp];
}
for(i = 1;i <= 120;++i){
if(a[i] != 0){
compare = i/2+5;
for(j = compare+1;j <= i;++j){//因为范围是从 i/2+5 -> i
if(j >= 100 && i > 100 || j < 100 && i <= 100 || j < 100 && i > 100) {
cnt += a[i]*a[j];//同岁的不止一个
if(i == j)//针对同岁、自己给自己发
if( a[i] >= 1)
cnt -= a[j];
}
}
}
}
printf("%lld",cnt);
return 0;
}
关键在于:同岁的也可以互相发信息,所以只要排除掉自己给自己发的次数就行了