天天看點

hdu 2838 Cow Sorting (樹狀數組+逆序對)

題目

題意:給你N個排列不規則的數,任務是把它從小到大排好,每次隻能交換相鄰兩個數,交換一次的代價為兩數之和,求最小代價

拿到這道題,我根本看不出這道題和樹狀數組有半毛錢關系,部落格之,全說用樹狀數組做,納尼。。。看來我還是太年輕。。

這道題還涉及到了逆序對,何為逆序對:對于一個包含N個非負整數的數組A[1..n],如果有i < j,且A[ i ]>A[ j ],則稱(A[ i] ,A[ j] )為數組A中的一個逆序對。

放到這道題中,如何求代價即:這有N個數,第i個數的代價  =   在i前面大于 i 的數字個數 *  i  + 在 i 前面大于 i 的數字之和。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define N 100001
int n;
struct node
{
	int cnt;
	__int64 sum;
}c[N];
int lowbit(int x)
{
	return x&(-x);
}
void update(int x,int s)
{
	while(x<=n)
	{
		c[x].cnt+=1;
		c[x].sum+=s;
		x+=lowbit(x);
	}
}
int getsum_cnt(int x)
{
	int ans=0;
	while(x>0)
	{
		ans+=c[x].cnt;
		x-=lowbit(x);
	}
	return ans;
}
__int64 getsum_sum(int x)
{
	__int64 ans=0;
	while(x>0)
	{
		ans+=c[x].sum;
		x-=lowbit(x);
	}
	return ans;
}
int main(void)
{
	while(~scanf("%d",&n))
	{
		int i;
		__int64 ans=0;
		memset(c,0,sizeof(c));
		for(i=1;i<=n;i++)
		{
			int x;
			scanf("%d",&x);
			update(x,x);
			__int64 k1=i-getsum_cnt(x);
			//getsum_cnt(x)是小于等于x的數字的個數,i-getsum_cnt(x)就是大于x的數字的個數
			if(k1!=0)
			{
				__int64 k2=getsum_sum(n)-getsum_sum(x);//[x+1,n]區間的數字值和,即[1,n]-[1,x]
				ans=ans+k1*x+k2;
			}

		}
		printf("%I64d\n",ans);
	}
}
           

樹狀數組求逆序對個數 

#include <bits/stdc++.h>
using namespace std;
const int Max = 1e3;
int c[Max],a[Max];
int n,t;
int Lowbit(int x)
{
    return x&(-x);
}
void update(int i,int x)
{
    while(i<=n)
    {
        c[i]+=x;
        i+=Lowbit(i);
    }
}
int sum(int x)
{
    int sum=0;
    while(x>0)
    {
        sum+=c[x];
        x-=Lowbit(x);
    }
    return sum;
}
int Getsum(int x1,int x2)
{
    return sum(x2)-sum(x1);
}
int main()
{
    int ans = 0;
    int x;cin>>n;
    memset(c,0,sizeof(c));
    for(int i=1;i<=n;i++)
    {
        cin>>x;
        update(x,1);
        int  k = i-sum(x);
        //sum(x)是[1,i]中,小于等于x的數字的個數,
        //則 i-sum(x)就是大于x的數字的個數
       // cout<<"k="<<k<<endl;
        ans+=k;
    }
    cout<<ans<<endl;
    return 0;
}