权值线段树

(1).可以计算出整个数组第k大\小
(2).可以计算出i-j在数组中出现的个数。
扩展:
1.计算逆序对个数,从第一个开始,每次添加一个数,记录前缀中大于a[i]的数的个数即可,利用到了上面的(1)。题目hdoj1394,代码:

#include<bits/stdc++.h>
using namespace std;
const int maxn=5000+11;
const int mod=1e9+7;
const double eps=0.000001;
typedef long long ll;
typedef long double lb;
typedef double db;
struct D{
    int l,r,w;
}tr[maxn*4];
int a[maxn],b[maxn],c[maxn];
void build(int rt,int l,int r){
    tr[rt].l=l,tr[rt].r=r,tr[rt].w=0;
    if(l==r) return;
    int mid=(l+r)>>1;
    build(rt<<1,l,mid);
    build(rt<<1|1,mid+1,r);
}
int query(int rt,int l,int r){
    if(tr[rt].l>=l&&tr[rt].r<=r){
        return tr[rt].w;
    }
    if(tr[rt].r<l||tr[rt].l>r) return 0;
    int sum=0;
    if(tr[rt<<1].r>=l) sum+=query(rt<<1,l,r);
    if(tr[rt<<1|1].l<=r) sum+=query(rt<<1|1,l,r);
    return sum;
}
void update(int rt,int x){
    if(tr[rt].l==tr[rt].r){
        tr[rt].w++;
        return;
    }
    if(tr[rt<<1].r>=x) update(rt<<1,x);
    if(tr[rt<<1|1].l<=x) update(rt<<1|1,x);
    tr[rt].w=tr[rt<<1].w+tr[rt<<1|1].w;
}
int main(){
    int n;
    while(scanf("%d",&n)!=EOF){
         build(1,0,n-1);
         int ans=0;
         for(int i=1;i<=n;++i){
            scanf("%d",&a[i]);
            int cnt=query(1,a[i]+1,n-1);
            ans+=cnt;
            c[i]=i-1-cnt;
            b[i]=a[i]-(i-1-cnt);
            update(1,a[i]);
         }
        int sum=ans;
        for(int i=1;i<=n;++i){
            sum=sum+n-1-2*b[i]-2*c[i];
            if(ans>sum) ans=sum;
        }
        printf("%d\n",ans);
    }
    system("pause");
}

洛谷p1637 代码:三元逆序对个数,每个数求出左边小于的个数L[i]和右边大于的个数R[i],相乘的和即可。

#include<bits/stdc++.h>
using namespace std;
const int maxn=3e4+11;
const int mod=1e9+7;
const double eps=0.000001;
typedef long long ll;
typedef long double lb;
typedef double db;
struct D{
    int l,r,w;
}t1[maxn*4],t2[maxn*4];
struct A{
    ll w,d;
}a[maxn];
int x[maxn],L[maxn],R[maxn];
void build(int rt,int l,int r){
    t1[rt].l=t2[rt].l=l,t1[rt].r=t2[rt].r=r,t1[rt].w=t2[rt].w=0;
    if(l==r) return;
    int mid=(l+r)>>1;
    build(rt<<1,l,mid);
    build(rt<<1|1,mid+1,r);
}
void update1(int rt,int x){
    if(t1[rt].l==t1[rt].r){
        t1[rt].w++;
        return;
    }
    if(t1[rt<<1].r>=x) update1(rt<<1,x);
    if(t1[rt<<1|1].l<=x) update1(rt<<1|1,x);
    t1[rt].w=t1[rt<<1].w+t1[rt<<1|1].w;
}
void update2(int rt,int x){
    if(t2[rt].l==t2[rt].r){
        t2[rt].w++;
        return;
    }
    if(t2[rt<<1].r>=x) update2(rt<<1,x);
    if(t2[rt<<1|1].l<=x) update2(rt<<1|1,x);
    t2[rt].w=t2[rt<<1].w+t2[rt<<1|1].w;
}
int query1(int rt,int l,int r){
    if(t1[rt].l>=l&&t1[rt].r<=r){
        return t1[rt].w;
    }
    if(t1[rt].l>r||t1[rt].r<l) return 0;
    int sum=0;
    if(t1[rt<<1].r>=l) sum+=query1(rt<<1,l,r);
    if(t1[rt<<1|1].l<=r) sum+=query1(rt<<1|1,l,r);
    return sum;
}
int query2(int rt,int l,int r){
    if(t2[rt].l>=l&&t2[rt].r<=r){
        return t2[rt].w;
    }
    if(t2[rt].l>r||t2[rt].r<l) return 0;
    int sum=0;
    if(t2[rt<<1].r>=l) sum+=query2(rt<<1,l,r);
    if(t2[rt<<1|1].l<=r) sum+=query2(rt<<1|1,l,r);
    return sum;
}
bool cmp(A a,A b){
    return a.w<b.w;
}
int main(){
    int n;
    cin>>n;
    for(int i=1;i<=n;++i){
        scanf("%d",&a[i].w);
        a[i].d=i;
    }
    sort(a+1,a+1+n,cmp);
    int j=1;
    for(int i=1;i<=n;++i){
        if(i>1&&a[i].w!=a[i-1].w) j++;
        x[a[i].d]=j;
    }
    build(1,1,j);
    for(int i=1;i<=n;++i){
        L[i]=query1(1,1,x[i]-1);
        update1(1,x[i]);
    }
    for(int i=n;i>=1;--i){
        R[i]=query2(1,x[i]+1,j); 
        update2(1,x[i]);
    }
    ll ans=0;
    for(int i=1;i<=n;++i){
        ans+=1ll*R[i]*L[i];
    }
    cout<<ans<<endl;
    system("pause");
}

待更新。。。。。。。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值