【题】【逆序对&容斥原理】NKOJ 3722 叮叮猫

NKOJ 3722 叮叮猫

时间限制 : - MS 空间限制 : 265536 KB
评测说明 : 时限1000ms

问题描述
叮叮猫，学名蜻蜓，无脊椎动物。一般体型较大，翅膀长而窄，膜质，有清 晰的网状翅脉。
有个叮叮猫飞到了nodgd房间里，nodgd赶紧用高速照相机连拍了n张清晰 的照片，以此分析叮叮猫的飞行轨迹。因为nodgd是个好人，所以在把数据给 你之前已经进行了离散化处理。简单地说，叮叮猫在第i张照片里飞行高度是ai， 其中所有ai都是1~n之间的整数（不一定是1~n的排列）。 nodgd定义了一种“欲扬先抑”的状态：如果存在三个整数i,j,k，满足条件1 ≤ i< j< k≤n，且aj< ai< ak，就是欲扬先抑的。
现在要求统计出有多少个这样的欲扬先抑三元组（i,j,k）

输入格式
第一行一个整数n，表示照片数。
第二行n个整数，表示每张照片里叮叮猫的飞行高度。

输出格式
输出一行一个整数，表示欲扬先抑三元组的个数。

样例输入
5
2 1 1 4 4

样例输出
4

提示
【数据范围】
对于30%的数据，n≤500；
对于50%的数据，n≤5000；
对于100%的数据，1≤n≤500000。

思路：
题目所要求的状态为{2，1，3}（注意不是{2，1，2}QAQ），所以可以转化为求{x，x，3}-（{1，2，3}+{2，2，3}）
{x，x，3}的状态数可用“在当前数左边，比当前数小的数的数目”求出.
{1，2，3}+{2，2，3}的状态数则为“当前数a左边所有小于 当前数 的数b / 的左边不大于b的数 / 的总数目”
用两个树状数组维护

代码：

#include<cstdio>
#include<iostream>
using namespace std;
const int need=500004;
#define ll long long 

int n;
ll c1[need],c2[need];

inline void in_(int &d)
{
    bool mark=false;char t=getchar();
    while(t<'0'||t>'9'){if(t=='-') mark=true;t=getchar();}
    for(d=0;t>='0'&&t<='9';t=getchar())d=(d<<3)+(d<<1)-'0'+t;
    if(mark)d=-d;
}  
char o[100];
inline void out_(int x)
{   
    int l=1;
    if(x<0) x=-x,putchar('-');  
    if(!x) putchar('0');  
    for(;x;x/=10) o[l++]=x%10+'0';  
    for(l--;l;l--)putchar(o[l]);  
    //putchar('\n');  
}  

#define lowbit(x) x&-x
ll  getsum(ll *c,int x)
{
    ll ans=0;
    for(;x>0;x-=lowbit(x)) ans+=c[x];
    return ans;
}
void modify(ll *c,int x,ll d)
{
    for(;x<=n;x+=lowbit(x)) c[x]+=d;
}


int main()
{
    in_(n);//scanf("%d",&n);
    long long ans=0,t;
    for(int i=1,a;i<=n;i++)
    {
        in_(a);//scanf("%d",&a);
        t=getsum(c1,a-1);
        ans+=t*(t-1)/2;
        t=getsum(c2,a-1);
        ans-=t;
        t=getsum(c1,a);
        modify(c2,a,t);
        modify(c1,a,1);
    }
    printf("%I64d",ans);
}