P4755-Beautiful Pair【笛卡尔树,线段树】

正题

题目链接:https://www.luogu.com.cn/problem/P4755

---

题目大意

(n)个数字的一个序列，求有多少个点对(i,j)满足(a_i imes a_jleq max{a_k}(kin[l,r]))

---

解题思路

如果构建一棵笛卡尔树的话那么两个点之间的(max)就在笛卡尔树的(LCA)位置。

所以对于每个位置维护一个线段树，然后每次暴力枚举小的那棵子树在大子树的线段树中查询即可。然后线段树合并或者启发式合并上去就好了。

建笛卡尔树的时候用( ext{RMQ})查询区间最大值然后递归下去就好了。

当然因为是乘法所以小的那个值域不会超过(sqrt{10^9})所以也可以树状数组+启发式合并。

这里写的是线段树的做法，时间复杂度都是(O(nlog^2 n))

注意(1)要特判就好了

---

code

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=1e5+10,L=20;
int n,a[N],lg[N],f[N][L+1],inf;
long long ans;
struct Seg_Tree{
    int cnt,w[N<<6],ls[N<<6],rs[N<<6];
    void Change(int &x,int L,int R,int pos,int val){
        if(!x)x=++cnt;w[x]+=val;
        if(L==R)return;
        int mid=(L+R)>>1;
        if(pos<=mid)Change(ls[x],L,mid,pos,val);
        else Change(rs[x],mid+1,R,pos,val);
        return;
    }
    int Ask(int x,int L,int R,int l,int r){
        if(!x||l>r)return 0;
        if(L==l&&R==r)return w[x];
        int mid=(L+R)>>1;
        if(r<=mid)return Ask(ls[x],L,mid,l,r);
        if(l>mid)return Ask(rs[x],mid+1,R,l,r);
        return Ask(ls[x],L,mid,l,mid)+Ask(rs[x],mid+1,R,mid+1,r);
    }
    int Merge(int x,int y,int L,int R){
        if(!x||!y)return x+y;
        int mid=(L+R)>>1;w[x]+=w[y];
        if(L==R)return x;
        ls[x]=Merge(ls[x],ls[y],L,mid);
        rs[x]=Merge(rs[x],rs[y],mid+1,R);
        return x;
    }
}T;
int Ask(int l,int r){
    int z=lg[r-l+1];
    int x=f[l][z],y=f[r-(1<<z)+1][z];
    return (a[x]>=a[y])?x:y;
}
int solve(int l,int r){
    if(l>r)return 0;
    int x=Ask(l,r),ls,rs;
    ls=solve(l,x-1);
    rs=solve(x+1,r);
    if(ls)ans+=T.Ask(ls,1,inf,1,1);
    if(rs)ans+=T.Ask(rs,1,inf,1,1);
    if(x-l<=r-x){
        for(int i=l;i<x;i++)
            ans+=T.Ask(rs,1,inf,1,a[x]/a[i]);
    }
    else{
        for(int i=x+1;i<=r;i++)
            ans+=T.Ask(ls,1,inf,1,a[x]/a[i]);
    }
    ls=T.Merge(ls,rs,1,inf);
    T.Change(ls,1,inf,a[x],1);
    return ls;
}
int main()
{
    // printf("%d
",sizeof(T)>>20);
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
        inf=max(inf,a[i]);
        ans+=(a[i]==1);
        f[i][0]=i;
    }
    inf=1e9;
    for(int i=2;i<=n;i++)lg[i]=lg[i>>1]+1;
    for(int j=1;(1<<j)<=n;j++)
        for(int i=1;i+(1<<j)-1<=n;i++){
            int x=f[i][j-1],y=f[i+(1<<j-1)][j-1];
            if(a[x]>=a[y])f[i][j]=x;
            else f[i][j]=y;
        }
    solve(1,n);
    printf("%lld",ans);
    return 0;
}