找朋友题解

限制

时间限制: 1 Sec 内存限制: 128 MB

题目描述

        根据社会学研究表明，人们都喜欢找和自己身高相近的人做朋友。现在有n名身高两两不相同的同学依次走入教室，调查人员想预测每个人在走入教室的瞬间最想和已经进入教室的哪个人做朋友。当有两名同学和这名同学的身高差一样时，这名同学会更想和高的那个人做朋友。比如一名身高为1.80米的同学进入教室时，有一名身高为1.79米的同学和一名身高为1.81米的同学在教室里，那么这名身高为1.80米的同学会更想和身高为1.81米的同学做朋友。对于第一个走入教室的同学我们不做预测。

输入

第一行：一个整数n，表示n个同学依次走进教室

接下来n行，每行一个浮点数a，表示同学的身高

输出

输出n-1行，分别表示第2到n个同学的朋友的编号

样例输入

5
1.1
1.3
1.2
1.123456
1.000001

样例输出

1
2
1
1

数据规模

n不超过10万，身高各不相同，精确到微米

分析

可以用链表做。数组l,r记录左右，数组a记录身高，数组s记录答案。

代码

#include<bits/stdc++.h>
using namespace std;
int n,i,s[100005],l[100005],r[100005],w[100005];
double a[100005],x,y;
int cmp(int x,int y){
	return a[x]<a[y];
}
int main(){
	scanf("%d",&n);
	for(i=1;i<=n;++i){
		scanf("%lf",&a[i]);
		w[i]=i;
	}
	sort(w+1,w+n+1,cmp);
	for(i=1;i<=n;++i) l[w[i]]=w[i-1],r[w[i]]=w[i+1];
	for(i=n;i>1;--i){
		x=y=1e9;
		if(l[i]>0) x=fabs(a[l[i]]-a[i]);
		if(r[i]>0) y=fabs(a[r[i]]-a[i]);
		if(y-1e-7>x) s[i]=l[i];
		else s[i]=r[i];
		r[l[i]]=r[i],l[r[i]]=l[i],l[i]=r[i]=a[i]=0;
	}
	for(i=2;i<=n;++i) printf("%d\n",s[i]);
    return 0;
}