题目大意:给出一个n个数的数列a,对于第i个元素ai定义fi=min(abs(ai−aj)),(1<=j<i),其中f1=a1。输出sum(fi)(1<=i<=n)
题解:稍有常识的人都能看出,这是一道平衡树模板,然而线段树这么神,当然要致辞它啦
我的收获:权值线段树神啊
#include<cstdio>
#include<algorithm>
#include<map>
#define N 50010
#define inf 0x7fffffff
using namespace std;
int n,a[N],d[N];
struct Data{
int x,ord;
}data[N];
bool cmp(Data a,Data b){
return a.x<b.x;
}
struct node{
int l,r,sum,mx,mn;
}t[N<<2];
void pushup(int rt){
t[rt].sum=t[rt<<1].sum+t[rt<<1|1].sum;
t[rt].mx=max(t[rt<<1].mx,t[rt<<1|1].mx);
t[rt].mn=min(t[rt<<1].mn,t[rt<<1|1].mn);
}
void build(int rt,int l,int r){
t[rt].l=l;t[rt].r=r;t[rt].mx=-1;t[rt].mn=inf-1;
if(l==r) return;
int mid=(l+r)>>1;
if(l<=mid) build(rt<<1,l,mid);
if(r>mid) build(rt<<1|1,mid+1,r);
}
void modify(int rt,int pos){
if(t[rt].l==t[rt].r){
t[rt].sum=1;
t[rt].mn=t[rt].l;
t[rt].mx=t[rt].l;
return;
}
int mid=(t[rt].l+t[rt].r)>>1;
if(pos<=mid) modify(rt<<1,pos);
if(pos>mid) modify(rt<<1|1,pos);
pushup(rt);
}
int querymx(int rt,int l,int r){
if(r<l) return -1;
if(t[rt].sum==0) return -1;
if(l<=t[rt].l&&t[rt].r<=r) return t[rt].mx;
int mid=(t[rt].l+t[rt].r)>>1,ret=-2;
if(l<=mid) ret=max(ret,querymx(rt<<1,l,r));
if(r>mid) ret=max(ret,querymx(rt<<1|1,l,r));
return ret;
}
int querymn(int rt,int l,int r){
if(r<l) return inf-1;
if(t[rt].sum==0) return inf-1;
if(l<=t[rt].l&&t[rt].r<=r) return t[rt].mn;
int mid=(t[rt].l+t[rt].r)>>1,ret=inf;
if(l<=mid) ret=min(ret,querymn(rt<<1,l,r));
if(r>mid) ret=min(ret,querymn(rt<<1|1,l,r));
return ret;
}
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++){
if(scanf("%d",&a[i])==EOF) a[i]=0;
data[i].ord=i;data[i].x=a[i];
}
sort(data+1,data+1+n,cmp);
int ord=1;d[1]=data[1].x;a[data[1].ord]=ord;
for(int i=2;i<=n;i++){
if(data[i].x!=data[i-1].x) ord++;
a[data[i].ord]=ord;d[ord]=data[i].x;
}
build(1,1,ord);
int ans=d[a[1]];modify(1,a[1]);
for(int i=2;i<=n;i++){
int x=querymx(1,1,a[i]-1),y=querymn(1,a[i],n),tmp=inf;
if(x!=-1) tmp=min(tmp,d[a[i]]-d[x]);
if(y!=inf-1) tmp=min(tmp,d[y]-d[a[i]]);
ans+=tmp;
modify(1,a[i]);
}
printf("%d\n",ans);
return 0;
}
当然,用双向链表维护也是支持的
#include <algorithm>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <queue>
using namespace std;
#define INF 0x3f3f3f3f
const int M=100010;
int n,rank[M];
int nxt[M],pre[M];
long long ans;
struct data{int id,v;}a[M];
bool operator <(data x,data y){return x.v<y.v;}
void work()
{
for(int i=n;i>1;i--)
{
int x=rank[i],l=INF,r=INF;
if(pre[x]>=1&&pre[x]<=n) l=a[x].v-a[pre[x]].v;
if(nxt[x]>=1&&nxt[x]<=n) r=a[nxt[x]].v-a[x].v;
ans+=min(l,r);
nxt[pre[x]]=nxt[x];
pre[nxt[x]]=pre[x];
}
printf("%lld\n",ans+a[rank[1]].v);
}
void init()
{
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%d",&a[i].v),a[i].id=i;
sort(a+1,a+1+n);
for(int i=1;i<=n;i++) rank[a[i].id]=i;
for(int i=1;i<=n;i++) pre[i]=i-1,nxt[i]=i+1;
}
int main()
{
init();
work();
return 0;
}