今天入门了树状数组 习题链接 https://blog.csdn.net/liuqiyao_01/article/details/26963913
离散化数据:用一个数组来记录每个值在数列中的排名,不能用map,会超时
开结构体存储每个数在数列中原来的位置,排序后用a数组求出原来状态下每个数的排名。
/* 线段树求逆序对的话是按顺序把数字插到数字对应的线段树下标里,然后统计该下标右边的数个数 树状数组求逆序对思路:按顺序一个个插到数组中,统计比它小的个数,逆序数=当前数的下标-当前比它小的个数 */ #include<iostream> #include<cstring> #include<cstdio> #include<algorithm> #define maxn 500500 #define ll long long using namespace std; struct node{ int val,pos; bool operator<(const node & a)const { return val<a.val; } }p[maxn]; int n,bit[maxn],a[maxn]; //树状数组bit[i]维护的信息是排名在一个区间内的数字个数 //数组a表示第i个数的排名是多少 void add(int i){ while(i<=n){ bit[i]+=1; i+=i&-i; } } int sum(int i){ int s=0; while(i>0){ s+=bit[i]; i-=i&-i; } return s; } void solve(){ for(int i=1;i<=n;i++){scanf("%d",&p[i].val);p[i].pos=i;} sort(p+1,p+1+n); for(int i=1;i<=n;i++)a[p[i].pos]=i; memset(bit,0,sizeof bit); ll ans=0; for(int i=1;i<=n;i++){ add(a[i]);//排名为a[i]的地方加1 ans+=i-sum(a[i]);//当前所有被加入的数个数-排名在第i的数之前的数=增加的逆序对数 } printf("%lld ",ans); } int main(){ while(scanf("%d",&n),n)solve(); }