hdu1394(線段樹)


鏈接:http://acm.hdu.edu.cn/showproblem.php?pid=1394

題意:就是給出一串數,當依次在將第一個數變為最后一個數的過程中,要你求它的最小逆序數。

思路:可以用樹狀數組和線段數做。這里我是用線段樹做的。建的是一棵空樹,然后每插入一個點之前,統計大於這個數的有多少個,直到所有的數都插入完成,就結果了逆序樹的統計。

要得出答案主要是利用了一個結論,如果是0到n的排列,那么如果把第一個數放到最后,對於這個數列,逆序數是減少a[i],而增加n-1-a[i]的.當然,你也可以是統計小於這個數的有多少個,然后再用已經插入樹中i個元素減去小於這個數的個數,得出的結果也是在一樣的。

反思:以前寫線段樹的時候,就一個簡單的更新求和要寫一百多行代碼,這次要感謝shiqi_614,在他那里學到了現在編寫線段樹的風格,謝謝。

#include<iostream>
using namespace std;
#define N 5005
struct 
{
	int l,r;
	int num;
}tree[4*N];
void creat(int i,int l,int r)
{
	int mid=(l+r)/2;
	tree[i].l=l;
	tree[i].r=r;
	tree[i].num=0;
	if(l==r)
	{
		return ;
	}
	creat(i*2,l,mid);
	creat(i*2+1,mid+1,r);
}
void updata(int i,int k)
{
	if(tree[i].l==k&&tree[i].r==k)
	{
		tree[i].num=1;
		return ;
	}
	int mid=(tree[i].l+tree[i].r)/2;
	if(k<=mid)
		updata(i*2,k);
	else
		updata(i*2+1,k);
	tree[i].num=tree[i*2].num+tree[i*2+1].num;
}
int getsum(int i,int k,int n)
{
	if(k<=tree[i].l&&tree[i].r<=n)
	{
		return tree[i].num;
	}
	else
	{
		int mid=(tree[i].l+tree[i].r)/2;
		int sum1=0,sum2=0;
		if(k<=mid)
			sum1=getsum(i*2,k,n);
		if(n>mid)
			sum2=getsum(i*2+1,k,n);
		return sum1+sum2;
	}
}
int main()
{
	int n;
	while(scanf("%d",&n)>0)
	{
		int a[N];
		creat(1,0,n-1);
		int i;
		int ans=0;
		for(i=0;i<n;i++)
		{
			scanf("%d",&a[i]);
			ans+=getsum(1,a[i]+1,n-1);
			updata(1,a[i]);
		}
		int minx=ans;
		for(i=0;i<n;i++)
		{
			ans=ans+n-2*a[i]-1;           //當a[i]由第一個變為最后一個時,要加上a[i]后面大於a[i]的數的個數,有n-1-a[i]個,要
			if(ans<minx)                    //減去a[i]后面小於a[i]的數的個數,有a[i]個(注意i是從0開始的)
				minx=ans;
		}
		printf("%d\n",minx);
	}
	return 0;
}

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM