第五届蓝桥杯——小朋友排队

小朋友排队
n 个小朋友站成一排。现在要把他们按身高从低到高的顺序排列，但是每次只能交换位置相邻的两个小朋友。
每个小朋友都有一个不高兴的程度。开始的时候，所有小朋友的不高兴程度都是0。
如果某个小朋友第一次被要求交换，则他的不高兴程度增加1，如果第二次要求他交换，则他的不高兴程度增加2（即不高兴程度为3），依次类推。当要求某个小朋友第k次交换时，他的不高兴程度增加k。
请问，要让所有小朋友按从低到高排队，他们的不高兴程度之和最小是多少。
如果有两个小朋友身高一样，则他们谁站在谁前面是没有关系的。
【数据格式】
输入的第一行包含一个整数n，表示小朋友的个数。
第二行包含 n 个整数 H1 H2 … Hn，分别表示每个小朋友的身高。
输出一行，包含一个整数，表示小朋友的不高兴程度和的最小值。
例如，输入：
3
3 2 1
程序应该输出：
9
【样例说明】
首先交换身高为3和2的小朋友，再交换身高为3和1的小朋友，再交换身高为2和1的小朋友，每个小朋友的不高兴程度都是3，总和为9。
【数据规模与约定】
    对于10%的数据， 1<=n<=10；
    对于30%的数据， 1<=n<=1000；
    对于50%的数据， 1<=n<=10000；
    对于100%的数据，1<=n<=100000，0<=Hi<=1000000。
资源约定：
峰值内存消耗 < 256M
CPU消耗  < 1000ms
请严格按要求输出，不要画蛇添足地打印类似：“请您输入...” 的多余内容。
所有代码放在同一个源文件中，调试通过后，拷贝提交该源码。
注意: main函数需要返回0
注意: 只使用ANSI C/ANSI C++ 标准，不要调用依赖于编译环境或操作系统的特殊函数。
注意: 所有依赖的函数必须明确地在源文件中 #include <xxx>， 不能通过工程设置而省略常用头文件。

提交时，注意选择所期望的编译器类型。

正反两次求每个小朋友的逆序数，再查表求和

树状数组优化加速

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <cmath>
#include <algorithm>
#include <vector>
#include <map>
#include <string>
#include <stack>
using namespace std;
typedef long long ll;
#define PI 3.1415926535897932
#define E 2.718281828459045
#define INF 0x3f3f3f3f
#define mod 1000000007


const int M=1005;
int n,m;
int cnt;
int sx,sy,sz;
int mp[M][M];
int pa[M*10],rankk[M];
int head[M*6],vis[M*10];
int dis[M][10];
ll prime[M*1000];
bool isprime[M*1000];
int lowcost[M],closet[M];
char st1[5050],st2[5050];
int len[M*6];
typedef pair<int ,int> ac;
vector<int> g[M*10];
int dp[M];
int sums[M*10];
int has[10500],has1[10500];
int month[13]= {0,31,59,90,120,151,181,212,243,273,304,334,0};
int dir[8][2]= {{0,1},{0,-1},{-1,0},{1,0},{1,1},{1,-1},{-1,1},{-1,-1}};


void getpri()
{
    ll i;
    int j;
    cnt=0;
    memset(isprime,false,sizeof(isprime));
    for(i=2; i<1000000LL; i++)
    {
        if(!isprime[i])prime[cnt++]=i;
        for(j=0; j<cnt&&prime[j]*i<1000000LL; j++)
        {
            isprime[i*prime[j]]=1;
            if(i%prime[j]==0)break;
        }
    }
}
ll qk_mul(ll a,ll b)
{
    ll t=0;
    while(b)
    {
        if(b&1)
            t=(t+a)%mod;
        a=(a<<1)%mod;
        b>>=1;
    }
    t%=mod;
    return t;
}
ll qk_mod(ll a,ll b)
{
    ll ans=1;
    while(b)
    {
        if(b&1) ans=qk_mul(ans,a);
        a=qk_mul(a,a);
        b>>=1;
    }
    ans%=mod;
    return ans;
}
int gcd(int a,int b)
{
    return b == 0 ? a : gcd(b, a%b);
}
int lcm(int a,int b)
{
    return a*b/gcd(a,b);
}
int findx(int t)
{
    if(t!=pa[t])
        pa[t]=findx(pa[t]);
    return pa[t];
}
void unionx(int x,int y)
{
    x=findx(x);
    y=findx(y);
    if(x!=y)
    {
        pa[y]=x;
    }
}


void init()
{
    for(int i=0; i<101; i++)
        pa[i]=i;
}
int heap[100005];
void push(int x)
{
    int i=++sz;
    while(i>1)  //i>0
    {
        int p=i/2; //(i-1)/2
        if(heap[p]<=x)break;
        heap[i]=heap[p];
        i=p;
    }
    heap[i]=x;
    /* a[++sz] = x;
     int t = a[sz];
     int tson = sz;
     while( (tson > 1)&&( a[tson/2] > t))
     {
         a[tson] = a[tson/2];
         tson = tson/2;
     }
     a[tson] = x;
    */
}
int pop()
{
    int ret=heap[1];//int ret=a[0]
    int x=heap[sz--];//a[--sz]
    int i=1;// i=0
    while(2*i<sz) //2*i+1
    {
        int a=i*2,b=i*2+1;
        if(b<sz&&heap[b]<heap[a])a=b;
        if(heap[a]>=x)break;
        heap[i]=heap[a];
        i=a;
    }
    heap[i]=x;
    return ret;
}
int lowbit(int i)
{
    return i&(-i);
}
void add(int i,int v)//修改值并向父节点修改
{
    while(i<=M)
    {
        sums[i]+=v; //c[i]开始都是0，每经过一个数，它的c[i]+1表示c[i]
        i+=lowbit(i);
    }
}
int sum(int i)//求和
{
    ll s=0;
    while(i>=1)
    {
        s+=sums[i];
        i-=lowbit(i);
    }
    return s;
}


char pre[7][7]=
{
    {'>','>','<','<','<','>','>'},
    {'>','>','<','<','<','>','>'},
    {'>','>','>','>','<','>','>'},
    {'>','>','>','>','<','>','>'},
    {'<','<','<','<','<','=','0'},
    {'>','>','>','>','0','>','>'},
    {'<','<','<','<','<','0','='}
};
int num[M*10],s[M*10],op[M*10],num2[M*10];
int main()
{
    int i,j;
    for(i=1;i<=M;i++){
            s[i]=s[i-1]+i; //表示该位置互换i次所需累加的愤怒值
           // printf("%d ",s[i]);
    }
    scanf("%d",&n);
    for(i=0;i<n;i++)
    {
        scanf("%d",&num[i]);
        add(num[i]+1,1);//num[i]+1往后标记，表示大于num[i]的数
        op[i]=i-sum(num[i]);//i-前i中小于num[i]的个数=前i中大于num[i]的个数
    }
    int ans=0;
    memset(sums,0,sizeof(sums));
    for(i=n-1;i>=0;i--)
    {
        add(num[i]+1,1);
        op[i]+=sum(num[i]);
        ans+=s[op[i]];
       // printf("%d ",s[i]);
    }
   // printf("\n");
    printf("%d\n",ans);
    return 0;
}

附树状数组求逆序数代码

http://blog.csdn.net/cattycat/article/details/5640838

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


const int N = 500005;


struct Node
{
int val;
int pos;
};


Node node[N];
int c[N], reflect[N], n;


bool cmp(const Node& a, const Node& b)
{
return a.val < b.val;
}


int lowbit(int x)
{
return x & (-x);
}


void update(int x)
{
while (x <= n)
{
c[x] += 1;
x += lowbit(x);
}
}


int getsum(int x)
{
int sum = 0;
while (x > 0)
{
sum += c[x];
x -= lowbit(x);
}
return sum;
}


int main()
{
while (scanf("%d", &n) != EOF && n)
{
for (int i = 1; i <= n; ++i) 
{
scanf("%d", &node[i].val);
node[i].pos = i;
}
sort(node + 1, node + n + 1, cmp);   //排序
for (int i = 1; i <= n; ++i) reflect[node[i].pos] = i;   //离散化
for (int i = 1; i <= n; ++i) c[i] = 0;   //初始化树状数组
long long ans = 0;
for (int i = 1; i <= n; ++i)
{
update(reflect[i]);
ans += i - getsum(reflect[i]);
}
printf("%lld\n", ans);
} 
return 0;
}