基础树状数组
LA 4329(2008年北京区域赛)
题目分析:首先是乘法原理加法原理计数,对a [i]当做裁判的情况,记c[i] d[i] 分别表示前面和后面小于它的数的个数,ai当裁判方法数就是 c[i]*(n-i-1-d[i])+d[i]*(i-c[i])了
具体的见《算法竞赛入门经典训练指南》197页
树状数组见194-197页
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=100001;
int C[maxn];
int a[20001],c[20001],d[20001];
int lowbit(int x) //x在二进制表达式中最右边的1所对应的值
{
return x&-x;
}
int sum(int x) //求前缀和
{
int ret=0;
while(x>0)
{
ret+=C[x]; //C[x]为以x结尾的水平长条内的元素之和
x-=lowbit(x);
}
return ret;
}
void add(int x,int d) //区间中修改一个x,使x增加d(有且仅有这些节点对应的长条包含被修改的元素)
{
while(x<=maxn)
{
C[x]+=d;
x+=lowbit(x);
}
}
int main()
{
int T;
cin>>T;
while(T--)
{
int n;
cin>>n;
memset(C,0,sizeof(C));
memset(c,0,sizeof(c));
memset(d,0,sizeof(d));
for(int i=0;i<n;i++)
{
cin>>a[i];
add(a[i],1);
c[i]=sum(a[i]-1);
}
memset(C,0,sizeof(C));
for(int i=n-1;i>=0;i--)
{
add(a[i],1);
d[i]=sum(a[i]-1);
}
long long ans=0;
for(int i=1;i<n-1;i++)
ans+=c[i]*(n-i-d[i]-1)+d[i]*(i-c[i]);
cout<<ans<<endl;
}
return 0;
}

浙公网安备 33010602011771号