hdu 2838 Cow Sorting

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

你可能感兴趣的:(线段树/树状数组)