当前位置: 移动技术网 > 移动技术>移动开发>IOS > 树状数组

树状数组

2020年07月18日  | 移动技术网移动技术  | 我要评论


先离散化处理在利用树状数组求一个数前面比他小的。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<iostream>
#include<iomanip>
#include<algorithm>
#include<stack>
#include<queue>
#include<cstdlib>
#include<string>
#include<vector>
#include<list>
#include<map>
#include<set>
#define fi first
#define se second
#define pb push_back
#define me(a,b) memset(a,b,sizeof(a))
#define INIT() std::ios::sync_with_stdio(false)
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int>P;
const int MAX_N=500000+5;
const int MAX_M=100000+5;
const int INF=0x7fffffff;
const int inf=1000000000;
const double EPS=1e-6;
const ull base=123;
const ll mod=998244353;
const double pi=4*atan(1.0);
struct node
{
    int pos;
    int x;
    int cnt;
} a[MAX_N];
int n;
int bit[MAX_N];
bool cmp(node a,node b)
{
    return a.x<b.x;
}
bool cmp2(node a,node b)
{
    return a.pos<b.pos;
}
int sum(int x)
{
    int s=0;
    while(x>0)
    {
        s+=bit[x];
        x-=x&-x;
    }
    return s;
}
void add(int x,int val)
{
    while(x<=n)
    {
        bit[x]+=val;
        x+=x&-x;
    }
}
int main()
{
    while(cin>>n&&n)
    {
        me(bit,0);
        int i;
        for(i=1; i<=n; i++)
        {
            a[i].pos=i;
            scanf("%d",&a[i].x);
        }
        sort(a+1,a+1+n,cmp);
        int cnt=1;
        a[1].cnt=1;
        for(i=2; i<=n; i++)
        {
            a[i].cnt=i;
        }
        sort(a+1,a+1+n,cmp2);
        ll ans=0;
        for(i=1; i<=n; i++)
        {
            add(a[i].cnt,1);
            ans+=(ll)(i-sum(a[i].cnt));
        }
        cout<<ans<<endl;
    }
}

本文地址:https://blog.csdn.net/ljy121121/article/details/107387764

如对本文有疑问, 点击进行留言回复!!

相关文章:

验证码:
移动技术网