码迷,mamicode.com
首页 > 编程语言 > 详细

codeforeces 540E(树状数组

时间:2015-07-25 07:05:55      阅读:121      评论:0      收藏:0      [点我收藏+]

标签:

题意:交换自然数中的若干对数,求交换后总共有多少逆序数对。

思路:因为题目数字范围比较大,不能直接用树状数组算,首先要离散化。然后一种算法是官方题解中根据逆序对数是否属于交换过的数分类讨论统计。我的算法是把没有交换的连续的数看成一个数,使用树状数组统计的时候直接加上这个区间的数字个数,这样就不需要繁琐(?)的讨论了。

技术分享
#include<iostream>
#include<map>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<vector>
#include<queue>
#include<stack>
#include<functional>
#include<set>
#define pb push_back
#define fs first
#define se second
using namespace std;
typedef long long ll;
typedef pair<ll,ll> P;
const int maxv=1e5*4;
struct BIT{
    ll a[maxv];
    void add(int p,int x){
        while(p<maxv){
            a[p]+=x;
            p+=p&-p;
        }
    }
    ll sum(int p){
        ll ans=0;
        while(p>0){
            ans+=a[p];
            p-=p&-p;
        }
        return ans;
    }
    ll after(int a){
        return sum(maxv-1)-sum(a);
    }
}B;
int a[maxv];
int b[maxv];
int c[maxv];
int mid[maxv];
int sq[maxv];
int n;
map<int,int> mii;
int main(){
    freopen("/home/files/CppFiles/in","r",stdin);
    cin>>n;
    for(int i=0;i<n;i++){
        scanf("%d%d",a+i,b+i);
        c[i*2]=a[i];
        c[i*2+1]=b[i];
    }
    sort(c,c+2*n);
    int h=unique(c,c+2*n)-c;
    for(int i=1;i<h;i++){
        mid[i]=c[i]-c[i-1]-1;
    }
    for(int i=0;i<h;i++){
        mii[c[i]]=i+1;
    }
    for(int i=0;i<maxv;i++){
        sq[i]=i;
    }
    for(int i=0;i<n;i++){
        swap(sq[mii[a[i]]],sq[mii[b[i]]]);
    }
    ll ans=0;
    for(int i=1;i<=h;i++){
        ll now=sq[i];
        ans+=B.after(now*2-1);
        B.add(now*2-1,1);
        ans+=mid[i]*B.after(i*2);
        B.add(2*i,mid[i]);
    }
    cout<<ans<<endl;
    return 0;
}
View Code

 

codeforeces 540E(树状数组

标签:

原文地址:http://www.cnblogs.com/Cw-trip/p/4675146.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!