线段树菜鸟一题+归并排序【求逆序数】POJ2299

题目链接：http://poj.org/problem?id=2299

归并排序解法链接：http://blog.csdn.net/lyy289065406/article/details/6647346

然后是自己写的线段树：

注意点在代码中。

 

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

#define lson  rt<<1,l,m
#define rson  rt<<1|1,m+1,r//2. 这一块的l,1分清楚

const int maxn=500005;

struct M
{
    int x;
    int id;
}m[maxn];

int b[maxn];
struct Tree
{
    int l,r,sum;
}tree[maxn*4];// 1.此处的结点maxn*4? 开maxn*2会RE

int cmp(struct M a,struct M b)
{
    return a.x<b.x;
}
void Pushup(int rt)
{
     tree[rt].sum=tree[rt<<1].sum+tree[rt<<1|1].sum;
}

void build(int rt,int l,int r)
{
    if(l==r) {tree[rt].sum=0;return;}
    else
    {
       int m=(l+r)>>1;
       build(lson);
       build(rson);
    }
    Pushup(rt);
    //tree[rt].sum=tree[rt<<1].sum+tree[rt<<1|1].sum;
}
void update(int p,int add,int rt,int l,int r)
{
    if(p>r) return;
    if(l==r)
    {
        tree[rt].sum+=add;
        return;
    }
    else
    {
        int m=(l+r)>>1;
        if(p<=m) update(p,add,lson);//3. 要p的作用哪。只更新一半 。因为下面有Pushup
        else update(p,add,rson);
    }
    Pushup(rt);
}
int query(int a,int b,int rt,int l,int r)
{
    int ans=0;
    if(a<=l&&b>=r)
    {
        return tree[rt].sum;
    }
    else
    {
        int m=(l+r)>>1;
        if(a<=m)
            ans+=query(a,b,lson);
        if(b>m) //4. 这地方是r>m果断而不是r>=m
            ans+=query(a,b,rson);
    }
    return ans;
}
int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        if(!n) break;

        for(int i=1;i<=n;i++)
        {
            scanf("%d",&m[i].x);
            m[i].id=i;
        }

        sort(m+1,m+1+n,cmp);
        
        //离散化
        b[m[1].id]=1;
        for(int i=2;i<=n;i++)
        {
            if(m[i].x==m[i-1].x)
              b[m[i].id]=b[m[i-1].id];
            else b[m[i].id]=i;
        }

        build(1,1,n);
    
        long long ans=0;
        for(int i=1;i<=n;i++)
        {
            ans+=query(b[i]+1,n,1,1,n);
            update(b[i],1,1,1,n);
        }
        printf("%lld\n",ans);

    }
    return 0;
}
//还有不懂的时候调试一下，出现奇怪的地方一般是自己错了。   或者模板敲错。