首页 > 代码库 > POJ 2299 Ultra-QuickSort(线段树+离散化)
POJ 2299 Ultra-QuickSort(线段树+离散化)
题目地址:POJ 2299
这题曾经用归并排序做过,线段树加上离散化也可以做。一般线段树的话会超时。
这题的数字最大到10^10次方,显然太大,但是可以利用下标,下标总共只有50w。可以从数字大的开始向树上加点,然后统计下标比它小即在它左边的数的个数。因为每加一个数的时候,比该数大的数已经加完了,这时候坐标在它左边的就是一对逆序数。
但是该题还有一个问题,就是数字重复的问题。这时候可以在排序的时候让下标大的在前面,这样就可以保证加点的时候下标比他小的数中不会出现重复的。
这题需要注意的是要用__int64。
代码如下:
#include <iostream> #include <cstdio> #include <string> #include <cstring> #include <stdlib.h> #include <math.h> #include <ctype.h> #include <queue> #include <map> #include <set> #include <algorithm> using namespace std; #define LL __int64 #define lson l, mid, rt<<1 #define rson mid+1, r, rt<<1|1 LL sum[2100000]; struct node { LL x, id; } fei[600000]; bool cmp(node x, node y) { if(x.x==y.x) return x.id>y.id; return x.x>y.x; } void PushUp(int rt) { sum[rt]=sum[rt<<1]+sum[rt<<1|1]; } void update(LL x, int l, int r, int rt) { if(l==r) { sum[rt]++; return ; } int mid=l+r>>1; if(x<=mid) update(x,lson); else update(x,rson); PushUp(rt); } LL query(int ll, int rr, int l, int r, int rt) { if(ll<=l&&rr>=r) { return sum[rt]; } LL ans=0; int mid=l+r>>1; if(ll<=mid) ans+=query(ll,rr,lson); if(rr>mid) ans+=query(ll,rr,rson); return ans; } int main() { int n, i, j; LL ans; while(scanf("%d",&n)!=EOF&&n) { memset(sum,0,sizeof(sum)); for(i=1; i<=n; i++) { scanf("%I64d",&fei[i].x); fei[i].id=i; } sort(fei+1,fei+n+1,cmp); ans=0; for(i=1; i<=n; i++) { ans+=query(1,fei[i].id,1,n,1); update(fei[i].id,1,n,1); } printf("%I64d\n",ans); } return 0; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。