树状数组的应用和变形

力扣:2179. 统计数组中好三元组数目
首先我只想说这个题终于过了呜呜呜呜
一开始的时候,做一个映射…得到一个数组
比如说[1,3,2,4,5]
实际上就是找这个数组里面的三元组有多少,形如(a,b,c),其中a<b<c,思路可以是这样的,找二元组:(a,b),(b,c),其中a<b,b<c,然后把它们存在一个数组里面,最后只要算x*y,再把他们加起来就ok了,那么咋存这个数组可以参考前文,需要注意的是,找比b大的数,在tree_node里面也要+1,这样做是保证在k和k+lowbit(k)之间的所有数可以知道有一个比他们小的,因为这个b以后不会再出现了,所以这样做是没问题的

树状数组的变形应用,主要体现在(b,c)的数量计算,你需要知道每一个元素,他的右边有多少是比他大的,这个咋做?
传统树状数组是一个数组向上传播,然后向下求和,那么是否可以向下传播,然后再向上求和,答案是可以的;这个可以参考前文是否会加重。
不过这里有一个误区:就是向上求和的时候,求到哪块为止?其实只要限制,k+lowbit(k)<max就可以了,而不用管max的最高位到max这个区间,你只要在max上加一个1,这样的话,在max最高位到max这一段上是由max来负责的。
贴个代码:

#include<stdio.h>
#include <string.h>

int index1[10000];
int nums[10000];//这个数组长度记得一会改了
long long tree_node[10000];
long long tree_new[10000];
long long reco_left[10000];
long long reco_right[10000];

int check(int x) {
    while (x > 1) {
        if (x % 2 == 1)
            return 0;
        x = x / 2;
    }
    return 1;
}

int lowbit(int x) {
    return x & (-x);
}

long long cal(long long p[], int x) {
    long long sum = 0;
    while (x > 0) {
        sum += p[x];
        x = x - lowbit(x);
    }
    return sum;
}

long long cal1(long long p[], int x, int n) {
    long long sum = 0;
    while (x <= n) {
        sum += p[x];
        //   printf(" x:%d %d\n",p[x],x);
        x = x + lowbit(x);
    }
    return sum;
}

long long goodTriplets(int nums1[], int nums1Size, int nums2[], int nums2Size) {

    //这么做的目的是把0给顶上去
    for (int i = 0; i < nums2Size; i++) {
        index1[nums2[i]] = i + 1;
    }
    for (int i = 0; i < nums1Size; i++) {
        nums[i] = index1[nums1[i]];
    }
    //下面实现一个树状数组

    reco_left[0] = 0;

    for (int i = 0; i < nums1Size + 1; i++) {
        tree_node[i] = 0;//初始数组元素数量其意义是小于当前元素的元素数量for
        tree_new[i] = 0;
        reco_left[i] = 0;
    }

    for (int i = 0; i < nums1Size; i++) {
        int k = nums[i];
        long long x = cal(tree_node, nums[i]);
        reco_left[nums[i]] = x;
        while (k < nums1Size + 1) {
            tree_node[k] = tree_node[k] + 1;
            k = k + lowbit(k);
        }
    }

    reco_right[0] = 0;
    for (int i = 1; i < nums1Size + 1; i++) {
        reco_right[i] = 0;

    }

    for (int i = nums1Size - 1; i >= 0; i--) {
        int k = nums[i];
        //这个数往下(左)传,然后往(右)算
        long long x = cal1(tree_new, k, nums1Size);
        reco_right[nums[i]] = x;
        //  printf("%d %d %d\n",nums[i],reco_right[nums[i]],x);
        while (k > 0) {
            tree_new[k] += 1;
            k = k - lowbit(k);
        }
        //然后往右算
    }

    long long sum = 0;

    for (int i = 1; i < nums1Size + 1; i++) {
        sum = sum + reco_right[i] * reco_left[i];
    }
    return sum;
}

int main(void) {
    int nums1[5] = {4, 0, 1, 3, 2};
    int nums2[5] = {4, 1, 0, 2, 3};
    int a = 5;
    int b = 5;
    int c = goodTriplets(nums1, a, nums2, b);
    printf("%d", c);
    }

结语:树状数组在做的时候不能把0算进来

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值