题目链接: http://acm.hdu.edu.cn/showproblem.php?pid=2838
题目大意:有n头(1<=n<=100,000)牛,每头牛有一个不同的rank(1<=rank<=100,000),牛的主人要按照rank给牛排序,并且只有相邻的牛能交换位置,且如果牛i和牛j交换,那么要话费的时间是rank[i]+rank[j],求最小花费的时间.
思路:因为相邻的牛才能交换,那么把rank小的先换到其最终位置就能保证时间最小,因为如果有个rank大的先别换到其最终位置但是还有一个rank小的在它后面,那么rank大的必然还要再交换.
用线段树或树状数组快速求需要交换的牛的rank和sum和需要交换的牛的头数cnt.
那么把牛x换到其最终位置的时间=sum[x]+cnt[x]*rank[x],换到最终位置后就把牛x所在的原位置清零.
代码:
#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <ctype.h>
#include <math.h>
#include <time.h>
#include <stack>
#include <queue>
#include <map>
#include <set>
#include <vector>
#include <string>
#include <iostream>
#include <algorithm>
using namespace std;
#define ull unsigned __int64
#define ll __int64
//#define ull unsigned long long
//#define ll long long
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define middle (l+r)>>1
#define mod 1000000
#define esp (1e-10)
const int INF=0x3F3F3F3F;
//const double pi=acos(-1.0);
const int N=100010;
int n,m,cnt[N];
ll sum[N];
struct node{
ll lv;
int pos;
bool operator < (const node& p) const {
return lv < p.lv;
}
}a[N];
int lowbit(int x){return x&(-x);}
void Update(int x,ll c1,int c2){
while(x<=n) sum[x]+=c1,cnt[x]+=c2,x+=lowbit(x);
}
void Query(int x,ll& r1,int& r2){
r1=r2=0;
while(x>0) r1+=sum[x],r2+=cnt[x],x-=lowbit(x);
}
int main(){
//freopen("1.in","r",stdin);
//freopen("1.out","w",stdout);
int i,j,k,r2;
ll ret,r1;
char op[2];
//int T,cas;scanf("%d",&T);for(cas=1;cas<=T;cas++)
while(~scanf("%d",&n)){
for(i=1;i<=n;i++){
scanf("%I64d",&a[i].lv),a[i].pos=i;
Update(i,a[i].lv,1);
}
sort(a+1,a+n+1);
for(i=1,ret=0;i<=n;i++){
Update(a[i].pos,-a[i].lv,-1);
Query(a[i].pos,r1,r2);
ret+=r1+r2*a[i].lv;
}
printf("%I64d\n",ret);
}
return 0;
}