树状数组逆序对c语言,牛客小白月赛 树状数组求逆序对+离散化

题目描述

出题人的妹子送了出题人一个手环，这个手环上有 n 个珠子，每个珠子上有一个数。

有一天，出题人和妹子分手了，想把这个手环从两个珠子间切开，并按顺时针顺序展开成一条链。

可以发现，这条链一共有 n 种可能性。求这 n 种可能性的逆序对数之积模 1000000007。

输入描述:

第一行一个数 n，表示珠子个数。

接下来一行 n 个数，以顺时针顺序给出每个珠子上的整数

输出描述:

一个数，表示答案。

示例1

输入

4

1 3 2 3

输出

24

说明

一共有 4 种方式：

1 3 2 3;3 1 3 2;2 3 1 3;3 2 3 1;

逆序对数分别为 1，3，2，4，积为 24。

备注:

n<=200000，-10^9<=珠子上的整数<=10^9。

思路：

离散化树状数组求逆序对。最后枚举每一个珠子作为最后一个珠子的情况计算贡献更新答案即可。

PS：离散化树状数组就是先对原序列排序，然后为每一个元素添加一个相对大小的标号，元素权值相同的元素标号相同，然后在标号范围上维护树状数组。

AC代码：

#include

#include

#include

#include

#include

#include

#include

#include

#include

#include

#include

#include

#define ll long long

using namespace std;

struct fun

{

int v,index;

}a[240000];

int c[240000];

int num[240000];

bool cmp1(fun aa,fun bb)

{

return aa.v

}

void add(int x,int v,int n)//注意这里的n指的是树状数组的长度，不是原a【】数组的长度.

{

for(;x<=n;x+=x&-x)c[x]=(c[x]+v)%1000000007;

}

int ask(int x)

{

int ans=0;

for(;x;x-=x&-x)ans=(ans+c[x])%1000000007;

return ans;

}

int ans[250000+10];

int mer[250000+10];

int cnt[250000+10];

int main()

{

int n;

scanf("%d",&n);

for(int i=1;i<=n;++i)scanf("%d",&a[i].v),a[i].index=i;

sort(a+1,a+1+n,cmp1);

num[a[1].index]=1;

int kk=1;

cnt[1]=1;//树状数组中<=1的元素有几个

for(int i=2;i<=n;++i)

{

if(a[i].v==a[i-1].v)num[a[i].index]=num[a[i-1].index];

else num[a[i].index]=num[a[i-1].index]+1,kk++;//kk！=n

cnt[kk]=i;

}

ll sum=0;

for(int i=1;i<=n;++i)

{

sum=(sum+i-1-ask(num[i])%1000000007)%1000000007;//ask查询<=num【i】的数的个数

add(num[i],1,kk);//注意树状数组范围

}

ll ans=sum%1000000007;

for(int i=1;i

{

sum=(sum-cnt[num[i]-1]+n-cnt[num[i]])%1000000007;//直接通过树状数组上顺序看对于1~n每一个元素有几个比它大的或者小的，注意cnt【】的含义。

sum=(sum+1000000007)%1000000007;

ans=(ans%1000000007*sum%1000000007)%1000000007;

}

printf("%lld\n",ans);

}

The end；