题目描述
链接:https://ac.nowcoder.com/acm/contest/1007/A
来源:牛客网
给定一个长度为 n 的序列 A,A 中的数各不相同。对于 A 中的每一个数 A_i,求:
min(1≤j<i) |A_i-A_j|
以及令上式取到最小值的 j(记为 P_i)。若最小值点不唯一,则选择使 A_j 较小的那个。
分析
做任何一道题我们首先想到的肯定是暴力,然后再慢慢优化,这道题一看,如果我们直接暴力枚举一个一个找,时间复杂度为O(n*n),那必然超时,这个时候我们就自然而然想到了链表,首先我们需要用nlogn的时间复杂度去排个序,这样我们每次找都只需要找他的前驱和后继,因为这三个从小到大排序后相邻,差值必然比前面的小,然后我们每次询问,比较,删除都是O(1)的时间复杂度,忽略常数项,总的时间复杂度就是O(nlogn),完美通过。
但除了链表,还有没有什么其他方法呢,当然有,那就是平衡树(STL set),同样的,我们只需要排好序后,查询前驱和后继就可以了,而二叉平衡树刚好就是一个支持动态插入,查询前驱,查询后继的数据结构。
理论有了,直接实践。
还有具体解释都在代码里了
链表
#include<bits/stdc++.h>
using namespace std;
int read(){
int x=0,f=1;char ch=getchar();
while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
while(isdigit(ch)){x=(x<<3)+(x<<1)+ch-48;ch=getchar();}
return x*f;
}
const int N=1e5+10;
int n;
struct data{
int a,id,nex,pre;
}s[N];
int num[N];
int nod[N];
int ans[N];
bool cmp(const data &x,const data &y)
{
if(x.a==y.a)
return x.id<y.id;
return x.a<y.a;
}
signed main()
{
n=read();
for(int i=1;i<=n;i++)
{
s[i].a=read();
s[i].id=i;
}
sort(s+1,s+1+n,cmp);
for(int i=1;i<=n;i++)
{
s[i].pre=i-1;//前驱
s[i].nex=i+1;//后继
num[s[i].id]=i;//表示原序列中的a位于链表的哪一位置
}
int k,fa,son;
for(int i=n;i>=2;i--)//从后往前数,排除干扰
{
k=num[i];//记录当前位置的编号
fa=s[k].pre;//前驱(比它小)
son=s[k].nex;//后继(比他大)
//每计算一个数,就将这个数删去
s[fa].nex=son;//后继指针指向后一个
s[son].pre=fa;//前驱指针指向前一个
//从逻辑上删去了s[k].a这个数
int p=s[son].a-s[k].a;//求与他最接近的两个数与他的差值
int q=abs(s[fa].a-s[k].a);
if(son==n+1) p=0x3fffffff;//没有后继
if(fa==0) q=0x3fffffff;//没有前驱
if(p<q)
{
ans[i]=p;//选择差值小
nod[i]=s[son].id;
}
else if(p>q)
{
ans[i]=q;//选择差值小
nod[i]=s[fa].id;
}
else
{
ans[i]=q;
nod[i]=s[fa].id;//选择序号小
}
}
for(int i=2;i<=n;i++)
cout<<ans[i]<<" "<<nod[i]<<endl;
return 0;
}
二叉平衡树
#include<bits/stdc++.h>
using namespace std;
int read(){
int x=0,f=1;char ch=getchar();
while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
while(isdigit(ch)){x=(x<<3)+(x<<1)+ch-48;ch=getchar();}
return x*f;
}
int n,a;
set<pair<int,int> > s;//第一个存数字,第二个存序号
signed main()
{
n=read();
a=read();//先把第一个读进去,因为他不参与计算
s.insert(make_pair(a,1));
for(int i=2;i<=n;i++)
{
a=read();
s.insert(make_pair(a,i));
set<pair<int,int> >::iterator it=s.find(make_pair(a,i));//定义一个指针,查找他的序号
pair<int,int> ans;//第一个存差值,第二个存序号
ans.first=0x7f7f7f7f;
if(++it!=s.end())//如果他还有后继
ans=make_pair((*it).first-a,(*it).second);//他的后继减它本身
it=s.find(make_pair(a,i));//指针重置
if(it--!=s.begin()&&ans.first>=a-(*it).first)//如果他还有前驱并且前驱的差值更小,选择更小的
ans=make_pair(a-(*it).first,(*it).second);
cout<<ans.first<<" "<<ans.second<<endl;
}
return 0;
}完美不!完美!撒花。

京公网安备 11010502036488号