hdu 1394 线段树or 树状数组

学习线段树，把以前的树状数组也复习一下。

Minimum Inversion Number

求最小的逆序数

#include<stdio.h>
struct st
{
    int l;
    int r;
    int sum;
}f[4*5002];
int date[5002];
void build(int l,int r,int n)
{
    int mid=(l+r)/2;
    f[n].l=l;
    f[n].r=r;
    if(l==r)
        f[n].sum=date[l];
    else
    {
        build(l,mid,n*2);
        build(mid+1,r,n*2+1);
        f[n].sum=f[n*2].sum+f[n*2+1].sum;
    }
}
void update(int wz,int num,int n)
{
    int mid=(f[n].l+f[n].r)/2,t;
    if(f[n].l==wz&&f[n].r==wz)
    {
        f[n].sum=f[n].sum+num;
        t=n/2;
        while(t)
        {
            f[t].sum=f[t*2].sum+f[t*2+1].sum;
            t=t/2;
        }
    }
    else if(mid>=wz)
        update(wz,num,n*2);
    else update(wz,num,n*2+1);
}
int getsum(int l,int r,int n)
{
    int mid=(f[n].l+f[n].r)/2;
    if(f[n].l==l&&f[n].r==r)
        return f[n].sum;
    else if(mid>=r)
        getsum(l,r,n*2);
    else if(mid<l)
        getsum(l,r,n*2+1);
    else 
    {
        return getsum(l,mid,n*2)+getsum(mid+1,r,n*2+1);
    }
}
int main()
{
    int i,k,n,sum,s;
    while(scanf("%d",&n)>0)
    {
        for(i=1;i<=n;i++)
            date[i]=0;
        build(1,n,1);
        sum=0;
        for(i=1;i<=n;i++)
        {
            scanf("%d",&date[i]);
            date[i]++;
        }
        for(i=1;i<=n;i++)
        {
            update(date[i],1,1);
            k=getsum(1,date[i],1);
            sum=sum+i-k;
        }
        s=sum;
        for(i=1;i<=n;i++)
        {
        //    printf("%d ",s);
            k=n-date[i]-(date[i]-1);
            s=s+k;
            if(s<sum)
                sum=s;    
        }
        printf("%d\n",sum);
    }
    return 0;
}

树状数组的代码

#include<stdio.h>
int a[5003],d[5003];
int lowbit(int x)
{
    return x&(-x);
}
int sum(int x,int *p)
{
    int k=0;
    while(x)
    {
        k=k+p[x];
        x=x-lowbit(x);
    }
    return k;
}
void add(int x,int n,int num,int *p)
{
    int i;
    for(i=x;i<=n;i=i+lowbit(i))
        p[i]=p[i]+num;
}
int main()
{
    int i,j,k,n,m,cz,k1,k2,min;
    while(scanf("%d",&n)>0)
    {
        for(i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            a[i]++;
            d[i]=0;
        }
        for(i=1,cz=0;i<=n;i++)
        {
            add(a[i],n,1,d);
            k1=i-sum(a[i],d);
            cz=k1+cz;
        }
    //    printf("%d\n",cz);
        for(i=1,k1=0,min=cz;i<n;i++)
        {
            k1=n-a[i];
            k2=-a[i]+1;
            min=k1+k2+min;
        //    printf("%d ",k1+k2);
            if(min<cz)
                cz=min;
        }
        printf("%d\n",cz);
    }
    return 0;
}