【值域线段树学习】 P1688 求逆序对

版权声明:http://blog.csdn.net/baidu_35643793 https://blog.csdn.net/baidu_35643793/article/details/78112415
 时间限制: 1 s
 空间限制: 128000 KB
 题目等级 : 黄金 Gold

题目描述 Description
给定一个序列a1,a2,…,an,如果存在i<j并且ai>aj,那么我们称之为逆序对,求逆序对的数目
数据范围:N<=1e5。Ai<=1e5。时间限制为1s。
输入描述 Input Description
第一行为n,表示序列长度,接下来的n行,第i+1行表示序列中的第i个数。
输出描述 Output Description
所有逆序对总数.
样例输入 Sample Input
4
3
2
3
2
样例输出 Sample Output
3

求逆序问题可以用树状数组解决;本题作为值域线段树入门题。

值域线段树:将a[i]作为一个节点编号,每输入一个a[i],使a[i]号节点的值++

代码:

#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;
typedef long long ll;

vector<int>v;
const int N=100010;
struct node
{
    int l,r;
    ll val;
    int mid()
    {
        return (l+r)>>1;
    }
}tree[N<<2];
int a[N+10];

int getid(int x)
{
    return lower_bound(v.begin(),v.end(),x)-v.begin()+1;
}

void push_up(int rt)
{
    tree[rt].val=tree[rt<<1].val+tree[rt<<1|1].val;
}

void build(int rt,int l,int r)
{
    tree[rt].l=l;
    tree[rt].r=r;
    tree[rt].val=0;
    if(l==r) return ;

    int m=tree[rt].mid();
    build(rt<<1,l,m);
    build(rt<<1|1,m+1,r);
    push_up(rt);
}

void update(int rt,int l,int val)
{
    if(tree[rt].l==l&&tree[rt].r==l)
    {
        tree[rt].val+=val;
        return ;
    }

    int m=tree[rt].mid();
    if(l<=m) update(rt<<1,l,val);
    if(l>m) update(rt<<1|1,l,val);
    push_up(rt);
}

ll query(int rt,int l,int r)
{
    if(tree[rt].l==l&&tree[rt].r==r)
        return tree[rt].val;

    int m=tree[rt].mid();
    if(r<=m) return query(rt<<1,l,r);
    else if(l>m) return query(rt<<1|1,l,r);
    else return query(rt<<1,l,m)+query(rt<<1|1,m+1,r);
}

int main()
{
    int n;
    scanf("%d",&n);
    for(int i=0;i<n;i++)
    {
        scanf("%d",&a[i]);
        v.push_back(a[i]);
    }

    build(1,1,N);
    ll ans=0;

    sort(v.begin(),v.end());
    v.erase(unique(v.begin(),v.end()),v.end());

    for(int i=0;i<n;i++)
    {
        int x;
        x=getid(a[i]);
        ll an=query(1,x+1,N);
//        printf("yes\n");
        ans+=an;
        update(1,x,1);
    }
    printf("%lld\n",ans);

    return 0;
}

//本代码用了离散化,虽然题目数据显然不需要离散化!

如果数据范围变大到1e6 这个值域线段树就不中了,树状数组可以!

展开阅读全文

没有更多推荐了,返回首页