HDU2492ping pang (线段树求逆序对)

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/liuqvanlu/article/details/78697183

题意:

一条大街上住着n个乒乓球爱好者,经常组织比赛切磋技术。每个人都有一个不同的技能值a[i]。每场比赛需要3个人,两名选手,一名裁判。他们有一个奇怪的规定,既裁判必须住在两名选手的中间,并且技能值也在两名选手直接。问一共能组织多少种比赛。 1<=T<=20, 3<=n<=20000, 1<=a[i]<=100000。

分析

假设c[i]为 a[1]到a[i-1]中比a[i]小的数量,d[i]为a[i+1]到a[n]中比a[i]小的数量。枚举2到n-1作为裁判,对于每个裁判i,都有c[i]*(n-i-d[i])+d[i]*(i-c[i]-1)种比赛方案。

用求逆序对的方法求出c和d的数组。而且因为a[i]<=100000,所以也不需要进行离散化。

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=20000+50;
const int maxa=100000+100;
int T,n,r;
int a[maxn],sumv[4*maxa] ;
int c[maxn],d[maxn];
int p,v;
void update(int o,int L,int R){
    if(L==R){
        sumv[o]+=v;
        return ;
    }
    int M=L+(R-L)/2;
    if(p<=M)update(2*o,L,M);
    if(p>M)update(2*o+1,M+1,R);
    sumv[o]=sumv[2*o+1]+sumv[2*o];
}
int ql,qr,_sum;
void query(int o,int L,int R){
    if(ql<=L&&qr>=R){
        _sum+=sumv[o];
        return ;
    }
    int M=L+(R-L)/2;
    if(ql<=M)query(2*o,L,M);
    if(qr>M)query(2*o+1,M+1,R);
}
int main(){
    scanf("%d",&T);
    for(int k=1;k<=T;k++){
        scanf("%d",&n);
        for(int i=1;i<=n;i++){
            cin>>a[i];
            r=max(r,a[i]);
        }
        memset(sumv,0,sizeof(sumv));
        for(int i=1;i<=n;i++){
            p=a[i];v=1;
            update(1,1,r);
            _sum=0;ql=1;qr=a[i];
            query(1,1,r);
            c[i]=_sum-1;
        }
       memset(sumv,0,sizeof(sumv));
       for(int i=n;i>=1;i--){
        p=a[i];v=1;
        update(1,1,r);
        _sum=0;ql=1;qr=a[i];
        query(1,1,r);
        d[i]=_sum-1;
       }
       long long ans=0;
       for(int i=2;i<n;i++){
          ans+=c[i]*(n-i-d[i])+(i-1-c[i])*d[i];
       }
       cout<<ans<<endl;
    }
return 0;
}


展开阅读全文

没有更多推荐了,返回首页