码迷,mamicode.com
首页 > 其他好文 > 详细

线段树求逆序对

时间:2018-03-21 16:35:43      阅读:158      评论:0      收藏:0      [点我收藏+]

标签:iostream   data   sum   scanf   add   ==   逆序对   好的   开始   

  • 思路:

离散化变成一个1-n的数组表示每个数的排名,然后按顺序插入各个数排名,并且查询比它排名大的数的个数。

这个离散化方法还是比较好的,思维难度和代码难度都比较小。

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <iostream>
using namespace std;
const int maxn=40005;
struct Node{
    int data,rank,index;
}num[maxn];
int n,sum[maxn<<2];
int t,L,R;
inline bool cmp1(const Node & a,const Node & b)
{
    return a.data<b.data;
}
inline bool cmp2(const Node & a,const Node & b)
{
    return a.index<b.index;
}
void add(int now,int l,int r)
{
    if(l==r)
    {
        sum[now]++;
        return ;
    }
    int mid=(l+r)>>1;
    if(t<=mid)add(now<<1,l,mid);
    else add(now<<1|1,mid+1,r);
    sum[now]=sum[now<<1]+sum[now<<1|1];
    return ;    
}
int query(int now,int l,int r)
{
    if(L<=l&&r<=R)
    {
        return sum[now];
    }
    int mid=(l+r)>>1;
    int ans=0;
    if(L<=mid)ans+=query(now<<1,l,mid);
    if(mid<R)ans+=query(now<<1|1,mid+1,r);
    return ans;
}
int main()
{
    int n,anss=0;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&num[i].data);
        num[i].index=i;
    }
    sort(num+1,num+1+n,cmp1);//离散化开始
    for(int i=1;i<=n;i++)num[i].rank=i;
    sort(num+1,num+1+n,cmp2);
    for(int i=1;i<=n;i++)
    {
        t=num[i].rank,L=t+1,R=n;
        add(1,1,n);
        anss+=query(1,1,n);
    }       
    cout<<anss<<endl;
    return 0;
}

线段树求逆序对

标签:iostream   data   sum   scanf   add   ==   逆序对   好的   开始   

原文地址:https://www.cnblogs.com/Rye-Catcher/p/8617470.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!