int lowbit(int x)
{
return x&-x;
}
void update(int x, int k)
{
for(int i = x; i <= n; i += lowbit(i))
t[i] += k;
}
int query(int x)
{
ll sum = 0;
for(int i = x; i;i -= lowbit(i))
sum += t[i];
return sum;
}
树状数组是动态维护前缀和
的数据结构
通过维护差分数组可以在 l o g 2 n log_2n log2n的时间复杂度下进行区间修改和区间查询;
int sum1[N],sum2[N];// 1.用来维护 d[i];2.用来维护 d[i]*i;
void update(int p,int x)
{
for(int i = p;i;i+=lowbit(i))
sum1[i] += x,sum2[i] += p*x;
}
ll query(int x)
{
ll res = 0;
for(int i = x; i; i -= lowbit(i))
res += (x+1)*sum1[i] - sum2[i];
return res;
}
用树状数组求逆序对
//结构体离散化数组;
struct node{
int data,id;
}a[N];
int new[N];
for i : 1 -> n cin >> a[i].data,a[i].id = i;
sort(...)// cmp{if a.data==b.data return a.id < b.id;return a.data < b. data};
for i:i -> n new[a[i].id] = i;
求逆序数
int new[N];//需要离散化得到的输入数组;
ll res = 0;
for(int i = 1; i <=n; i++){
update(new[i],1);
res += (i - query(new[i]));// (该数所在位置 - 比他小的数 - 1),逆序对第二个数;
}
离散化+树状数组+逆序对
题号 |
---|
51Nod - 1107 |
CodeForces 1042D |
出题人的手环(牛客) |
HDU - 1394 |