Luogu P1637 三元上升子序列【权值线段树】By cellur925

题目传送门

emmm..不开结构体的线段树真香！

首先我们知道“三元上升子序列”的个数就是对于序列中的每个数，**它左边比他小的数*它右边比他大的数**。但是如何快速求出这两个数？

我们用到权值线段树来维护。一般我们的线段树都是以下标延伸的，但是这里我们用的是权值，一般需要离散化，效果相当于一个桶。

这部分讲解请移步绝世好文

第一次我们从\(1\)~\(n\)循环是为了找它左边的，而找比他小的值是在线段树的\(1\)~\(seq[i]-1\)中找。第二次我们从\(n\)~\(1\)循环是为了找它右边的，而找比他大的值是用在线段树的\(seq[i]+1\)~\(n\)中找实现的。

不用结构体因为方便清空，\(tong\)数组记录线段树的权值，注意开\(4\)倍。

#include<cstdio>
#include<algorithm>
#include<cstring>
#define maxn 100090

using namespace std;
typedef long long ll;

int n;
ll ans,sma[maxn],bigg[maxn];
int seq[maxn],tmp[maxn],tong[maxn<<2];

int ask(int p,int l,int r,int L,int R)
{
if(L<=l&&r<=R) return tong
;
int mid=(l+r)>>1,qwq=0;
if(L<=mid) qwq+=ask(p<<1,l,mid,L,R);
if(R>mid) qwq+=ask(p<<1|1,mid+1,r,L,R);
return qwq;
}

void change(int p,int l,int r,int x)
{
if(l==x&&r==x)
{
tong[p]++;
return ;
}
int mid=(l+r)>>1;
if(x<=mid) change(p<<1,l,mid,x);
else if(x>mid) change(p<<1|1,mid+1,r,x);
tong[p]=tong[p<<1]+tong[p<<1|1];
}

int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&seq[i]),tmp[i]=seq[i];
sort(tmp+1,tmp+1+n);
int m=unique(tmp+1,tmp+1+n)-(tmp+1);
for(int i=1;i<=n;i++)
seq[i]=lower_bound(tmp+1,tmp+1+m,seq[i])-tmp;
for(int i=1;i<=n;i++)
{
if(seq[i]!=1) sma[i]=ask(1,1,n,1,seq[i]-1);
change(1,1,n,seq[i]);
}
memset(tong,0,sizeof(tong));
for(int i=n;i>=1;i--)
{
if(seq[i]!=n) bigg[i]=ask(1,1,n,seq[i]+1,n);
change(1,1,n,seq[i]);
}
for(int i=1;i<=n;i++)
ans+=sma[i]*bigg[i];
printf("%lld\n",ans);
return 0;
}

#include<cstdio>
#include<algorithm>
#include<cstring>
#define maxn 500090

using namespace std;
typedef long long ll;

int n;
ll ans,lef[maxn];
int tmp[maxn],seq[maxn],tong[maxn<<2];

int ask(int p,int l,int r,int L,int R)
{
if(L<=l&&r<=R) return tong[p];
int mid=(l+r)>>1,qwq=0;
if(L<=mid) qwq+=ask(p<<1,l,mid,L,R);
if(R>mid) qwq+=ask(p<<1|1,mid+1,r,L,R);
return qwq;
}

void change(int p,int l,int r,int x)
{
if(l==x&&r==x)
{
tong[p]++;
return ;
}
int mid=(l+r)>>1;
if(x<=mid) change(p<<1,l,mid,x);
else if(x>mid) change(p<<1|1,mid+1,r,x);
tong[p]=tong[p<<1]+tong[p<<1|1];
}

int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%d",&seq[i]),tmp[i]=seq[i];
sort(tmp+1,tmp+1+n);
int m=unique(tmp+1,tmp+1+n)-(tmp+1);
for(int i=1;i<=n;i++)
seq[i]=lower_bound(tmp+1,tmp+1+m,seq[i])-tmp;
for(int i=1;i<=n;i++)
{
if(seq[i]!=n) lef[i]=ask(1,1,n,seq[i]+1,n);
change(1,1,n,seq[i]);
}
for(int i=1;i<=n;i++)
ans+=lef[i];
printf("%lld\n",ans);
return 0;
}