HDU 5273 Dylans loves sequence

点击打开链接

有一个长度为1000的数组，有10^5个询问一段区间[L,R]之内有多少个逆序对。

因为数组长度为1000，所以可以把结果保存下来然后O(1)查询，求逆序对有很多方法，树状数组线段树或者用分治归并。

第一次用线段树写逆序对。。。

线段树求逆序对的方法就是将数组降序排序，相同的元素按下标降序排序，排好序之后，再按照顺序插入线段树并标记，线段树的叶子节点的位置代表元素原来的位置，统计在当前元素之前已经插了有多少元素，因为是按降序排列的，所以之前插入的元素一定比当前元素大。这样求得的是当前元素之前一共有多少比它大的元素，记为a[i]，逆序对就是枚举区间段所有元素，求sum a[i]。

所以只要枚举端点即可，总复杂度是O(n^2logn)。

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn = 1010;
#define ls rt << 1
#define rs rt << 1 | 1
#define lson l,mid,ls
#define rson mid+1,r,rs

struct node{
    int val;
    int pos;
}num[maxn];

bool cmp(node a,node b){
    if(a.val == b.val) return a.pos > b.pos;
    return a.val > b.val;
}

int ans[maxn][maxn];

struct no{
    int sum;
}Tree[maxn<<2];

void build(){
    memset(Tree,0,sizeof(Tree));
}

void pushup(int rt){
    Tree[rt].sum = Tree[ls].sum +Tree[rs].sum;
}

void update(int x,int l,int r,int rt){
     if(l == r && l == x){
        Tree[rt].sum = 1;
        return;
     }
     int mid = (l + r)>>1;
     if(x <= mid) update(x,lson);
     else update(x,rson);
     pushup(rt);
}

int query(int L,int R,int l,int r,int rt){
     if(L > R) return 0;
     if(L <= l&&R >= r){
        return Tree[rt].sum;
     }
     int mid = (l + r) >> 1,ret = 0;
     if(L <= mid) ret += query(L,R,lson);
     if(R >= mid+1) ret += query(L,R,rson);
     return ret;
}


int main()
{
    int n,q,tmp[maxn];
    node a[maxn];
    cin>>n>>q;
    for(int i = 1;i <= n;i++){
        scanf("%d",&num[i].val);
        num[i].pos = i;
    }
    memset(ans,0,sizeof(ans));
    for(int i = 1;i <= n;i++){//枚举开头
        for(int j = i;j <= n;j++) a[j] = num[j];
        sort(a+i,a+n+1,cmp);            //对当前段排序
        for(int j = i;j <= n;j++){
            update(a[j].pos,1,n,1);
            tmp[a[j].pos] = query(i,a[j].pos - 1,1,n,1);//tmp保存的是当前元素前面有多少个比他大的元素
        }
        for(int j = i;j <= n;j++){
            ans[i][j] = ans[i][j-1] + tmp[j];
        }
        build();
    }
    while(q--){
        int l,r;
        scanf("%d%d",&l,&r);
        cout<<ans[l][r]<<endl;
    }
    return 0;
}

其实仔细想想还有另外一种做法，设dp[i][j]为从i到j比i大的元素个数，那么有dp[i][j] =dp[i+1][j] + num[i]>num[j]?1:0 这个过程是O(n^2)的，设ans[i][j]为i到j的逆序对，那么它满足ans[i][j] = ans[i][j-1] + dp[i][j];

这样整个过程就是O(n^2)的。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;
const int maxn = 1010;

int ans[maxn][maxn],k[maxn][maxn];
int num[maxn];

int main()
{
    int n,q;
    cin>>n>>q;
    for(int i = 1;i <= n;i++) scanf("%d",&num[i]);
    memset(ans,0,sizeof(ans));
    for(int j = n;j >= 1;j--){
        for(int i = j; i >= 1;i--){
            ans[i][j] = ans[i+1][j];
            if(num[i] > num[j]) ans[i][j]++;
        }
    }
    for(int i = 1;i <= n;i++)
    for(int j = 1;j <= n;j++){
        k[i][j] = k[i][j-1] + ans[i][j];
    }
    while(q--){
        int l,r;
        scanf("%d%d",&l,&r);
        cout<<k[l][r]<<endl;
    }
    return 0;
}