hdu 2838 Cow Sorting

来源：互联网发布：大数据分析书籍推荐编辑：程序博客网时间：2024/05/02 04:42

题目链接: 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;}