力扣: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算进来