BZOJ 1145: [CTSC2008]图腾totem 数据结构维护，思维题

1145: [CTSC2008]图腾totem

Time Limit: 30 Sec Memory Limit: 162 MB
Submit: 446 Solved: 269

Description

　　在完成了古越州圆盘密码的研究之后，考古学家小布来到了南美大陆的西部。相传很久以前在这片土地上生活

着两个部落，一个部落崇拜闪电，另一个部落崇拜高山，他们分别用闪电和山峰的形状作为各自部落的图腾。小布

的团队在山洞里发现了一幅巨大的壁画，壁画上被标记出了N个点，经测量发现这N个点的水平位置和竖直位置是两

两不同的。小布认为这幅壁画所包含的信息仅与这N个点的相对位置有关，因此不妨设坐标分别为(1, y1) , (2, y

2), …, (n, yn)，其中y1~yn是1~N的一个排列。小布的团队打算研究在这幅壁画中包含着多少个图腾，其中闪电

图腾的定义图示如下（图腾的形式只与4个纵坐标值的相对大小排列顺序有关）：

　　崇拜高山的部落有两个氏族，因而山峰图腾有如下两种形式，左边为A类，右边为B类（同样，图腾的形式也都

只与4个纵坐标值的大小排列顺序有关）：

　　小布的团队希望知道，这N个点中两个部落图腾数目的差值。因此在本题中，你需要帮助小布的团队编写一个

程序，计算闪电图腾数目减去山峰图腾数目的值，由于该值可能绝对值较大，本题中只需输出该值对16777216的余

数（注意余数必为正值，例如-1对16777216的余数为16777215）。

Input

　　第一行包含一个整数N，为点的数目。接下来一行包含N个整数，分别为y1, y2, …, yn。保证y1, y2, …, yn

是1~N的一个排列。N ≤ 200000

Output

　　仅包含一个数，表示闪电图腾数目与山峰图腾数目的差值对16777216的余数。

Sample Input

【样例输入一】

5

1 5 3 2 4

【样例输入二】

4

1 2 4 3

Sample Output

【样例输出一】

0

【样例输出二】

16777215

HINT

样例一中共有1个闪电图腾（1324）和1个B类山峰图腾（1532）。样例二中仅有一个A类山峰图腾（1243），故差值为-1，答案为16777215。

---

mod XGG，引用一发PPT的内容

#include<cstring>
#include<iostream>
#include<cstdio>
#define N 200010
const int XGG=16777216;
using namespace std;
long long C[N],L[N],R[N];
int H[N],n;
int lowbit(int x){return x&(-x);}
void modify(int loc,int val){
    while(loc<=n){
        C[loc]+=val;
        loc+=lowbit(loc);
    }
}
long long query(int loc){
    long long temp=0;
    while(loc){
        temp+=C[loc];
        loc-=lowbit(loc);
    }
    return temp;
}
int solve1_2_(){
    memset(C,0,sizeof(C));
    long long temp=0;
    for(register int i=1;i<=n;i++){
        temp+=(L[i]*(i-1)-L[i]*(L[i]-1)/2-query(H[i]))*(n-R[i]-i);
        temp%=XGG; 
        modify(H[i],i);//所以H[i]的左边又有i个点了,维护进去 
    }//存的是所有小于h[i]高度的左下角点的左边有多少点 
    return temp;
}
int solve1234(){
    memset(C,0,sizeof(C));
    long long temp=0;
    for(register int i=1;i<=n;i++){
        temp+=query(H[i])*(n-i-R[i]);
        temp%=XGG;
        modify(H[i],L[i]);
    }
    return temp;
}
int solve13__(){
    memset(C,0,sizeof(C));
    long long temp=0;
    for(register int i=1;i<=n;i++){
        temp+=(L[i]*(H[i]-1)-L[i]*(L[i]-1)/2-query(H[i]))*(n-i-R[i]);
        temp%=XGG;
        modify(H[i],H[i]);
    }
    return temp;
}
int solve1___(){
    memset(C,0,sizeof(C));
    long long temp=0;
    for(register int i=1;i<=n;i++){
        long long x=n-i-R[i];//此为其右上的点
        if(x<3) continue;
        temp+=x*(x-1)*(x-2)/6;
        temp%=XGG; 
    }
    return temp;
}
int main(){
    scanf("%d",&n);
    for(register int i=1;i<=n;i++){
        scanf("%d",&H[i]);
        L[i]=query(H[i]);
        R[i]=H[i]-L[i]-1;
        modify(H[i],1);
    }
    printf("%d\n",(solve1_2_()+solve1234()-solve1___()+solve13__()+XGG)%XGG);
    return 0;
}