首页 > 代码库 > hdu 2838 coew sorting(树状数组)
hdu 2838 coew sorting(树状数组)
求逆序对的和
分析:其实这个结果和逆序数有关,对某个位置i,如果前面比他大的有x个,那么a[i]至少要加x次
如果后面有y个比a[i]小,那么a[i]至少要加y次,也就是说用两个树状数组来分别维护当前位置时前面有多少比他大,后面有多少个比他小
如求逆序数两两的总和: 如 3 2 1 :sum=(3+2)+(3+1)+(2+1)=12; 1 2 3: sum=0;
对于新插入的一个元素,运用树状数组,可以求得比它小的元素的个数,比它小的元素的和,在它之前的元素的总和。
而对于每一个新元素,其sum[m]=m*(比它大的元素个数)+(前i个元素的和)-(比它小的元素的和)。
然后累加得解。
可得出公式
#include <cstdio> #include <cstring> #include <algorithm> using namespace std; typedef __int64 ll; struct node { int id; ll sum; }s[100005]; int n; //更新个个节点 void add(int i,int j,int k){ while(i<=n){ s[i].id+=j; s[i].sum+=k; i+=i&-i; } } //此函数求逆序对的个数 int get_id(int i){ int idd=0; while(i>0){ idd+=s[i].id; i-=i&-i; } return idd; } //求前i个数的和 ll get_sum(int i){ ll summ=0; while(i>0){ summ+=s[i].sum; i-=i&-i; } return summ; } int main(){ while(~scanf("%d",&n)) { int x; memset(s,0,sizeof(s)); ll ans=0,k; for(int i=1;i<=n;i++) { scanf("%d",&x); add(x,1,x); ll k=(i-get_id(x)); if(k==0) continue; ans+=(k*x+get_sum(n)-get_sum(x)); } printf("%I64d\n",ans); } }
hdu 2838 coew sorting(树状数组)
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。