【Codeforces Beta Round #57 (Div. 2) E. Enemy is weak】

博客围绕 Codeforces 61E 问题展开,问题是求满足 i > j > k 且 arr[i] > arr[j] > arr[k] 的元素对数量。通过枚举 j,利用树状数组分别维护其左边比它大的元素数量和右边比它小的元素数量,将二者相乘得到结果。

Codeforces 61E
题意 就是问你有多少对 i > j > k 使得 arr[i] > arr[j] > arr[k]
那么我们枚举 j
左边有多少个比他大的 用树状数组维护
右边有多少个比他小的 用树状数组维护
乘起来即可

/*
    if you can't see the repay
    Why not just work step by step
    rubbish is relaxed
    to ljq
*/
#include <cstdio>
#include <cstring>
#include <iostream>
#include <queue>
#include <cmath>
#include <map>
#include <stack>
#include <set>
#include <sstream>
#include <vector>
#include <stdlib.h>
#include <algorithm>
using namespace std;

#define dbg(x) cout<<#x<<" = "<< (x)<< endl
#define dbg2(x1,x2) cout<<#x1<<" = "<<x1<<" "<<#x2<<" = "<<x2<<endl
#define dbg3(x1,x2,x3) cout<<#x1<<" = "<<x1<<" "<<#x2<<" = "<<x2<<" "<<#x3<<" = "<<x3<<endl
#define max3(a,b,c) max(a,max(b,c))
#define min3(a,b,c) min(a,min(b,c))

typedef pair<int,int> pll;
typedef long long ll;
const int inf = 0x3f3f3f3f;
const int _inf = 0xc0c0c0c0;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const ll _INF = 0xc0c0c0c0c0c0c0c0;
const ll mod =  (int)1e9+7;

ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
ll ksm(ll a,ll b,ll mod){int ans=1;while(b){if(b&1) ans=(ans*a)%mod;a=(a*a)%mod;b>>=1;}return ans;}
ll inv2(ll a,ll mod){return ksm(a,mod-2,mod);}
void exgcd(ll a,ll b,ll &x,ll &y,ll &d){if(!b) {d = a;x = 1;y=0;}else{exgcd(b,a%b,y,x,d);y-=x*(a/b);}}//printf("%lld*a + %lld*b = %lld\n", x, y, d);

/*namespace sgt
{
    #define lc (rt<<1)
    #define rc (rt<<11)
    #define mid ((l+r)>>1)

    #undef lc
    #undef rc
    #undef mid
}*/
const int MAX_N = 1000025;
int C[MAX_N<<1],C_[MAX_N<<1],arr[MAX_N],b[MAX_N];
void add(int x,int v)
{
    for(x;x<MAX_N;x+=x&(-x))
        C[x]+=v;
}
int getsum(int x)
{
    int res = 0;
    for(;x;x-=x&(-x))
        res+=C[x];
    return res;
}
void add_(int x,int v)
{
    for(x;x<MAX_N;x+=x&(-x))
        C_[x]+=v;
}
int getsum_(int x)
{
    int res = 0;
    for(;x;x-=x&(-x))
        res+=C_[x];
    return res;
}
int main()
{
    //ios::sync_with_stdio(false);
    //freopen("a.txt","r",stdin);
    //freopen("b.txt","w",stdout);
    int n;
    ll ans = 0;
    scanf("%d",&n);
    for(int i = 1;i<=n;++i) scanf("%d",&arr[i]),b[i] = arr[i];
    sort(b+1,b+1+n);
    int sz = unique(b+1,b+1+n)-b-1;
    for(int i = 1;i<=n;++i) arr[i] = lower_bound(b+1,b+1+sz,arr[i])-b,add_(arr[i],1);
    for(int i = 1;i<=n;++i)
    {
        add_(arr[i],-1);
        int tmp = i-1-getsum(arr[i]),tmp_ = getsum_(arr[i]);
        ans+=1ll*tmp*tmp_;
        add(arr[i],1);
    }
    printf("%lld\n",ans);
    //fclose(stdin);
    //fclose(stdout);
    //cout << "time: " << (long long)clock() * 1000 / CLOCKS_PER_SEC << " ms" << endl;
    return 0;
}

评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值