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;
}