这类问题就是求一段序列中的逆序数::
逆序数就是对于一个序列:
1 3 4 2
假设正序:1 2 3 4
那么原序列的逆序数为2
仔细观察,发现这个就是算冒泡排序交换相邻元素的次数。
但是这个算法复杂度为O(n^2)吗,一看数据范围,要tle的。
求逆序数的方法有::
线段树||树状数组||归并
具体分析一下,这个题要求的是这按照x降序排序后序列y的逆序数。
线段树写法:
#include <iostream>
#include <stdio.h>
#include <math.h>
#include <algorithm>
#include <queue>
#include <stack>
#include <set>
#include <string.h>
#include <map>
#include <stdlib.h>
#include <time.h>
#include <vector>
#define MAX_N 1e7
#define INF 0x3f3f3f3f
#define FAST ios::sync_with_stdio(false)
#define ll long long
//#define int long long
#define endl '\n'
using namespace std;
const int maxn=2e5+7;
struct node{
int l,r;
int sum;
}tr[maxn*4];
void pushup(int u){
tr[u].sum=tr[u<<1].sum+tr[u<<1|1].sum;
}
void build(int u,int l,int r){
node th;
th.l=l;
th.r=r;
th.sum=0;
tr[u]=th;
if(l==r){
tr[u].sum=0;
return;//默认节点都为0
}
int mid=(l+r)>>1;
build(u<<1,l,mid);
build(u<<1|1,mid+1,r);
}
int query(int u,int l0,int r0){
if(tr[u].l>=l0&&tr[u].r<=r0){
return tr[u].sum;
}
int mid=(tr[u].l+tr[u].r)>>1;
int sum=0;
if(l0<=mid)
sum+=query(u<<1,l0,r0);
if(r0>=mid+1)
sum+=query(u<<1|1,l0,r0);
return sum;
}
void modify(int u,int x){
//把区间为x的节点修改为1
if(tr[u].l==tr[u].r&&tr[u].l==x){
tr[u].sum=1;
return;
}
else{
int mid=(tr[u].l+tr[u].r)>>1;
if(x<=mid){
modify(u<<1,x);
}
else{
modify(u<<1|1,x);
}
pushup(u);
}
}
struct sold{
int x,y;
}a[maxn];
bool cmp(sold a,sold b){
return a.x>b.x;
}
int main(){
int n;
cin>>n;
memset(tr,0,sizeof tr);
build(1,0,200000);
for(int i=1;i<=n;i++){
cin>>a[i].x>>a[i].y;
}
sort(a+1,a+1+n,cmp);
ll ans=0;
for(int i=1;i<=n;i++){
ans+=query(1,0,a[i].y);//查询每次前缀和
modify(1,a[i].y);
}
cout<<ans<<endl;
}
树状数组写法:
#include <iostream>
#include <stdio.h>
#include <math.h>
#include <algorithm>
#include <queue>
#include <stack>
#include <set>
#include <string.h>
#include <map>
#include <stdlib.h>
#include <time.h>
#include <vector>
#define INF 0x3f3f3f3f
#define ll long long
using namespace std;
const int maxn=2e5+10;
int n;
int tr[maxn];
struct soldier{
int x,y;
}a[maxn];
int lowbit(int x){
return x&-x;//查询x二进制表示下最低位的1
}
void add(int x){
for(int i=x;i<=n;i+=lowbit(i)){
tr[i]++;//更新树状数组
}
}
int sum(int x){
int res=0;
for(int i=x;i>=1;i-=lowbit(i)){
res+=tr[i];//树状数组求前缀和
}
return res;
}
bool cmp(soldier a,soldier b){
return a.x>b.x;
};
int main(){
cin>>n;
for(int i=1;i<=n;i++){
cin>>a[i].x>>a[i].y;
}
memset(tr,0,sizeof tr);
sort(a+1,a+1+n,cmp);//按x排序,保证x合法
ll ans=0;
for(int i=1;i<=n;i++){
add(a[i].y);//查询小于当前y且x合法的数量
ans+=sum(a[i].y-1);//每次都把当前的y加入,相当于这个y所对应的战士的x对于所有之后加入战士的x都合法
}
cout<<ans<<endl;
}
归并写法: