逆序对（树状数组OR归并排序）

最新推荐文章于 2021-12-08 18:59:21 发布

原创最新推荐文章于 2021-12-08 18:59:21 发布 · 164 阅读

0 ·

CC 4.0 BY-SA版权

本文提供了解决POJ 2299问题的两种方法：使用树状数组和归并排序。树状数组方法通过快速读取优化了查询效率，而归并排序方法则通过分治策略计算逆序对数量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接：http://poj.org/problem?id=2299

这题用了快读，poj上G++交一直WA，C++尼玛居然就对了，气死我了。

树状数组：

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <cctype>
#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <queue>
#include <map>
#include <set>
#include <sstream>
#include<iomanip>
using namespace std;
typedef long long ll;
#define REW(a,b) memset(a,b,sizeof(a))
#define inf int(0x3f3f3f3f)
#define mod int(1e9+7)
#define P pair<int,int>
#define pi acos(-1)
inline int read()
{
    int X=0,w=0; char ch=0;
    while(!isdigit(ch)) {w|=ch=='-';ch=getchar();}
    while(isdigit(ch)) X=(X<<3)+(X<<1)+(ch^48),ch=getchar();
    return w?-X:X;
}
int n,a[500008],qw[500008];
struct as{
int q,w;}b[500008];
bool cmp(as a,as b)
{
	return a.q<b.q;
}
int sum(int x)
{
    int s=0;
    while(x>0)
    {
        s+=a[x];
        x-=x&-x;
    }
    return s;
}
void add(int x,int y)
{
    while(x<=n)
    {
        a[x]+=y;
        x+=x&-x;
    }
    return;
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	while(cin>>n&&n)
    {
        REW(a,0);
        REW(qw,0);
        int e=n;
        ll s=0;
        for(int i=1;i<=n;i++)
		{
			b[i].q=read();
			b[i].w=i;
		}
		sort(b+1,b+n+1,cmp);
		for(int i=1;i<=n;i++)
		{
			qw[b[i].w]=i;
		}
		for(int i=1;i<=n;i++)
		{
			add(qw[i],1);
			s+=i-sum(qw[i]);
		}
		cout<<s<<endl;
    }
    return 0;
}

归并排序：

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <cctype>
#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <queue>
#include <map>
#include <set>
#include <sstream>
#include<iomanip>;
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const ll inff = 0x3f3f3f3f3f3f3f3f;
#define FOR(i,a,b) for(int i(a);i<=(b);++i)
#define FOL(i,a,b) for(int i(a);i>=(b);--i)
#define REW(a,b) memset(a,b,sizeof(a))
#define inf int(0x3f3f3f3f)
#define si(a) scanf("%d",&a)
#define sl(a) scanf("%lld",&a)
#define sd(a) scanf("%lf",&a)
#define ss(a) scanf("%s",a)
#define mod ll(1e9+7)
#define pb push_back
#define eps 1e-6
#define lc d<<1
#define rc d<<1|1
#define Pll pair<ll,ll>
#define P pair<int,int>
#define pi acos(-1)
#define VI vector<vector<int>>
using namespace std;
const int N=5e5+8;
int a[N],b[N],n;
ll ans;
void getsort(int l,int r)
{
    if(l>=r) return;
    int mid=(l+r)>>1;
    getsort(l,mid);
    getsort(mid+1,r);
    int x=l,y=mid+1,num=0;
    while(x<=mid&&y<=r)
    {
        if(a[x]<=a[y]) b[++num]=a[x++];
        else b[++num]=a[y++],ans+=mid-x+1;
    }
    while(x<=mid) b[++num]=a[x++];
    while(y<=r) b[++num]=a[y++];
    FOR(i,1,num) a[l+i-1]=b[i];
}
int main()
{
    cin.tie(0);
    cout.tie(0);
    while(cin>>n&&n)
    {
        ans=0;
        FOR(i,1,n) si(a[i]);
        getsort(1,n);
        cout<<ans<<endl;
    }
    return 0;
}
/*

*/