首页 > 代码库 > POJ 2299 Ultra-QuickSort
POJ 2299 Ultra-QuickSort
求逆序对个数。
暴力 n^2 TLE妥妥的。要么 归并排序的时候统计,要么线段树或者数状数组优化。
时间复杂度都是 n*logn
线段树求逆序数怎么写呢。
例如样例的
9 1 0 5 4
其位置pos 为
0 1 2 3 4
再来一个数组设为
0 0 0 0 0
线段树先插入 (9,0),把插入的位置 0 变成 1。
1 0 0 0 0
然后统计 left=0,right=(0-1) 由于right比left 小,不进行查询。
再插入(5,3) ,把插入的位置 0 变成 1。
1 0 0 1 0
然后统计 left=0,right=(3-1) 对线段树进行查询,查询 0~2 的和。
再插入 (4,4)………
再插入 (1,1)………
再插入 (0,2)………
这样就分别求出 这些数左边比他们大的数。加起来就是这个数列逆序数的总数。
线段树的 插入和查询 都是 logn的时间复杂度。总共n次。为n*logn。
不过注意的是 题意中说了最后答案很大,所以统计的时候需要用long long 加起来。
PS:我就WA了一发。
#include<cstdio> #include<cstring> #include<string> #include<queue> #include<algorithm> #include<map> #include<stack> #include<iostream> #include<list> #include<set> #include<bitset> #include<vector> #include<cmath> #define INF 0x7fffffff #define eps 1e-8 #define LL long long #define PI 3.141592654 #define CLR(a,b) memset(a,b,sizeof(a)) #define FOR(i,a,b) for(int i=a;i<b;i++) #define FOR_(i,a,b) for(int i=a;i>=b;i--) #define pub push_back #define puf push_front #define pob pop_back #define pof pop_front #define mp make_pair #define ft first #define sd second #define sf scanf #define pf printf #define sz(v) ((int)(v).size()) #define all(v) (v).begin(),(v).end() #define acfun std::ios::sync_with_stdio(false) #define SIZE 3000000 +1 #define MOD 1000000007 using namespace std; struct lx { int a,p; bool friend operator < (lx a,lx b) { return a.a>b.a; } }; lx l[SIZE]; int n; int t[SIZE]; int site; void update(int o,int l,int r) { if(l==r)t[o]=1; else { int m=(l+r)>>1; if(site<=m)update(o*2,l,m); else update(o*2+1,m+1,r); t[o]=t[o*2]+t[o*2+1]; } } int ql,qr; int query(int o,int l,int r) { if(l>=ql&&r<=qr)return t[o]; int m=(l+r)>>1; int ans=0; if(ql<=m)ans+=query(o*2,l,m); if(qr>m)ans+=query(o*2+1,m+1,r); return ans; } int main() { while(sf("%d",&n),n) { FOR(i,1,n+1) { sf("%d",&l[i].a); l[i].p=i; } sort(l+1,l+n+1); CLR(t,0); LL ans=0; //fuck long long FOR(i,1,n+1) { site=l[i].p; update(1,1,n); ql=1,qr=site-1; if(ql<=qr) ans+=query(1,1,n); //pf("%lld==\n",ans); } pf("%lld\n",ans); } }
POJ 2299 Ultra-QuickSort
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。