1009. Triple Inversions (35)

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/yhf_naive/article/details/61418486

感想:1。树状数组先求二阶的逆序数(权重为1),然后再用二阶逆序数求三阶逆序数(权重为二阶逆序数)(三阶逆序数可以看成带二阶逆序数权重的二阶逆序数)

2.数据范围会超过int所以用long long

#include<iostream>
#include<string>
using namespace std;\
long long a[200010];
long long b[200010];
int c[200010];
int N;
void insert(int i,int t){
	while(i<=2*N){
		a[i]+=t;
		i+=(i&-i);
	}
}
long long read(int i){
	long long sum=0;
	while(i>0){
		sum+=a[i];
		i-=(i&-i);
	}
	return sum;
}
int main(){
	int i,temp;
	long long ss=0;
	cin>>N;
	for(i=0;i<N;i++){
		cin>>temp;
		c[i]=N-temp+1;
		insert(N-temp+1,1);
		b[i]=read(N-temp+1)-1;
	}
	for(i=0;i<=2*N;i++)
		a[i]=0;
	for(i=0;i<N;i++){
		insert(c[i],b[i]);
		ss+=read(c[i])-b[i];
	}
	cout<<ss;
} 


猜你喜欢

转载自blog.csdn.net/yhf_naive/article/details/61418486